Update cpp/src/neighbors/detail/cagra/search_single_cta_kernel-inl.cuh

Co-authored-by: Artem M. Chirkin <[email protected]>
rapidsai · Dec 3, 2024 · 4ad78b3 · 4ad78b3
1 parent e8ad300
commit 4ad78b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/neighbors/detail/cagra/search_single_cta_kernel-inl.cuh b/cpp/src/neighbors/detail/cagra/search_single_cta_kernel-inl.cuh
@@ -478,7 +478,7 @@ RAFT_DEVICE_INLINE_FUNCTION void move_invalid_to_end_of_list(IdxT* const index_a
       } else {
         // Check if the index is invalid
         const auto I_found_invalid = (index == invalid_index);
-        const auto who_has_invalid = __ballot_sync(~0u, I_found_invalid);
+        const auto who_has_invalid = raft::ballot(I_found_invalid);
         // if a value that is loaded by a smaller lane id thread, shift the array
         if (who_has_invalid << (warp_size - lane_id)) {
           index_array[i - 1]    = index;