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

ivf_flat::index: hide implementation details #747

Merged
merged 147 commits into from
Aug 24, 2022
Merged

ivf_flat::index: hide implementation details #747

merged 147 commits into from
Aug 24, 2022

Commits on May 13, 2022

  1. inital commit and formatting cleanup

    @achirkin
    achirkin committed May 13, 2022
    Configuration menu
    Copy the full SHA
    35ab60d View commit details
    Browse the repository at this point in the history

Commits on May 16, 2022

  1. update save/load index function to work with cuann benchmark suite, s… 

    …eparate out cuivfl handle init in separate func
    @achirkin
    achirkin committed May 16, 2022
    Configuration menu
    Copy the full SHA
    24e8c4d View commit details
    Browse the repository at this point in the history

  2. Added benchmarks.

    @achirkin
    achirkin committed May 16, 2022
    Configuration menu
    Copy the full SHA
    cb8bcd2 View commit details
    Browse the repository at this point in the history

Commits on May 17, 2022

  1. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    884723c View commit details
    Browse the repository at this point in the history

  2. Add a missing parameter docs

    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    8c4a0a0 View commit details
    Browse the repository at this point in the history

  3. Adapt to the changes in the warpsort api

    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    070fd05 View commit details
    Browse the repository at this point in the history

  4. cleanup: use WarpSize constant

    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    83b6630 View commit details
    Browse the repository at this point in the history

  5. cleanup: remove unnecessary helpers

    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    3a2703c View commit details
    Browse the repository at this point in the history

  6. Use a more efficient warp_sort_filtered

    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    31bbaec View commit details
    Browse the repository at this point in the history

  7. Recover files that have only non-relevant changes to reduce the size … 

    …of the PR
    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    4b40181 View commit details
    Browse the repository at this point in the history

  8. wip: replacing explicit allocations with rmm buffers

    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    7e3041c View commit details
    Browse the repository at this point in the history

  9. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 17, 2022
    Configuration menu
    Copy the full SHA
    f6556b7 View commit details
    Browse the repository at this point in the history

Commits on May 18, 2022

  1. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 18, 2022
    Configuration menu
    Copy the full SHA
    f75761f View commit details
    Browse the repository at this point in the history

  2. Update cpp/include/raft/spatial/knn/detail/ann_quantized_faiss.cuh 

    Co-authored-by: Tamas Bela Feher <tfeher@nvidia.com>
    @achirkin @tfeher
    achirkin and tfeher authored May 18, 2022
    Configuration menu
    Copy the full SHA
    dd558b4 View commit details
    Browse the repository at this point in the history

  3. Update cpp/include/raft/spatial/knn/detail/ann_quantized_faiss.cuh 

    Co-authored-by: Tamas Bela Feher <tfeher@nvidia.com>
    @achirkin @tfeher
    achirkin and tfeher authored May 18, 2022
    Configuration menu
    Copy the full SHA
    94b3cbe View commit details
    Browse the repository at this point in the history

  4. wip: replace cudaMemcpy with raft::copy

    @achirkin
    achirkin committed May 18, 2022
    Configuration menu
    Copy the full SHA
    2be45a9 View commit details
    Browse the repository at this point in the history

  5. Simplified some cudaMemcpy invocations

    @achirkin
    achirkin committed May 18, 2022
    Configuration menu
    Copy the full SHA
    30c32a9 View commit details
    Browse the repository at this point in the history

Commits on May 19, 2022

  1. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    c8e7b4d View commit details
    Browse the repository at this point in the history

  2. Refactoring with helper functions

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    150a438 View commit details
    Browse the repository at this point in the history

  3. Make the scratch buf 3x L2 cache size

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    ddfb8cc View commit details
    Browse the repository at this point in the history

  4. Remove serialization code for now

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    b788e2e View commit details
    Browse the repository at this point in the history

  5. remove obsolete comment

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    3e1c14d View commit details
    Browse the repository at this point in the history

  6. Add a missing sync

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    a001999 View commit details
    Browse the repository at this point in the history

  7. Rename ann_quantized_faiss

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    2d08271 View commit details
    Browse the repository at this point in the history

  8. wip from manual allocations to rmm: updated some parts with pointer r… 

    …equirements CPU/GPU
    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    0f88aaa View commit details
    Browse the repository at this point in the history

  9. wip from manual allocations to rmm

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    363dfc9 View commit details
    Browse the repository at this point in the history

  10. fix style

    @achirkin
    achirkin committed May 19, 2022
    Configuration menu
    Copy the full SHA
    e5399f8 View commit details
    Browse the repository at this point in the history

Commits on May 20, 2022

  1. Set minimum memory pool size in radix_topk to 256 bytes

    @achirkin
    achirkin committed May 20, 2022
    Configuration menu
    Copy the full SHA
    306f5bf View commit details
    Browse the repository at this point in the history

  2. wip malloc-to-rmm: removed most of the manual allocations

    @achirkin
    achirkin committed May 20, 2022
    Configuration menu
    Copy the full SHA
    fd7d2ba View commit details
    Browse the repository at this point in the history

  3. misc cleanup

    @achirkin
    achirkin committed May 20, 2022
    Configuration menu
    Copy the full SHA
    403667a View commit details
    Browse the repository at this point in the history

  4. Refactoing; used raft::handle in place of cublas handle everywhere

    @achirkin
    achirkin committed May 20, 2022
    Configuration menu
    Copy the full SHA
    4c6d563 View commit details
    Browse the repository at this point in the history

  5. Fix the value type at runtime (use templates instead of runtime dtype)

    @achirkin
    achirkin committed May 20, 2022
    Configuration menu
    Copy the full SHA
    3ae52ea View commit details
    Browse the repository at this point in the history

  6. ceildiv

    @achirkin
    achirkin committed May 20, 2022
    Configuration menu
    Copy the full SHA
    6fecd7f View commit details
    Browse the repository at this point in the history

  7. Use rmm's memory pool in place of explicitly allocated buffers

    @achirkin
    achirkin committed May 20, 2022
    Configuration menu
    Copy the full SHA
    174854f View commit details
    Browse the repository at this point in the history

  8. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 20, 2022
    Configuration menu
    Copy the full SHA
    b45b14c View commit details
    Browse the repository at this point in the history

Commits on May 24, 2022

  1. Use raft logging

    @achirkin
    achirkin committed May 24, 2022
    Configuration menu
    Copy the full SHA
    ca1aaad View commit details
    Browse the repository at this point in the history

  2. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 24, 2022
    Configuration menu
    Copy the full SHA
    4228a02 View commit details
    Browse the repository at this point in the history

  3. Updated logging and nvtx markers

    @achirkin
    achirkin committed May 24, 2022
    Configuration menu
    Copy the full SHA
    70d84ec View commit details
    Browse the repository at this point in the history

  4. clang-format

    @achirkin
    achirkin committed May 24, 2022
    Configuration menu
    Copy the full SHA
    f9c12f8 View commit details
    Browse the repository at this point in the history

  5. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 24, 2022
    Configuration menu
    Copy the full SHA
    17968e4 View commit details
    Browse the repository at this point in the history

  6. Use the recommended logger header

    @achirkin
    achirkin committed May 24, 2022
    Configuration menu
    Copy the full SHA
    957ac94 View commit details
    Browse the repository at this point in the history

Commits on May 25, 2022

  1. Use warpsort for smaller k

    @achirkin
    achirkin committed May 25, 2022
    Configuration menu
    Copy the full SHA
    ccfbccc View commit details
    Browse the repository at this point in the history

  2. Using raft helpers

    @achirkin
    achirkin committed May 25, 2022
    Configuration menu
    Copy the full SHA
    7819397 View commit details
    Browse the repository at this point in the history

  3. Determine the template parameters Capacity and Veclen recursively

    @achirkin
    achirkin committed May 25, 2022
    Configuration menu
    Copy the full SHA
    510c467 View commit details
    Browse the repository at this point in the history

Commits on May 26, 2022

  1. wip: refactoring and reducing duplicate calls

    @achirkin
    achirkin committed May 26, 2022
    Configuration menu
    Copy the full SHA
    c5087be View commit details
    Browse the repository at this point in the history

Commits on May 27, 2022

  1. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 27, 2022
    Configuration menu
    Copy the full SHA
    f850a4a View commit details
    Browse the repository at this point in the history

  2. Refactor and document ann_ivf_flat_kernel

    @achirkin
    achirkin committed May 27, 2022
    Configuration menu
    Copy the full SHA
    c5f1c89 View commit details
    Browse the repository at this point in the history

  3. Documenting and refactoring the kernel

    @achirkin
    achirkin committed May 27, 2022
    Configuration menu
    Copy the full SHA
    7b2b9ff View commit details
    Browse the repository at this point in the history

Commits on May 30, 2022

  1. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed May 30, 2022
    Configuration menu
    Copy the full SHA
    913edfb View commit details
    Browse the repository at this point in the history

Commits on May 31, 2022

  1. Add a case of high dimensionality

    @achirkin
    achirkin committed May 31, 2022
    Configuration menu
    Copy the full SHA
    b1208ed View commit details
    Browse the repository at this point in the history

  2. Add more sync into the test to detect device errors

    @achirkin
    achirkin committed May 31, 2022
    Configuration menu
    Copy the full SHA
    a30ade5 View commit details
    Browse the repository at this point in the history

  3. Add more sync into the test to detect device errors

    @achirkin
    achirkin committed May 31, 2022
    Configuration menu
    Copy the full SHA
    84db732 View commit details
    Browse the repository at this point in the history

  4. Allow large batch sizes and document more functions

    @achirkin
    achirkin committed May 31, 2022
    Configuration menu
    Copy the full SHA
    346afb2 View commit details
    Browse the repository at this point in the history

  5. Add a lower bound on expected recall

    @achirkin
    achirkin committed May 31, 2022
    Configuration menu
    Copy the full SHA
    fc201b5 View commit details
    Browse the repository at this point in the history

  6. Compure required memory dynamically

    @achirkin
    achirkin committed May 31, 2022
    Configuration menu
    Copy the full SHA
    4021ea2 View commit details
    Browse the repository at this point in the history

  7. readability quickfix

    @achirkin
    achirkin committed May 31, 2022
    Configuration menu
    Copy the full SHA
    ea8b1c4 View commit details
    Browse the repository at this point in the history

  8. Correct the smem size for the warpsort and add launch bounds

    @achirkin
    achirkin committed May 31, 2022
    Configuration menu
    Copy the full SHA
    d8a034a View commit details
    Browse the repository at this point in the history

Commits on Jun 1, 2022

  1. Add couple checks against floating point exceptions

    @achirkin
    achirkin committed Jun 1, 2022
    Configuration menu
    Copy the full SHA
    d97d248 View commit details
    Browse the repository at this point in the history

Commits on Jun 2, 2022

  1. Don't run kmeans on empty dataset

    @achirkin
    achirkin committed Jun 2, 2022
    Configuration menu
    Copy the full SHA
    2e64037 View commit details
    Browse the repository at this point in the history

  2. Order all ops by a cuda stream

    @achirkin
    achirkin committed Jun 2, 2022
    Configuration menu
    Copy the full SHA
    9ed50ac View commit details
    Browse the repository at this point in the history

  3. Update comments

    @achirkin
    achirkin committed Jun 2, 2022
    Configuration menu
    Copy the full SHA
    1f9352c View commit details
    Browse the repository at this point in the history

  4. Suggest replacing _cuann_sqsum

    @achirkin
    achirkin committed Jun 2, 2022
    Configuration menu
    Copy the full SHA
    c048af2 View commit details
    Browse the repository at this point in the history

  5. wip: refactoting utils

    @achirkin
    achirkin committed Jun 2, 2022
    Configuration menu
    Copy the full SHA
    96f39a8 View commit details
    Browse the repository at this point in the history

  6. minor comments

    @achirkin
    achirkin committed Jun 2, 2022
    Configuration menu
    Copy the full SHA
    888daeb View commit details
    Browse the repository at this point in the history

Commits on Jun 3, 2022

  1. ann_utils refactoring, docs, and clang-tidy

    @achirkin
    achirkin committed Jun 3, 2022
    Configuration menu
    Copy the full SHA
    e6ff267 View commit details
    Browse the repository at this point in the history

Commits on Jun 7, 2022

  1. Merge branch 'branch-22.06' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jun 7, 2022
    Configuration menu
    Copy the full SHA
    426f713 View commit details
    Browse the repository at this point in the history

  2. Refactor tests and reduce their memory footprint

    @achirkin
    achirkin committed Jun 7, 2022
    Configuration menu
    Copy the full SHA
    bacb402 View commit details
    Browse the repository at this point in the history

  3. Refactored and documents ann_kmeans_balanced

    @achirkin
    achirkin committed Jun 7, 2022
    Configuration menu
    Copy the full SHA
    4042b28 View commit details
    Browse the repository at this point in the history

  4. Use memory_resource for temp data in kmeans

    @achirkin
    achirkin committed Jun 7, 2022
    Configuration menu
    Copy the full SHA
    bb5726b View commit details
    Browse the repository at this point in the history

Commits on Jun 8, 2022

  1. Address clang-tidy and other refactoring suggestions

    @achirkin
    achirkin committed Jun 8, 2022
    Configuration menu
    Copy the full SHA
    810c26b View commit details
    Browse the repository at this point in the history

  2. Move part of the index building onto gpu

    @achirkin
    achirkin committed Jun 8, 2022
    Configuration menu
    Copy the full SHA
    042c410 View commit details
    Browse the repository at this point in the history

Commits on Jun 15, 2022

  1. Document the index building kernel

    @achirkin
    achirkin committed Jun 15, 2022
    Configuration menu
    Copy the full SHA
    7ace0fb View commit details
    Browse the repository at this point in the history

  2. Merge branch 'branch-22.08' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jun 15, 2022
    Configuration menu
    Copy the full SHA
    e9c0d49 View commit details
    Browse the repository at this point in the history

  3. Added a dims padding todo

    @achirkin
    achirkin committed Jun 15, 2022
    Configuration menu
    Copy the full SHA
    3515715 View commit details
    Browse the repository at this point in the history

  4. Move kmeans-related allocations and routines to ann_kmeans_balanced.cuh

    @achirkin
    achirkin committed Jun 15, 2022
    Configuration menu
    Copy the full SHA
    6bd6560 View commit details
    Browse the repository at this point in the history

  5. Add documentation to the build_optimized_kmeans

    @achirkin
    achirkin committed Jun 15, 2022
    Configuration menu
    Copy the full SHA
    2811814 View commit details
    Browse the repository at this point in the history

Commits on Jun 16, 2022

  1. Using mdarrays and structured index

    @achirkin
    achirkin committed Jun 16, 2022
    Configuration menu
    Copy the full SHA
    fc3e46e View commit details
    Browse the repository at this point in the history

Commits on Jun 17, 2022

  1. Fixed a memory leak and introduced a few assertions to check pointer … 

    …alignment
    @achirkin
    achirkin committed Jun 17, 2022
    Configuration menu
    Copy the full SHA
    fb8c4b1 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'branch-22.08' into fea-knn-ivf-flat

    @cjnolet
    cjnolet committed Jun 17, 2022
    Configuration menu
    Copy the full SHA
    f3b2cb2 View commit details
    Browse the repository at this point in the history

  3. Refactoring build_optimized_kmeans

    @achirkin
    achirkin committed Jun 17, 2022
    Configuration menu
    Copy the full SHA
    092d428 View commit details
    Browse the repository at this point in the history

  4. A few smaller refactorings for kmeans

    @achirkin
    achirkin committed Jun 17, 2022
    Configuration menu
    Copy the full SHA
    fbcb16b View commit details
    Browse the repository at this point in the history

Commits on Jun 20, 2022

  1. Add docs to public methods of the handle

    @achirkin
    achirkin committed Jun 20, 2022
    Configuration menu
    Copy the full SHA
    29ca199 View commit details
    Browse the repository at this point in the history

Commits on Jun 21, 2022

  1. Made the metric be a part of the index struct and set the greater_ = … 

    …true for similarity metrics
    @achirkin
    achirkin committed Jun 21, 2022
    Configuration menu
    Copy the full SHA
    38b3cec View commit details
    Browse the repository at this point in the history

  2. Do not persist grid_dim_x between searches

    @achirkin
    achirkin committed Jun 21, 2022
    Configuration menu
    Copy the full SHA
    d19bb5f View commit details
    Browse the repository at this point in the history

  3. Refactor names according to clang-tidy

    @achirkin
    achirkin committed Jun 21, 2022
    Configuration menu
    Copy the full SHA
    9094707 View commit details
    Browse the repository at this point in the history

  4. Refactor the usage of stream and params

    @achirkin
    achirkin committed Jun 21, 2022
    Configuration menu
    Copy the full SHA
    325e201 View commit details
    Browse the repository at this point in the history

  5. Refactor api to have symmetric index/search params

    @achirkin
    achirkin committed Jun 21, 2022
    Configuration menu
    Copy the full SHA
    2a3eb33 View commit details
    Browse the repository at this point in the history

Commits on Jun 22, 2022

  1. refactor away ivf_flat_index

    @achirkin
    achirkin committed Jun 22, 2022
    Configuration menu
    Copy the full SHA
    867beca View commit details
    Browse the repository at this point in the history

  2. Add the memory resource argument to warp_sort_topk

    @achirkin
    achirkin committed Jun 22, 2022
    Configuration menu
    Copy the full SHA
    059a6c0 View commit details
    Browse the repository at this point in the history

  3. update docs

    @achirkin
    achirkin committed Jun 22, 2022
    Configuration menu
    Copy the full SHA
    df17b5b View commit details
    Browse the repository at this point in the history

Commits on Jun 23, 2022

  1. Allow empty mesoclusters

    @achirkin
    achirkin committed Jun 23, 2022
    Configuration menu
    Copy the full SHA
    fe9ced1 View commit details
    Browse the repository at this point in the history

  2. Add low-dimensional and non-veclen-aligned-dimensional test cases

    @achirkin
    achirkin committed Jun 23, 2022
    Configuration menu
    Copy the full SHA
    91fdcbb View commit details
    Browse the repository at this point in the history

  3. Refactor and document loadAndComputeDist

    @achirkin
    achirkin committed Jun 23, 2022
    Configuration menu
    Copy the full SHA
    be14c63 View commit details
    Browse the repository at this point in the history

  4. Minor renamings

    @achirkin
    achirkin committed Jun 23, 2022
    Configuration menu
    Copy the full SHA
    eeb4601 View commit details
    Browse the repository at this point in the history

  5. Add 8bit int types to knn benchmarks

    @achirkin
    achirkin committed Jun 23, 2022
    Configuration menu
    Copy the full SHA
    025e5a5 View commit details
    Browse the repository at this point in the history

Commits on Jun 24, 2022

  1. Fix incorrect data mapping for int8 types

    @achirkin
    achirkin committed Jun 24, 2022
    Configuration menu
    Copy the full SHA
    3821366 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'branch-22.08' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jun 24, 2022
    Configuration menu
    Copy the full SHA
    d596842 View commit details
    Browse the repository at this point in the history

Commits on Jun 27, 2022

  1. Introduce kIndexGroupSize constant

    @achirkin
    achirkin committed Jun 27, 2022
    Configuration menu
    Copy the full SHA
    a29baa7 View commit details
    Browse the repository at this point in the history

  2. Cleanup ann_quantized

    @achirkin
    achirkin committed Jun 27, 2022
    Configuration menu
    Copy the full SHA
    546bef8 View commit details
    Browse the repository at this point in the history

  3. Add several type aliases and helpers for creating mdarrays

    @achirkin
    achirkin committed Jun 27, 2022
    Configuration menu
    Copy the full SHA
    32d0d2e View commit details
    Browse the repository at this point in the history

Commits on Jun 28, 2022

  1. Remove unnecessary inlines and fix docs

    @achirkin
    achirkin committed Jun 28, 2022
    Configuration menu
    Copy the full SHA
    5f427c0 View commit details
    Browse the repository at this point in the history

  2. More refactoring and a few forceinlines

    @achirkin
    achirkin committed Jun 28, 2022
    Configuration menu
    Copy the full SHA
    c581fe2 View commit details
    Browse the repository at this point in the history

Commits on Jun 29, 2022

  1. Add a helper for creating pool_memory_resource when it makes sense

    @achirkin
    achirkin committed Jun 29, 2022
    Configuration menu
    Copy the full SHA
    805e78c View commit details
    Browse the repository at this point in the history

  2. Force move the mdarrays when creating index to avoid copying them

    @achirkin
    achirkin committed Jun 29, 2022
    Configuration menu
    Copy the full SHA
    a4973e6 View commit details
    Browse the repository at this point in the history

  3. Minor refactorings

    @achirkin
    achirkin committed Jun 29, 2022
    Configuration menu
    Copy the full SHA
    68c267e View commit details
    Browse the repository at this point in the history

  4. Add nvtx annotations to the outermost ANN calls for better performanc… 

    …e analysis
    @achirkin
    achirkin committed Jun 29, 2022
    Configuration menu
    Copy the full SHA
    f2b8ed8 View commit details
    Browse the repository at this point in the history

  5. Add a few more test cases and annotations for them

    @achirkin
    achirkin committed Jun 29, 2022
    Configuration menu
    Copy the full SHA
    f91c7f7 View commit details
    Browse the repository at this point in the history

  6. Fix a typo

    @achirkin
    achirkin committed Jun 29, 2022
    Configuration menu
    Copy the full SHA
    84b1c5b View commit details
    Browse the repository at this point in the history

Commits on Jun 30, 2022

  1. Move ensure_integral_extents to the detail folder

    @achirkin
    achirkin committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    afc1f6a View commit details
    Browse the repository at this point in the history

  2. Lift the requirement to have query pointers aligned with Veclen

    @achirkin
    achirkin committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    3a10f86 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'branch-22.08' into enh-mdarray-helpers

    @achirkin
    achirkin committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    9f5c64c View commit details
    Browse the repository at this point in the history

  4. Use move semantics for the index everywhere, but try to keep it const… 

    … where possible
    @achirkin
    achirkin committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    1afd667 View commit details
    Browse the repository at this point in the history

  5. Update documentation

    @achirkin
    achirkin committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    73ce9e1 View commit details
    Browse the repository at this point in the history

  6. Remove the debug path USE_FAISS

    @achirkin
    achirkin committed Jun 30, 2022
    Configuration menu
    Copy the full SHA
    2a45645 View commit details
    Browse the repository at this point in the history

Commits on Jul 1, 2022

  1. Add a type trait for checking if the conversion between two numeric t… 

    …ypes is narrowing
    @achirkin
    achirkin committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    75a48b4 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'branch-22.08' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    ed25cae View commit details
    Browse the repository at this point in the history

  3. Support 32bit and unsigned indices in bruteforce KNN

    @achirkin
    achirkin committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    388200c View commit details
    Browse the repository at this point in the history

  4. Merge branch 'enh-mdarray-helpers' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    f08df83 View commit details
    Browse the repository at this point in the history

  5. Merge branch 'enh-knn-bruteforce-uint32' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    9200886 View commit details
    Browse the repository at this point in the history

  6. Make index type a template parameter

    @achirkin
    achirkin committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    14bfe02 View commit details
    Browse the repository at this point in the history

  7. Revert the api changes as much as possible and deprecate the old api

    @achirkin
    achirkin committed Jul 1, 2022
    Configuration menu
    Copy the full SHA
    1283cbe View commit details
    Browse the repository at this point in the history

Commits on Jul 4, 2022

  1. Remove the stream argument from the public API

    @achirkin
    achirkin committed Jul 4, 2022
    Configuration menu
    Copy the full SHA
    e73b259 View commit details
    Browse the repository at this point in the history

Commits on Jul 5, 2022

  1. Merge branch 'branch-22.08' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jul 5, 2022
    Configuration menu
    Copy the full SHA
    8e7ffb8 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'branch-22.08' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jul 5, 2022
    Configuration menu
    Copy the full SHA
    5f5dc0d View commit details
    Browse the repository at this point in the history

Commits on Jul 6, 2022

  1. Simplify kmeans::predict a little bit

    @achirkin
    achirkin committed Jul 6, 2022
    Configuration menu
    Copy the full SHA
    03ebbe0 View commit details
    Browse the repository at this point in the history

Commits on Jul 7, 2022

  1. Factor out predict from the other ops in kmeans for use outside of th… 

    …e module
    @achirkin
    achirkin committed Jul 7, 2022
    Configuration menu
    Copy the full SHA
    cde7f97 View commit details
    Browse the repository at this point in the history

Commits on Jul 20, 2022

  1. Add new function extend(index, new_vecs, new_inds) to ivf_flat

    @achirkin
    achirkin committed Jul 20, 2022
    Configuration menu
    Copy the full SHA
    305bbcd View commit details
    Browse the repository at this point in the history

Commits on Jul 21, 2022

  1. Merge branch 'branch-22.08' into fea-knn-ivf-flat

    @achirkin
    achirkin committed Jul 21, 2022
    Configuration menu
    Copy the full SHA
    76c383f View commit details
    Browse the repository at this point in the history

  2. Improve the docs

    @achirkin
    achirkin committed Jul 21, 2022
    Configuration menu
    Copy the full SHA
    7f640a9 View commit details
    Browse the repository at this point in the history

  3. Fix using non-existing log function

    @achirkin
    achirkin committed Jul 21, 2022
    Configuration menu
    Copy the full SHA
    2e9eda5 View commit details
    Browse the repository at this point in the history

  4. Hide all data components from ifv_flat::index and expose immutable views

    @achirkin
    achirkin committed Jul 21, 2022
    Configuration menu
    Copy the full SHA
    dc62a0f View commit details
    Browse the repository at this point in the history

Commits on Jul 22, 2022

  1. Replace thurst::exclusive_scan with thrust::inclusive_scan to avoid a… 

    …n out-of-bounds read of the last entry
    @achirkin
    achirkin committed Jul 22, 2022
    Configuration menu
    Copy the full SHA
    fb841c3 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'fea-knn-ivf-flat' into enh-knn-ivf-flat-hide-impl

    @achirkin
    achirkin committed Jul 22, 2022
    Configuration menu
    Copy the full SHA
    04bb5dc View commit details
    Browse the repository at this point in the history

  3. ann_common.h: remove deps on cuda code, so that the file can be inclu… 

    …ded in .cpp files as before (in cuml)
    @achirkin
    achirkin committed Jul 22, 2022
    Configuration menu
    Copy the full SHA
    c95ea85 View commit details
    Browse the repository at this point in the history

  4. ann_common.h: remove deps on cuda code, so that the file can be inclu… 

    …ded in .cpp files as before (in cuml)
    @achirkin
    achirkin committed Jul 22, 2022
    Configuration menu
    Copy the full SHA
    0c72ee8 View commit details
    Browse the repository at this point in the history

  5. Make helper overloads inline for linking in cuml

    @achirkin
    achirkin committed Jul 22, 2022
    Configuration menu
    Copy the full SHA
    0196695 View commit details
    Browse the repository at this point in the history

  6. Split processing.hpp into *.cuh and *.hpp to avoid incomplete types

    @achirkin
    achirkin committed Jul 22, 2022
    Configuration menu
    Copy the full SHA
    eb15639 View commit details
    Browse the repository at this point in the history

Commits on Jul 25, 2022

  1. WIP: investigating segmentation fault in cuml test

    @achirkin
    achirkin committed Jul 25, 2022
    Configuration menu
    Copy the full SHA
    e4b2b39 View commit details
    Browse the repository at this point in the history

Commits on Jul 26, 2022

  1. Revert the wip-changes from the last commit

    @achirkin
    achirkin committed Jul 26, 2022
    Configuration menu
    Copy the full SHA
    6bc0fcb View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'origin/fea-knn-ivf-flat' into enh-knn-i… 

    …vf-flat-hide-impl
    @achirkin
    achirkin committed Jul 26, 2022
    Configuration menu
    Copy the full SHA
    f599aaf View commit details
    Browse the repository at this point in the history

Commits on Jul 28, 2022

  1. Merge branch 'branch-22.08' into enh-knn-ivf-flat-hide-impl

    @achirkin
    achirkin committed Jul 28, 2022
    Configuration menu
    Copy the full SHA
    a191410 View commit details
    Browse the repository at this point in the history

  2. Enhance documentation

    @achirkin
    achirkin committed Jul 28, 2022
    Configuration menu
    Copy the full SHA
    317ddf3 View commit details
    Browse the repository at this point in the history

  3. Fix couple typos in docs

    @achirkin
    achirkin committed Jul 28, 2022
    Configuration menu
    Copy the full SHA
    114fb63 View commit details
    Browse the repository at this point in the history

  4. Change the data indexing to size_t to make sure the total size (size*… 

    …dim) fits
    @achirkin
    achirkin committed Jul 28, 2022
    Configuration menu
    Copy the full SHA
    1d283ae View commit details
    Browse the repository at this point in the history

Commits on Aug 2, 2022

  1. Merge branch 'branch-22.08' into enh-knn-ivf-flat-hide-impl

    @achirkin
    achirkin committed Aug 2, 2022
    Configuration menu
    Copy the full SHA
    a9bd2d6 View commit details
    Browse the repository at this point in the history

  2. Make ivf_flat::index look a little bit more like knn::sparse api

    @achirkin
    achirkin committed Aug 2, 2022
    Configuration menu
    Copy the full SHA
    f9d55a7 View commit details
    Browse the repository at this point in the history

  3. Test both overloads of

    @achirkin
    achirkin committed Aug 2, 2022
    Configuration menu
    Copy the full SHA
    fef6dac View commit details
    Browse the repository at this point in the history