feat: add hnsw-rabitq support

.gitmodules

@@ -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"]
+	path = thirdparty/RaBitQ-Library/RaBitQ-Library-0.1
+	url = https://github.com/VectorDB-NTU/RaBitQ-Library.git

CMakeLists.txt

@@ -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)

examples/c++/db/main.cc

@@ -110,10 +110,10 @@ Doc create_doc(const uint64_t doc_id, const CollectionSchema &schema,
             std::vector<double>(field->dimension(), double(doc_id + 0.1)));
         break;
       case DataType::VECTOR_FP16:
-        new_doc.set<std::vector<float16_t>>(
-            field->name(), std::vector<float16_t>(
-                               field->dimension(),
-                               static_cast<float16_t>(float(doc_id + 0.1))));
+        new_doc.set<std::vector<zvec::float16_t>>(
+            field->name(), std::vector<zvec::float16_t>(
+                               field->dimension(), static_cast<zvec::float16_t>(
+                                                       float(doc_id + 0.1))));
         break;
       case DataType::VECTOR_INT8:
         new_doc.set<std::vector<int8_t>>(
@@ -127,16 +127,17 @@ Doc create_doc(const uint64_t doc_id, const CollectionSchema &schema,
         break;
       case DataType::SPARSE_VECTOR_FP16: {
         std::vector<uint32_t> indices;
-        std::vector<float16_t> values;
+        std::vector<zvec::float16_t> values;
         for (uint32_t i = 0; i < 100; i++) {
           indices.push_back(i);
-          values.push_back(float16_t(float(doc_id + 0.1)));
+          values.push_back(zvec::float16_t(float(doc_id + 0.1)));
         }
-        std::pair<std::vector<uint32_t>, std::vector<float16_t>>
+        std::pair<std::vector<uint32_t>, std::vector<zvec::float16_t>>
             sparse_float_vec;
         sparse_float_vec.first = indices;
         sparse_float_vec.second = values;
-        new_doc.set<std::pair<std::vector<uint32_t>, std::vector<float16_t>>>(
+        new_doc.set<
+            std::pair<std::vector<uint32_t>, std::vector<zvec::float16_t>>>(
             field->name(), sparse_float_vec);
         break;
       }