Browse Source

Merge pull request '修复智算NPU任务积分消耗完后不能自动停止的问题' (#3572) from fix-2901 into V20230215

Reviewed-on: https://openi.pcl.ac.cn/OpenI/aiforge/pulls/3572
Reviewed-by: ychao_1983 <ychao_1983@sina.com>
dev-csh
ychao_1983 2 years ago
parent
commit
7108f766f9
1 changed files with 5 additions and 8 deletions
  1. +5
    -8
      routers/repo/cloudbrain.go

+ 5
- 8
routers/repo/cloudbrain.go View File

@@ -1353,14 +1353,11 @@ func StopJobs(cloudBrains []*models.Cloudbrain) {
logErrorAndUpdateJobStatus(err, taskInfo)
}
} else if taskInfo.Type == models.TypeC2Net {
if taskInfo.JobType == string(models.JobTypeTrain) {
err := retry(3, time.Second*30, func() error {
_, err := grampus.StopJob(taskInfo.JobID)
return err
})
logErrorAndUpdateJobStatus(err, taskInfo)
}

err := retry(3, time.Second*30, func() error {
_, err := grampus.StopJob(taskInfo.JobID, taskInfo.JobType)
return err
})
logErrorAndUpdateJobStatus(err, taskInfo)
}
}
}


Loading…
Cancel
Save