fix atomic scope

rapidsai · Jun 15, 2024 · 530de96 · 530de96
1 parent dabad8f
commit 530de96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/strings/replace/replace.cu b/cpp/src/strings/replace/replace.cu
@@ -258,7 +258,7 @@ CUDF_KERNEL void count_targets_kernel(replace_parallel_chars_fn fn,
   auto const total = block_reduce(temp_storage).Reduce(count, cub::Sum());
 
   if ((lane_idx == 0) && (total > 0)) {
-    cuda::atomic_ref<int64_t, cuda::thread_scope_block> ref{*d_output};
+    cuda::atomic_ref<int64_t, cuda::thread_scope_device> ref{*d_output};
     ref.fetch_add(total, cuda::std::memory_order_relaxed);
   }
 }