|
|
@@ -346,6 +346,24 @@ func CloudBrainRestart(ctx *context.Context) { |
|
|
|
break |
|
|
|
} |
|
|
|
|
|
|
|
var hasSameResource bool |
|
|
|
if gpuInfos == nil { |
|
|
|
json.Unmarshal([]byte(setting.GpuTypes), &gpuInfos) |
|
|
|
} |
|
|
|
for _, resourceType := range gpuInfos.GpuInfo { |
|
|
|
if resourceType.Queue == task.GpuQueue { |
|
|
|
hasSameResource = true |
|
|
|
continue |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
if !hasSameResource { |
|
|
|
log.Error("has no same resource, can not restart", ctx.Data["MsgID"]) |
|
|
|
resultCode = "-1" |
|
|
|
errorMsg = "the job's version is too old and can not be restarted" |
|
|
|
break |
|
|
|
} |
|
|
|
|
|
|
|
count, err := models.GetCloudbrainCountByUserID(ctx.User.ID, string(models.JobTypeDebug)) |
|
|
|
if err != nil { |
|
|
|
log.Error("GetCloudbrainCountByUserID failed:%v", err, ctx.Data["MsgID"]) |
|
|
|