From c3fb37cd2548aef3a061093ab69ed6a137bb4eb0 Mon Sep 17 00:00:00 2001 From: Dante Gama Dessavre Date: Wed, 26 Aug 2026 09:58:42 -0500 Subject: [PATCH] FIX stream ordering for brute-force L2 sqrt post-processing --- cpp/src/neighbors/detail/knn_brute_force.cuh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cpp/src/neighbors/detail/knn_brute_force.cuh b/cpp/src/neighbors/detail/knn_brute_force.cuh index 76ed71fe57..7888282084 100644 --- a/cpp/src/neighbors/detail/knn_brute_force.cuh +++ b/cpp/src/neighbors/detail/knn_brute_force.cuh @@ -476,8 +476,10 @@ void brute_force_knn_impl( metric == cuvs::distance::DistanceType::LpUnexpanded) { DistType p = 0.5; // standard l2 if (metric == cuvs::distance::DistanceType::LpUnexpanded) p = 1.0 / metricArg; + raft::resources stream_handle(handle); + raft::resource::set_cuda_stream(stream_handle, stream); raft::linalg::map( - handle, + stream_handle, raft::make_device_vector_view(res_D, n * k), [p] __device__(DistType input) { return powf(fabsf(input), p); }, raft::make_const_mdspan(