From 4ad78b378b6e992427099197c93a99da5a632741 Mon Sep 17 00:00:00 2001
From: tsuki <12711693+enp1s0@users.noreply.github.com>
Date: Tue, 3 Dec 2024 23:27:19 +0900
Subject: [PATCH] Update
 cpp/src/neighbors/detail/cagra/search_single_cta_kernel-inl.cuh

Co-authored-by: Artem M. Chirkin <9253178+achirkin@users.noreply.github.com>
---
 cpp/src/neighbors/detail/cagra/search_single_cta_kernel-inl.cuh | 2 +-
 1 file changed, 1 insertion(+), 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
index 62086e398..f63231f17 100644
--- 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;