minor bug fix in warp synchronous code (#7029)

apache · Jul 13, 2017 · b5615f5 · b5615f5
1 parent fa2c0a3
commit b5615f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/tensor/dot-inl.cuh b/src/operator/tensor/dot-inl.cuh
@@ -71,7 +71,7 @@ struct DotCsrDnsDnsVectorKernel {
     for (int j = low+lane; j < high; j+=32) {
       sum += data_l[j] * data_r[col_idx_l[j]*num_cols_r + kcol];
     }
-    vals[threadIdx.x] = sum;
+    vals[threadIdx.x] = sum; __syncwarp();
 
     // Parallel reduction in shared memory
     if (lane < 16) {vals[threadIdx.x] += vals[threadIdx.x+16];} __syncwarp();