support cagra

Signed-off-by: yusheng.ma <[email protected]>
zilliztech · Nov 22, 2023 · 36297e8 · 36297e8
1 parent adbf8e0
commit 36297e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/index/gpu_raft/gpu_raft_ivf_pq_config.h b/src/index/gpu_raft/gpu_raft_ivf_pq_config.h
@@ -48,9 +48,9 @@ struct GpuRaftIvfPqConfig : public IvfPqConfig {
             .for_search();
         KNOWHERE_CONFIG_DECLARE_FIELD(m).set_default(8).description("m").set_range(0, 65536).for_train();
         KNOWHERE_CONFIG_DECLARE_FIELD(nbits)
-            .set_default(0)
+            .set_default(8)
             .description("nbits")
-            .set_range(0, 8)  // Declared in base but limited here
+            .set_range(4, 8)  // Declared in base but limited here
             .for_train();
         KNOWHERE_CONFIG_DECLARE_FIELD(kmeans_n_iters)
             .description("iterations to search for kmeans centers")