From 0f5f0d00191e87df94991972b9015cec57474c3d Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Thu, 25 Nov 2021 11:45:13 +0800
Subject: [PATCH 1/2] add gpu type

---
 templates/repo/cloudbrain/new.tmpl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/templates/repo/cloudbrain/new.tmpl b/templates/repo/cloudbrain/new.tmpl
index b3b827558..83905ce08 100755
--- a/templates/repo/cloudbrain/new.tmpl
+++ b/templates/repo/cloudbrain/new.tmpl
@@ -165,7 +165,7 @@
                     </div>
                     <input  id="store_category" type="hidden" name="get_benchmark_category">
 
-                    <div class="inline required field cloudbrain_benchmark">
+                    <div class="inline required field">
                         <label>GPU类型</label>
                         <select id="cloudbrain_gpu_type" class="ui search dropdown" placeholder="选择GPU类型" style='width:385px' name="gpu_type">
                             {{range .gpu_types}}

From 651ffee9a4b7f3e39eb572fe63c34adde8dcf8ba Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Thu, 25 Nov 2021 15:10:00 +0800
Subject: [PATCH 2/2] add V100

---
 routers/repo/cloudbrain.go | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 1e1f7bf38..28f3a0184 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -200,7 +200,7 @@ func CloudBrainCreate(ctx *context.Context, form auth.CreateCloudBrainForm) {
 	command := form.Command
 	uuid := form.Attachment
 	jobType := form.JobType
-	gpuQueue := setting.JobType
+	gpuQueue := form.GpuType
 	codePath := setting.JobPath + jobName + cloudbrain.CodeMountPath
 	resourceSpecId := form.ResourceSpecId
 
@@ -243,7 +243,6 @@ func CloudBrainCreate(ctx *context.Context, form auth.CreateCloudBrainForm) {
 
 	benchmarkPath := setting.JobPath + jobName + cloudbrain.BenchMarkMountPath
 	if setting.IsBenchmarkEnabled && jobType == string(models.JobTypeBenchmark) {
-		gpuQueue = form.GpuType
 		var gpuType string
 		for _, gpuInfo := range gpuInfos.GpuInfo {
 			if gpuInfo.Queue == gpuQueue {