Update src/cuda/CUDAStream.cu

UoB-HPC · May 13, 2024 · eede871 · eede871
1 parent 28c8cc3
commit eede871
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/cuda/CUDAStream.cu b/src/cuda/CUDAStream.cu
@@ -237,6 +237,7 @@ __global__ void dot_kernel(const T * a, const T * b, T* sums, int array_size)
     if (tidx < offset) smem[tidx] += smem[tidx+offset];
   }
 
+  // First thread writes to host memory directly from the device
   if (tidx == 0) sums[blockIdx.x] = smem[tidx];
 }