add sync

345737d0 · chengduoZH · 187e23a7 · 345737d0
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/platform/cuda_device_function.h paddle/fluid/platform/cuda_device_function.h +1 -0

未找到文件。
--- a/paddle/fluid/platform/cuda_device_function.h
+++ b/paddle/fluid/platform/cuda_device_function.h
@@ -63,6 +63,7 @@ __device__ T reduceSum(T val, int tid, int len) {
    val += platform::CudaShuffleDownSync(mask, val, offset);

  if (tid < warpSize) shm[tid] = 0;
+  __syncthreads();

  if (tid % warpSize == 0) {
    shm[tid / warpSize] = val;