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(),