fix randomly fail on large size(2nd try)

rapidsai · Dec 8, 2024 · 45292f0 · 45292f0
1 parent fec566a
commit 45292f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/include/raft/sparse/convert/detail/bitmap_to_csr.cuh b/cpp/include/raft/sparse/convert/detail/bitmap_to_csr.cuh
@@ -230,8 +230,8 @@ RAFT_KERNEL __launch_bounds__(bitmap_to_csr_tpb)
     int l_bits    = raft::detail::popc(l_bitmap);
     int l_sum_32b = 0;
     BlockScan(scan_storage).InclusiveSum(l_bits, l_sum_32b);
-    __syncthreads();
     l_sum = l_sum_32b + g_sum - l_bits;
+    __syncthreads();
 
 #pragma unroll
     for (int i = 0; i < BITS_PER_BITMAP; i++) {