From efd42c9f34fc16b28e94578e7e8460685fa2012d Mon Sep 17 00:00:00 2001 From: Louis Sugy Date: Sun, 15 Jan 2023 17:26:50 +0100 Subject: [PATCH] Squared norm fix follow-up (change was lost in merge conflict) (#1144) This change was part of #1141 but was accidentally lost while merging conflicts with #1133 Authors: - Louis Sugy (https://github.com/Nyrio) Approvers: - Corey J. Nolet (https://github.com/cjnolet) URL: https://github.com/rapidsai/raft/pull/1144 --- cpp/include/raft/spatial/knn/detail/ivf_flat_search.cuh | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/cpp/include/raft/spatial/knn/detail/ivf_flat_search.cuh b/cpp/include/raft/spatial/knn/detail/ivf_flat_search.cuh index fac8519a03..b139d8df8c 100644 --- a/cpp/include/raft/spatial/knn/detail/ivf_flat_search.cuh +++ b/cpp/include/raft/spatial/knn/detail/ivf_flat_search.cuh @@ -1133,8 +1133,7 @@ void search_impl(const handle_t& handle, static_cast(n_queries), raft::linalg::L2Norm, true, - stream, - raft::sqrt_op()); + stream); utils::outer_add(query_norm_dev.data(), (IdxT)n_queries, index.center_norms()->data_handle(),