Minor cleanup

NVIDIA · Nov 20, 2024 · 703da6a · 703da6a
1 parent bc601e1
commit 703da6a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/include/cuco/detail/open_addressing/open_addressing_ref_impl.cuh b/include/cuco/detail/open_addressing/open_addressing_ref_impl.cuh
@@ -1230,11 +1230,10 @@ class open_addressing_ref_impl {
                 num_matches[i] = __popc(exists[i]);
               }
 
-              auto const total_matches =
-                thrust::reduce(thrust::seq, num_matches, num_matches + bucket_size);
-
               int32_t output_idx;
               if (lane_id == 0) {
+                auto const total_matches =
+                  thrust::reduce(thrust::seq, num_matches, num_matches + bucket_size);
                 auto ref =
                   cuda::atomic_ref<int32_t, cuda::thread_scope_block>{counters[flushing_tile_id]};
                 output_idx = ref.fetch_add(total_matches, cuda::memory_order_relaxed);