Browse Source

Merge branch 'V20220926' into fix-2069

tags/v1.22.9.2^2
liuzx 3 years ago
parent
commit
46ed205606
3 changed files with 8 additions and 15 deletions
  1. +1
    -1
      models/cloudbrain.go
  2. +4
    -0
      routers/repo/cloudbrain.go
  3. +3
    -14
      routers/repo/grampus.go

+ 1
- 1
models/cloudbrain.go View File

@@ -279,7 +279,7 @@ func (task *Cloudbrain) IsRunning() bool {
}

func ConvertDurationToStr(duration int64) string {
if duration == 0 {
if duration <= 0 {
return DURATION_STR_ZERO
}
return util.AddZero(duration/3600) + ":" + util.AddZero(duration%3600/60) + ":" + util.AddZero(duration%60)


+ 4
- 0
routers/repo/cloudbrain.go View File

@@ -1862,6 +1862,10 @@ func SyncCloudbrainStatus() {
oldStatus := task.Status
task.Status = grampus.TransTrainJobStatus(result.JobInfo.Status)
task.Duration = result.JobInfo.RunSec

if task.Duration < 0 {
task.Duration = 0
}
task.TrainJobDuration = models.ConvertDurationToStr(task.Duration)

if task.StartTime == 0 && result.JobInfo.StartedAt > 0 {


+ 3
- 14
routers/repo/grampus.go View File

@@ -242,13 +242,6 @@ func GrampusTrainJobGpuCreate(ctx *context.Context, form auth.CreateGrampusTrain
return
}

errStr := checkSpecialPool(ctx, "GPU")
if errStr != "" {
grampusTrainJobNewDataPrepare(ctx, grampus.ProcessorTypeGPU)
ctx.RenderWithErr(errStr, tplGrampusTrainJobGPUNew, &form)
return
}

//check count limit
count, err := models.GetGrampusCountByUserID(ctx.User.ID, string(models.JobTypeTrain), models.GPUResource)
if err != nil {
@@ -461,13 +454,6 @@ func GrampusTrainJobNpuCreate(ctx *context.Context, form auth.CreateGrampusTrain
return
}

errStr := checkSpecialPool(ctx, "NPU")
if errStr != "" {
grampusTrainJobNewDataPrepare(ctx, grampus.ProcessorTypeNPU)
ctx.RenderWithErr(errStr, tplGrampusTrainJobGPUNew, &form)
return
}

//check count limit
count, err := models.GetGrampusCountByUserID(ctx.User.ID, string(models.JobTypeTrain), models.NPUResource)
if err != nil {
@@ -732,6 +718,9 @@ func GrampusTrainJobShow(ctx *context.Context) {
task.Status = grampus.TransTrainJobStatus(result.JobInfo.Status)
if task.Status != result.JobInfo.Status || result.JobInfo.Status == models.GrampusStatusRunning {
task.Duration = result.JobInfo.RunSec
if task.Duration < 0 {
task.Duration = 0
}
task.TrainJobDuration = models.ConvertDurationToStr(task.Duration)

if task.StartTime == 0 && result.JobInfo.StartedAt > 0 {


Loading…
Cancel
Save