From 985e63cbf27cd5d7dc5db781de6e78fa42436d2f Mon Sep 17 00:00:00 2001
From: groot <yihua.mo@zilliz.com>
Date: Thu, 23 Sep 2021 21:50:11 +0800
Subject: [PATCH] Reduce cpp unittest time cost (#8424)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
---
 internal/core/src/index/unittest/test_nsg.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internal/core/src/index/unittest/test_nsg.cpp b/internal/core/src/index/unittest/test_nsg.cpp
index 29283680c2..e3ba6cf5d1 100644
--- a/internal/core/src/index/unittest/test_nsg.cpp
+++ b/internal/core/src/index/unittest/test_nsg.cpp
@@ -43,7 +43,7 @@ class NSGInterfaceTest : public DataGen, public ::testing::Test {
         milvus::knowhere::FaissGpuResourceMgr::GetInstance().InitDevice(DEVICE_GPU0, MB * 200, MB * 600, 1);
 #endif
         int nsg_dim = 256;
-        Generate(nsg_dim, 20000, nq);
+        Generate(nsg_dim, nb, nq);
         index_ = std::make_shared<milvus::knowhere::NSG_NM>();
 
         train_conf = milvus::knowhere::Config{{milvus::knowhere::meta::DIM, 256},