diff --git a/cpp/unittest/db/db_tests.cpp b/cpp/unittest/db/db_tests.cpp index 67d61a05c29ef96b35db9e82c876561f657a7d92..50724eddc3d513ab084b217f3f4e0cc0a5a9e41a 100644 --- a/cpp/unittest/db/db_tests.cpp +++ b/cpp/unittest/db/db_tests.cpp @@ -378,11 +378,11 @@ TEST_F(DBTest, INDEX_TEST) { ASSERT_TRUE(stat.ok()); engine::TableIndex index_out; - stat = db_->DescribeIndex(table_info.table_id_, index); + stat = db_->DescribeIndex(table_info.table_id_, index_out); ASSERT_TRUE(stat.ok()); ASSERT_EQ(index.engine_type_, index_out.engine_type_); ASSERT_EQ(index.nlist_, index_out.nlist_); - ASSERT_EQ(index.metric_type_, index_out.metric_type_); + ASSERT_EQ(table_info.metric_type_, index_out.metric_type_); stat = db_->DropIndex(table_info.table_id_); ASSERT_TRUE(stat.ok()); diff --git a/cpp/unittest/db/utils.cpp b/cpp/unittest/db/utils.cpp index 5b8d82522d320f02aacefbe09ebe9f55d0cbde8d..f598fb1f8ac1d0dbfaf89437c007f832609a03e3 100644 --- a/cpp/unittest/db/utils.cpp +++ b/cpp/unittest/db/utils.cpp @@ -13,6 +13,7 @@ #include "db/Factories.h" #include "db/Options.h" #include "server/ServerConfig.h" +#include "knowhere/index/vector_index/gpu_ivf.h" INITIALIZE_EASYLOGGINGPP @@ -59,6 +60,8 @@ engine::Options BaseTest::GetOptions() { void DBTest::SetUp() { BaseTest::SetUp(); + zilliz::knowhere::FaissGpuResourceMgr::GetInstance().InitDevice(0, 1024*1024*200, 1024*1024*300, 2); + server::ConfigNode& config = server::ServerConfig::GetInstance().GetConfig(server::CONFIG_CACHE); config.AddSequenceItem(server::CONFIG_GPU_IDS, "0"); @@ -84,6 +87,8 @@ void DBTest::TearDown() { db_->DropAll(); delete db_; + zilliz::knowhere::FaissGpuResourceMgr::GetInstance().Free(); + engine::ResMgrInst::GetInstance()->Stop(); engine::SchedInst::GetInstance()->Stop();