From 8dd2051a9a3676f65fb0281397062212e5c3fbec Mon Sep 17 00:00:00 2001 From: dragondriver Date: Fri, 2 Jul 2021 18:20:14 +0800 Subject: [PATCH] Fix 5045, decrease the nb and nlist in unittest (#6282) Signed-off-by: dragondriver --- internal/core/unittest/test_index_wrapper.cpp | 2 +- .../core/unittest/test_utils/indexbuilder_test_utils.h | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/internal/core/unittest/test_index_wrapper.cpp b/internal/core/unittest/test_index_wrapper.cpp index 28ff1b6a2a..3d787d9d8b 100644 --- a/internal/core/unittest/test_index_wrapper.cpp +++ b/internal/core/unittest/test_index_wrapper.cpp @@ -25,7 +25,7 @@ #include "indexbuilder/utils.h" #include "test_utils/indexbuilder_test_utils.h" -constexpr int64_t NB = 10000; +constexpr int64_t NB = 1000; namespace indexcgo = milvus::proto::indexcgo; using Param = std::pair; diff --git a/internal/core/unittest/test_utils/indexbuilder_test_utils.h b/internal/core/unittest/test_utils/indexbuilder_test_utils.h index 6dd18a76be..b0d1c79c81 100644 --- a/internal/core/unittest/test_utils/indexbuilder_test_utils.h +++ b/internal/core/unittest/test_utils/indexbuilder_test_utils.h @@ -48,7 +48,7 @@ generate_conf(const milvus::knowhere::IndexType& index_type, const milvus::knowh return milvus::knowhere::Config{ {milvus::knowhere::meta::DIM, DIM}, {milvus::knowhere::meta::TOPK, K}, - {milvus::knowhere::IndexParams::nlist, 100}, + {milvus::knowhere::IndexParams::nlist, 16}, {milvus::knowhere::IndexParams::nprobe, 4}, {milvus::knowhere::IndexParams::m, 4}, {milvus::knowhere::IndexParams::nbits, 8}, @@ -59,7 +59,7 @@ generate_conf(const milvus::knowhere::IndexType& index_type, const milvus::knowh return milvus::knowhere::Config{ {milvus::knowhere::meta::DIM, DIM}, {milvus::knowhere::meta::TOPK, K}, - {milvus::knowhere::IndexParams::nlist, 1024}, + {milvus::knowhere::IndexParams::nlist, 16}, {milvus::knowhere::IndexParams::nprobe, 4}, {milvus::knowhere::Metric::TYPE, metric_type}, {milvus::knowhere::INDEX_FILE_SLICE_SIZE_IN_MEGABYTE, 4}, @@ -71,7 +71,7 @@ generate_conf(const milvus::knowhere::IndexType& index_type, const milvus::knowh return milvus::knowhere::Config{ {milvus::knowhere::meta::DIM, DIM}, {milvus::knowhere::meta::TOPK, K}, - {milvus::knowhere::IndexParams::nlist, 100}, + {milvus::knowhere::IndexParams::nlist, 16}, {milvus::knowhere::IndexParams::nprobe, 4}, {milvus::knowhere::IndexParams::nbits, 8}, {milvus::knowhere::Metric::TYPE, metric_type}, @@ -84,7 +84,7 @@ generate_conf(const milvus::knowhere::IndexType& index_type, const milvus::knowh return milvus::knowhere::Config{ {milvus::knowhere::meta::DIM, DIM}, {milvus::knowhere::meta::TOPK, K}, - {milvus::knowhere::IndexParams::nlist, 100}, + {milvus::knowhere::IndexParams::nlist, 16}, {milvus::knowhere::IndexParams::nprobe, 4}, {milvus::knowhere::IndexParams::m, 4}, {milvus::knowhere::IndexParams::nbits, 8},