From ea0443c84eec080316a5181f0d144ec4daede940 Mon Sep 17 00:00:00 2001 From: "xj.lin" Date: Thu, 5 Sep 2019 10:37:12 +0800 Subject: [PATCH] MS-476 1. default value Former-commit-id: 57341d845d864f22604890004af28708f9fffea9 --- cpp/src/scheduler/SchedInst.cpp | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/cpp/src/scheduler/SchedInst.cpp b/cpp/src/scheduler/SchedInst.cpp index f775979e..e805c535 100644 --- a/cpp/src/scheduler/SchedInst.cpp +++ b/cpp/src/scheduler/SchedInst.cpp @@ -39,12 +39,9 @@ StartSchedulerService() { enable_executor)); if (res.lock()->Type() == ResourceType::GPU) { - auto pinned_memory = resconf.GetInt64Value(server::CONFIG_RESOURCE_PIN_MEMORY); - auto temp_memory = resconf.GetInt64Value(server::CONFIG_RESOURCE_TEMP_MEMORY); - auto resource_num = resconf.GetInt64Value(server::CONFIG_RESOURCE_NUM); - if (pinned_memory == 0) pinned_memory = 300; - if (temp_memory == 0) temp_memory = 300; - if (resource_num == 0) resource_num = 2; + auto pinned_memory = resconf.GetInt64Value(server::CONFIG_RESOURCE_PIN_MEMORY, 300); + auto temp_memory = resconf.GetInt64Value(server::CONFIG_RESOURCE_TEMP_MEMORY, 300); + auto resource_num = resconf.GetInt64Value(server::CONFIG_RESOURCE_NUM, 2); pinned_memory = 1024 * 1024 * pinned_memory; temp_memory = 1024 * 1024 * temp_memory; knowhere::FaissGpuResourceMgr::GetInstance().InitDevice(device_id, pinned_memory, temp_memory, resource_num); -- GitLab