MS-476 1. default value

Former-commit-id: 57341d845d864f22604890004af28708f9fffea9

MS-476 1. default value
Former-commit-id: 57341d845d864f22604890004af28708f9fffea9
ea0443c8 · xj.lin · b2488256 · ea0443c8
隐藏空白更改
内联并排

Showing with 3 addition and 6 deletion

cpp/src/scheduler/SchedInst.cpp cpp/src/scheduler/SchedInst.cpp +3 -6

未找到文件。
--- 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);