|
|
|
@@ -17,6 +17,7 @@ import ( |
|
|
|
"code.gitea.io/gitea/modules/notification" |
|
|
|
"code.gitea.io/gitea/modules/setting" |
|
|
|
"code.gitea.io/gitea/modules/storage" |
|
|
|
"code.gitea.io/gitea/services/cloudbrain/resource" |
|
|
|
uuid "github.com/satori/go.uuid" |
|
|
|
) |
|
|
|
|
|
|
|
@@ -69,14 +70,16 @@ func saveModelByParameters(jobId string, versionName string, name string, versio |
|
|
|
cloudType = models.TypeCloudBrainTwo |
|
|
|
} else if aiTask.ComputeResource == models.GPUResource { |
|
|
|
cloudType = models.TypeCloudBrainOne |
|
|
|
var ResourceSpecs *models.ResourceSpecs |
|
|
|
json.Unmarshal([]byte(setting.ResourceSpecs), &ResourceSpecs) |
|
|
|
for _, tmp := range ResourceSpecs.ResourceSpec { |
|
|
|
if tmp.Id == aiTask.ResourceSpecId { |
|
|
|
flaverName := ctx.Tr("cloudbrain.gpu_num") + ": " + fmt.Sprint(tmp.GpuNum) + " " + ctx.Tr("cloudbrain.cpu_num") + ": " + fmt.Sprint(tmp.CpuNum) + " " + ctx.Tr("cloudbrain.memory") + "(MB): " + fmt.Sprint(tmp.MemMiB) + " " + ctx.Tr("cloudbrain.shared_memory") + "(MB): " + fmt.Sprint(tmp.ShareMemMiB) |
|
|
|
aiTask.FlavorName = flaverName |
|
|
|
} |
|
|
|
spec, err := resource.GetAndCheckSpec(ctx.User.ID, int64(aiTask.ResourceSpecId), models.FindSpecsOptions{ |
|
|
|
JobType: models.JobType(aiTask.JobType), |
|
|
|
ComputeResource: models.GPU, |
|
|
|
Cluster: models.OpenICluster, |
|
|
|
AiCenterCode: models.AICenterOfCloudBrainOne}) |
|
|
|
if err == nil { |
|
|
|
flaverName := "GPU: " + fmt.Sprint(spec.AccCardsNum) + "*" + spec.AccCardType + ",CPU: " + fmt.Sprint(spec.CpuCores) + "," + ctx.Tr("cloudbrain.memory") + ": " + fmt.Sprint(spec.MemGiB) + "GB," + ctx.Tr("cloudbrain.shared_memory") + ": " + fmt.Sprint(spec.ShareMemGiB) + "GB" |
|
|
|
aiTask.FlavorName = flaverName |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
accuracy := make(map[string]string) |
|
|
|
|