Fix new distance

rapidsai · Nov 23, 2024 · ed10ed2 · ed10ed2
1 parent 3d35cdc
commit ed10ed2
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
@@ -491,7 +491,7 @@ RAFT_DEVICE_INLINE_FUNCTION void move_invalid_to_end_of_list(IdxT* const index_a
   }
   if (lane_id == 0) {
     index_array[array_length - 1]    = invalid_index;
-    distance_array[array_length - 1] = -1;
+    distance_array[array_length - 1] = utils::get_max_value<float>();
   }
 }