diff --git a/core/src/server/Config.h b/core/src/server/Config.h index 4e4923ee07a29b8d9c52a1b160a2cab489e30129..281a832d5720db999453cca8d88ef454221234ad 100644 --- a/core/src/server/Config.h +++ b/core/src/server/Config.h @@ -172,9 +172,10 @@ class Config { CheckEngineConfigOmpThreadNum(const std::string& value); #ifdef MILVUS_GPU_VERSION - /* gpu resource config */ Status CheckEngineConfigGpuSearchThreshold(const std::string& value); + + /* gpu resource config */ Status CheckGpuResourceConfigEnable(const std::string& value); Status @@ -307,9 +308,10 @@ class Config { SetEngineConfigOmpThreadNum(const std::string& value); #ifdef MILVUS_GPU_VERSION - /* gpu resource config */ Status SetEngineConfigGpuSearchThreshold(const std::string& value); + + /* gpu resource config */ Status SetGpuResourceConfigEnable(const std::string& value); Status diff --git a/core/unittest/server/test_config.cpp b/core/unittest/server/test_config.cpp index 3a24c02a45c12e3844aabbe5f2c9c1924b653fd1..791876ee8bb40ed46dd3413ab932ee72b7625286 100644 --- a/core/unittest/server/test_config.cpp +++ b/core/unittest/server/test_config.cpp @@ -404,10 +404,10 @@ TEST_F(ConfigTest, SERVER_CONFIG_INVALID_TEST) { ASSERT_FALSE(s.ok()); #ifdef MILVUS_GPU_VERSION - /* gpu resource config */ s = config.SetEngineConfigGpuSearchThreshold("-1"); ASSERT_FALSE(s.ok()); + /* gpu resource config */ s = config.SetGpuResourceConfigEnable("ok"); ASSERT_FALSE(s.ok());