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

Migrate IVF-Flat from RAFT #94

Merged
merged 40 commits into from
May 16, 2024
Merged

Migrate IVF-Flat from RAFT #94

Show file tree
Hide file tree
Changes from 35 commits
Commits
Show all changes
40 commits
Select commit Hold shift + click to select a range
3c40a06
Init IVFPQ migration
lowener Apr 11, 2024
c5a3d3a
Migrate ivf-pq build steps to cuvs
lowener Apr 15, 2024
3568f12
Migrate ivf-pq extend to cuvs
lowener Apr 15, 2024
69bf330
Migrate ivf-pq serialize to cuvs
lowener Apr 16, 2024
7a5e8a0
Migrate compute similarity
lowener Apr 17, 2024
4b86f94
Migrate search
lowener Apr 18, 2024
8c8aaed
Add Test for IVFPQ
lowener Apr 19, 2024
e444f4b
Temporarily remove IVFPQ Search with filter
lowener Apr 19, 2024
5228a68
Reset exceptions
lowener Apr 19, 2024
3f924f2
Merge branch 'branch-24.06' into 24.06-ivfpq-migration
lowener Apr 19, 2024
e8907ee
Add bitset to cuvs, add search_with_filter to ivf-pq
lowener Apr 23, 2024
64809e8
Remove `uint32_t` ivfpq compute_similarity instantiation
lowener Apr 23, 2024
e88f82c
Add bitset to compute similarity instantiation
lowener Apr 23, 2024
36bd63b
Add NVTX cuvs domain
lowener Apr 24, 2024
54d2159
migration
divyegala Apr 24, 2024
43cee07
format
divyegala Apr 24, 2024
b43b0f3
more formatting
divyegala Apr 25, 2024
9c4781d
Expose only public API on includes
lowener Apr 30, 2024
6e7591a
Merge branch 'branch-24.06' into 24.06-ivfpq-migration
cjnolet May 1, 2024
b6ff0f9
.cu instead of .cpp
divyegala May 1, 2024
c9ed948
cmake changes
divyegala May 1, 2024
0a71eb8
building
divyegala May 1, 2024
5a719ba
Move headers
lowener May 3, 2024
091fe02
Separate header and impl for filters
lowener May 6, 2024
f4481ea
intermediate stage
divyegala May 8, 2024
69ca137
merging mickael
divyegala May 8, 2024
5d27699
Merge remote-tracking branch 'upstream/branch-24.06' into 24.06-ivffl…
divyegala May 8, 2024
ee58fc2
passing tests
divyegala May 8, 2024
69a10a6
search with filter, more serialization
divyegala May 8, 2024
96d86f7
pre-commit
divyegala May 8, 2024
6d9813c
style check
divyegala May 8, 2024
edb2f79
address review
divyegala May 14, 2024
8f48df5
merge upstream
divyegala May 14, 2024
311f4f4
fix include
divyegala May 14, 2024
817017c
address review
divyegala May 14, 2024
d8c10d4
address review
divyegala May 14, 2024
f961210
combine ivf_flat_helper functions into ivf_flat
divyegala May 14, 2024
831df78
fix ivf flat search
divyegala May 14, 2024
e71b2af
codepacker inside helper namespace
divyegala May 15, 2024
8f20107
Merge remote-tracking branch 'upstream/branch-24.06' into 24.06-ivffl…
divyegala May 15, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 10 additions & 12 deletions cpp/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -379,18 +379,16 @@ add_library(
src/neighbors/detail/refine_host_int8_t_float.cpp
src/neighbors/detail/refine_host_uint8_t_float.cpp
src/neighbors/ivf_flat_index.cpp
src/neighbors/ivf_flat/ivf_flat_build_float_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_build_int8_t_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_build_uint8_t_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_extend_float_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_extend_int8_t_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_extend_uint8_t_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_search_float_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_search_int8_t_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_search_uint8_t_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_serialize_float_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_serialize_int8_t_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_serialize_uint8_t_int64_t.cpp
src/neighbors/ivf_flat/ivf_flat_build_extend_float_int64_t.cu
src/neighbors/ivf_flat/ivf_flat_build_extend_int8_t_int64_t.cu
src/neighbors/ivf_flat/ivf_flat_build_extend_uint8_t_int64_t.cu
src/neighbors/ivf_flat/ivf_flat_helpers.cu
src/neighbors/ivf_flat/ivf_flat_search_float_int64_t.cu
src/neighbors/ivf_flat/ivf_flat_search_int8_t_int64_t.cu
src/neighbors/ivf_flat/ivf_flat_search_uint8_t_int64_t.cu
src/neighbors/ivf_flat/ivf_flat_serialize_float_int64_t.cu
src/neighbors/ivf_flat/ivf_flat_serialize_int8_t_int64_t.cu
src/neighbors/ivf_flat/ivf_flat_serialize_uint8_t_int64_t.cu
src/neighbors/ivf_pq_index.cpp
src/neighbors/ivf_pq/ivf_pq_build_common.cu
src/neighbors/ivf_pq/ivf_pq_serialize.cu
Expand Down
Loading
Loading