Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Prerequisites for FAISS HNSW #731

Merged
merged 6 commits into from
Jul 29, 2024

Conversation

alexanderguzhva
Copy link
Collaborator

A collection of small tweaks that might be needed in the future.

  • Adds another statistics for HNSW called nhops, which is a number of different nodes visited through a single query search
  • AVX-512 code for PQDecoder8
  • Binary heap utilities for data stored in std::pair<dis,ids> format
  • Fix for HNSW::upper_beam != 1 search use case
  • AVX-512 code HNSW::MinimaxHeap::pop_min

Additionally, fixes a bug in ScalarQuantizerCoder_avx512

Corresponding baseline Faiss PR: facebookresearch/faiss#3692

/kind improvement

Signed-off-by: Alexandr Guzhva <alexanderguzhva@gmail.com>
Signed-off-by: Alexandr Guzhva <alexanderguzhva@gmail.com>
Signed-off-by: Alexandr Guzhva <alexanderguzhva@gmail.com>
Signed-off-by: Alexandr Guzhva <alexanderguzhva@gmail.com>
…gned

Signed-off-by: Alexandr Guzhva <alexanderguzhva@gmail.com>
Signed-off-by: Alexandr Guzhva <alexanderguzhva@gmail.com>
Copy link

codecov bot commented Jul 25, 2024

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 71.97%. Comparing base (3c46f4c) to head (ca6908e).
Report is 130 commits behind head on main.

Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff            @@
##           main     #731       +/-   ##
=========================================
+ Coverage      0   71.97%   +71.97%     
=========================================
  Files         0       70       +70     
  Lines         0     5160     +5160     
=========================================
+ Hits          0     3714     +3714     
- Misses        0     1446     +1446     

see 70 files with indirect coverage changes

@mergify mergify bot added the ci-passed label Jul 25, 2024
Copy link
Collaborator

@liliu-z liliu-z left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

/lgtm

template <class C>
inline void heap_push(
size_t k,
std::pair<typename C::T, typename C::TI>* bh,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Curios about why we need to add another function with pair

@sre-ci-robot
Copy link
Collaborator

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: alexanderguzhva, liliu-z

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@liliu-z
Copy link
Collaborator

liliu-z commented Jul 29, 2024

/lgtm

@sre-ci-robot sre-ci-robot merged commit 752e5e0 into zilliztech:main Jul 29, 2024
13 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants