From ae6ddd0aa89490a409c3795da656bc5dfe28a7c9 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Wed, 13 Jul 2022 14:42:13 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/setting/setting.go | 32 ++++++++++++++++++--------------
 1 file changed, 18 insertions(+), 14 deletions(-)

diff --git a/modules/setting/setting.go b/modules/setting/setting.go
index d206ed9a3..14d968b9e 100755
--- a/modules/setting/setting.go
+++ b/modules/setting/setting.go
@@ -453,20 +453,22 @@ var (
 	DecompressOBSTaskName   string
 
 	//cloudbrain config
-	CBAuthUser         string
-	CBAuthPassword     string
-	RestServerHost     string
-	JobPath            string
-	CBCodePathPrefix   string
-	JobType            string
-	GpuTypes           string
-	SpecialPools       string
-	DebugServerHost    string
-	ResourceSpecs      string
-	MaxDuration        int64
-	TrainGpuTypes      string
-	TrainResourceSpecs string
-	MaxDatasetNum      int
+	CBAuthUser             string
+	CBAuthPassword         string
+	RestServerHost         string
+	JobPath                string
+	CBCodePathPrefix       string
+	JobType                string
+	GpuTypes               string
+	SpecialPools           string
+	DebugServerHost        string
+	ResourceSpecs          string
+	MaxDuration            int64
+	TrainGpuTypes          string
+	TrainResourceSpecs     string
+	InferenceGpuTypes      string
+	InferenceResourceSpecs string
+	MaxDatasetNum          int
 
 	//benchmark config
 	IsBenchmarkEnabled     bool
@@ -1312,6 +1314,8 @@ func NewContext() {
 	MaxDuration = sec.Key("MAX_DURATION").MustInt64(14400)
 	TrainGpuTypes = sec.Key("TRAIN_GPU_TYPES").MustString("")
 	TrainResourceSpecs = sec.Key("TRAIN_RESOURCE_SPECS").MustString("")
+	InferenceGpuTypes = sec.Key("INFERENCE_GPU_TYPES").MustString("")
+	InferenceResourceSpecs = sec.Key("INFERENCE_RESOURCE_SPECS").MustString("")
 	SpecialPools = sec.Key("SPECIAL_POOL").MustString("")
 	
 	MaxDatasetNum = sec.Key("MAX_DATASET_NUM").MustInt(5)