Skip to content

feat: add hnsw-rabitq support #69

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

Merged
egolearner merged 62 commits into from
Mar 19, 2026
Merged

feat: add hnsw-rabitq support #69

Show file tree
Hide file tree
Changes from 26 commits
Commits
Show all changes
62 commits
Select commit Hold shift + click to select a range
0e5727a
feat: add hnsw-rabitq
egolearner Feb 4, 2026
f3a4214
undefine transform
egolearner Feb 5, 2026
b4794f6
support config rotator type
egolearner Feb 5, 2026
676c457
support sample_count
egolearner Feb 5, 2026
fd330c0
fix ut
egolearner Feb 5, 2026
5977529
refactor: update interface
egolearner Feb 6, 2026
9713221
refactor: update interface
egolearner Feb 6, 2026
78127fd
update rabitq index params
egolearner Feb 6, 2026
af3f172
fix interface update
egolearner Feb 7, 2026
2e153b4
fix searcher test
egolearner Feb 7, 2026
0427555
fix streamer test
egolearner Feb 7, 2026
4c6f28a
streamer support bf and add more ut
egolearner Feb 8, 2026
aac0ea3
rm env check
egolearner Feb 8, 2026
d2809c3
add collection ut
egolearner Feb 9, 2026
2f0ccbc
add schema check
egolearner Feb 9, 2026
420d791
add rabitq query param binding
egolearner Feb 9, 2026
32e2ed0
add integration test
egolearner Feb 10, 2026
733fa46
fix local_builder
egolearner Feb 10, 2026
f381039
cleanup dist calculator
egolearner Feb 10, 2026
090913d
add RaBitQ-Library submodule
egolearner Feb 10, 2026
afbf50c
cleanup rabitq converter/reformer
egolearner Feb 10, 2026
5431adb
Merge branch 'main' into feat/rabitq
egolearner Feb 10, 2026
3b06f74
add files
egolearner Feb 10, 2026
822af9a
disable build on mac
egolearner Feb 11, 2026
36aa0a7
disable Feature_Optimize_HNSW_RABITQ
egolearner Feb 11, 2026
86f84bd
disable python/tests/test_collection_hnsw_rabitq.py:13
egolearner Feb 11, 2026
df922f2
check avx2/avx512
egolearner Feb 18, 2026
91f706d
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Feb 22, 2026
80adf29
fix refine
egolearner Feb 25, 2026
f57c56e
fix mac ci
egolearner Feb 25, 2026
9dc8b64
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Feb 25, 2026
56bbcc5
add dimension check
egolearner Feb 26, 2026
04817ec
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Feb 26, 2026
f8713c5
fix mac ci
egolearner Feb 26, 2026
986e557
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Feb 26, 2026
f05a120
fix ci
egolearner Feb 26, 2026
42a30d6
add missing lib
egolearner Feb 26, 2026
e5fb71b
fix centroids selection for Cosine/InnerProduct
egolearner Feb 28, 2026
8549247
rename hnsw-rabitq to hnsw_rabitq
egolearner Feb 28, 2026
d93d32b
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Feb 28, 2026
6570070
address comments
egolearner Mar 11, 2026
3797862
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Mar 12, 2026
1729682
fix compile
egolearner Mar 12, 2026
94837be
fix rabitqlib name
egolearner Mar 12, 2026
27307ef
fix mac compile
egolearner Mar 12, 2026
6d99bcd
check avx2/avx512 support
egolearner Mar 12, 2026
a98b0f3
runtime check again compile-time
egolearner Mar 12, 2026
7d7ce96
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Mar 12, 2026
4227cfc
check AUTO_DETECT_ARCH
egolearner Mar 12, 2026
2879c70
rm debug log
egolearner Mar 12, 2026
2a6252c
fix
egolearner Mar 13, 2026
365d878
rabitq use avx2 by default
egolearner Mar 13, 2026
debb4af
add missing file
egolearner Mar 13, 2026
65f6a62
fix search_bf/group_by dist
egolearner Mar 16, 2026
ad5a807
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Mar 16, 2026
de48e3a
address comments
egolearner Mar 16, 2026
cdbc6cc
fix typo
egolearner Mar 17, 2026
3fb3261
address comments
egolearner Mar 17, 2026
3e1a4ad
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Mar 18, 2026
84b046e
rabitq support disable id_map
egolearner Mar 18, 2026
b11b34a
address comments
egolearner Mar 18, 2026
5e747de
Merge remote-tracking branch 'origin/main' into feat/rabitq
egolearner Mar 19, 2026
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
3 changes: 3 additions & 0 deletions .gitmodules
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,3 +39,6 @@
path = thirdparty/magic_enum/magic_enum-0.9.7
url = https://github.com/Neargye/magic_enum.git
ignore = all
[submodule "thirdparty/thirdparty/RaBitQ-Library/RaBitQ-Library-0.1"]
Comment thread
greptile-apps[bot] marked this conversation as resolved.
Outdated
path = thirdparty/RaBitQ-Library/RaBitQ-Library-0.1
url = https://github.com/VectorDB-NTU/RaBitQ-Library.git
10 changes: 10 additions & 0 deletions CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,16 @@ message(STATUS "BUILD_PYTHON_BINDINGS:${BUILD_PYTHON_BINDINGS}")
option(BUILD_TOOLS "Build tools" ON)
message(STATUS "BUILD_TOOLS:${BUILD_TOOLS}")

if(CMAKE_SYSTEM_NAME STREQUAL "Linux" AND CMAKE_SYSTEM_PROCESSOR MATCHES "x86_64|amd64|AMD64")
set(RABITQ_SUPPORTED ON)
add_definitions(-DRABITQ_SUPPORTED=1)
message(STATUS "RaBitQ support enabled for Linux x86_64")
else()
set(RABITQ_SUPPORTED OFF)
add_definitions(-DRABITQ_SUPPORTED=0)
message(STATUS "RaBitQ support disabled - only supported on Linux x86_64")
endif()

cc_directory(thirdparty)
cc_directories(src)
cc_directories(tests)
Expand Down
Loading
Loading