Skip to content

Commit

Permalink
Squared norm fix follow-up (change was lost in merge conflict) (#1144)
Browse files Browse the repository at this point in the history
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: #1144
  • Loading branch information
Nyrio authored Jan 15, 2023
1 parent 2af2749 commit efd42c9
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions cpp/include/raft/spatial/knn/detail/ivf_flat_search.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -1133,8 +1133,7 @@ void search_impl(const handle_t& handle,
static_cast<IdxT>(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(),
Expand Down

0 comments on commit efd42c9

Please sign in to comment.