From 36297e8c07fb9c5ae4f1e89993b4efa23d91ee2a Mon Sep 17 00:00:00 2001 From: "yusheng.ma" Date: Wed, 22 Nov 2023 19:10:59 +0800 Subject: [PATCH] support cagra Signed-off-by: yusheng.ma --- src/index/gpu_raft/gpu_raft_ivf_pq_config.h | 4 ++-- 1 file changed, 2 insertions(+), 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 index c689cf69f..5eda80d63 100644 --- 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")