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] 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 {