From c235b0ae1a773f62697b23a42e90c3a48f3abbf6 Mon Sep 17 00:00:00 2001 From: Zefeng Yin Date: Wed, 28 Jan 2026 16:16:23 +0800 Subject: [PATCH] change single dist to batch dist --- src/core/algorithm/hnsw/hnsw_searcher.cc | 8 ++++---- src/core/algorithm/hnsw/hnsw_streamer.cc | 6 +++--- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/core/algorithm/hnsw/hnsw_searcher.cc b/src/core/algorithm/hnsw/hnsw_searcher.cc index ac3cdefd..2e066873 100644 --- a/src/core/algorithm/hnsw/hnsw_searcher.cc +++ b/src/core/algorithm/hnsw/hnsw_searcher.cc @@ -276,7 +276,7 @@ int HnswSearcher::search_bf_impl(const void *query, const IndexQueryMeta &qmeta, continue; } if (!ctx->filter().is_valid() || !ctx->filter()(entity_.get_key(id))) { - dist_t dist = ctx->dist_calculator().dist(id); + dist_t dist = ctx->dist_calculator().batch_dist(id); std::string group_id = group_by(id); @@ -299,7 +299,7 @@ int HnswSearcher::search_bf_impl(const void *query, const IndexQueryMeta &qmeta, continue; } if (!ctx->filter().is_valid() || !ctx->filter()(entity_.get_key(id))) { - dist_t dist = ctx->dist_calculator().dist(id); + dist_t dist = ctx->dist_calculator().batch_dist(id); ctx->topk_heap().emplace(id, dist); } } @@ -364,7 +364,7 @@ int HnswSearcher::search_bf_by_p_keys_impl( if (!ctx->filter().is_valid() || !ctx->filter()(pk)) { node_id_t id = entity_.get_id(pk); if (id != kInvalidNodeId) { - dist_t dist = ctx->dist_calculator().dist(id); + dist_t dist = ctx->dist_calculator().batch_dist(id); std::string group_id = group_by(id); auto &topk_heap = ctx->group_topk_heaps()[group_id]; @@ -387,7 +387,7 @@ int HnswSearcher::search_bf_by_p_keys_impl( if (!ctx->filter().is_valid() || !ctx->filter()(pk)) { node_id_t id = entity_.get_id(pk); if (id != kInvalidNodeId) { - dist_t dist = ctx->dist_calculator().dist(id); + dist_t dist = ctx->dist_calculator().batch_dist(id); ctx->topk_heap().emplace(id, dist); } } diff --git a/src/core/algorithm/hnsw/hnsw_streamer.cc b/src/core/algorithm/hnsw/hnsw_streamer.cc index e737ce1e..be01f5d0 100644 --- a/src/core/algorithm/hnsw/hnsw_streamer.cc +++ b/src/core/algorithm/hnsw/hnsw_streamer.cc @@ -709,7 +709,7 @@ int HnswStreamer::search_bf_impl( } if (!ctx->filter().is_valid() || !ctx->filter()(entity_.get_key(id))) { - dist_t dist = ctx->dist_calculator().dist(id); + dist_t dist = ctx->dist_calculator().batch_dist(id); std::string group_id = group_by(id); @@ -802,7 +802,7 @@ int HnswStreamer::search_bf_by_p_keys_impl( if (!ctx->filter().is_valid() || !ctx->filter()(pk)) { node_id_t id = entity_.get_id(pk); if (id != kInvalidNodeId) { - dist_t dist = ctx->dist_calculator().dist(id); + dist_t dist = ctx->dist_calculator().batch_dist(id); std::string group_id = group_by(id); auto &topk_heap = ctx->group_topk_heaps()[group_id]; @@ -828,7 +828,7 @@ int HnswStreamer::search_bf_by_p_keys_impl( if (!filter.is_valid() || !filter(pk)) { node_id_t id = entity_.get_id(pk); if (id != kInvalidNodeId) { - dist_t dist = ctx->dist_calculator().dist(id); + dist_t dist = ctx->dist_calculator().batch_dist(id); topk.emplace(id, dist); } }