Repository navigation
perf: ANN optimizations - SwissHashSet, AVX2 distance, insert_fast - #1252
Merged
Merged
Conversation
| } | ||
|
|
||
| template <typename data_t> | ||
| void PQL2Distance<data_t>::prepopulate_chunkwise_distances(const float *query_vec, float *dist_vec) |
Contributor
There was a problem hiding this comment.
this function isn't called in current scenarios, the change should not take effect
Deyu Kong (cave-g-f)
force-pushed
the
deyu/agentic2
branch
from
July 13, 2026 09:44
51edf73 to
18d127e
Compare
Port hot-path optimizations from parent repo PR 6610806 (commit 844664001): - swiss_hash_set.h: New SSE2/SIMD-optimized Swiss table hash set for uint32_t keys with O(occupied_slots) clear and ~87.5% load factor. Replaces tsl::robin_set for inserted_into_pool_rs. - neighbor.h: Add NeighborMinHeap::insert_fast() with AVX2-accelerated linear scan over a contiguous float distance array (_dist_arr) for faster insertion-point finding. - index.cpp: Refactor iterate_to_fixed_point() to interleave prefetch, distance computation, and immediate insert; remove separate dist_scratch allocation; mark nodes visited and process neighbors under the same shared lock (eliminates redundant lock/unlock). - pq_l2_distance.cpp: AVX2 inner loop for chunk-wise PQ centroid distance computation (8-wide _mm256_fmadd_ps). - scratch.h/.cpp: Switch _inserted_into_pool_rs type to SwissHashSet and skip clear() on empty containers.
Deyu Kong (cave-g-f)
force-pushed
the
deyu/agentic2
branch
from
July 13, 2026 09:45
18d127e to
17c0a25
Compare
Deyu Kong (cave-g-f)
merged commit Jul 13, 2026
d759aed
into
jegao/LabelHotFix
3 of 39 checks passed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Port hot-path optimizations from parent repo PR 6610806 (commit 844664001):
Reference Issues/PRs
What does this implement/fix? Briefly explain your changes.
Any other comments?