aparler/PyNorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add comment on reduce cuda

Browse source
This commit is contained in:
lucasdelimanogueira 2024-06-04 11:21:56 -03:00
parent 12eb38a940
commit b56ed81cd7
2 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

0
.gitmodules vendored
View file

2
norch/csrc/cuda.cu
View file

@ -136,7 +136,7 @@ __global__ void sum_tensor_cuda_kernel(float* data, float* result_data, int size
__syncthreads();
// Perform block-wise reduction
for (int s = blockDim.x / 2; s > 0; s >>= 1) {
for (int s = blockDim.x / 2; s > 0; s >>= 1) { // s>>=1 --> s = s/2
if (tid < s) {
partial_sum[tid] += partial_sum[tid + s];
}