diff --git a/cpp/src/scheduler/SchedInst.cpp b/cpp/src/scheduler/SchedInst.cpp index 67c12b8b1d..f775979e6b 100644 --- a/cpp/src/scheduler/SchedInst.cpp +++ b/cpp/src/scheduler/SchedInst.cpp @@ -42,6 +42,9 @@ StartSchedulerService() { 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; pinned_memory = 1024 * 1024 * pinned_memory; temp_memory = 1024 * 1024 * temp_memory; knowhere::FaissGpuResourceMgr::GetInstance().InitDevice(device_id, pinned_memory, temp_memory, resource_num);