This commit is contained in:
groot 2019-11-23 14:36:22 +08:00
parent 2b805489a1
commit 8aac809ad7
2 changed files with 5 additions and 3 deletions

View File

@ -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

View File

@ -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());