diff --git a/core/src/index/unittest/test_ivf.cpp b/core/src/index/unittest/test_ivf.cpp index 3b3e557c4d77637468f086b631b985ef3b8a8380..4aaddb108b6e454e7d0376283e091b14f28be0ce 100644 --- a/core/src/index/unittest/test_ivf.cpp +++ b/core/src/index/unittest/test_ivf.cpp @@ -96,10 +96,10 @@ class ParamGenerator { auto tempconf = std::make_shared(); tempconf->d = DIM; tempconf->gpu_id = device_id; - tempconf->nlist = 100; - tempconf->nprobe = 16; + tempconf->nlist = 25; + tempconf->nprobe = 4; tempconf->k = K; - tempconf->m = 8; + tempconf->m = 4; tempconf->nbits = 8; tempconf->metric_type = knowhere::METRICTYPE::L2; return tempconf;