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.

Sign up for GitHub

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

Jump to bottom

Support graph-based approximate nearest neighbors search (cagra on nn_descent/ivf_pq) #714

Merged
merged 15 commits into from
Sep 3, 2024
Merged

Support graph-based approximate nearest neighbors search (cagra on nn_descent/ivf_pq) #714

merged 15 commits into from
Sep 3, 2024

Commits on Aug 30, 2024

  1. two cagra running and two tests passed 

    Signed-off-by: Jinfeng <[email protected]>
    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    c3cf04c View commit details
    Browse the repository at this point in the history

  2. add index parameters intermediate_graph_degree, graph_degree, and bui… 

    …ld_algo
    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    63d5d86 View commit details
    Browse the repository at this point in the history

  3. add docstring for cagra and ivfpq

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    adc5c53 View commit details
    Browse the repository at this point in the history

  4. add docstring for cagra build index params

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    4425102 View commit details
    Browse the repository at this point in the history

  5. use get_default_params_from_func to separate index params and search … 

    …params
    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    047f688 View commit details
    Browse the repository at this point in the history

  6. get cagra params working and tested

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    4a423f6 View commit details
    Browse the repository at this point in the history

  7. add cuvs to ci and main README

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    23ecc68 View commit details
    Browse the repository at this point in the history

  8. add cuvs to pip install command

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    b4fd36b View commit details
    Browse the repository at this point in the history

  9. add a link to docstring and get list working properly

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    2709759 View commit details
    Browse the repository at this point in the history

  10. revise testing to assert ann distances indices shape

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    bff46e9 View commit details
    Browse the repository at this point in the history

  11. enable cagra works for multi_cols pd.dataframe input

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    3d85325 View commit details
    Browse the repository at this point in the history

  12. get graph_degree and itopk_size tested

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    573f3c6 View commit details
    Browse the repository at this point in the history

  13. revise description of itopk_size to be more accurate

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    8902da7 View commit details
    Browse the repository at this point in the history

  14. validate itopk_size before calling cuml

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    e14b1d0 View commit details
    Browse the repository at this point in the history

  15. revise per comments

    @lijinf2
    lijinf2 committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    788f1ac View commit details
    Browse the repository at this point in the history