add npu

routers/api/v1/repo/modelarts.go

@@ -182,7 +182,7 @@ func GetModelArtsTrainJobVersion(ctx *context.APIContext) {
 		}
 		if oldStatus != job.Status {
 			notification.NotifyChangeCloudbrainStatus(job, oldStatus)
-			if models.IsTrainJobTerminal(job.Status) {
+			if models.IsTrainJobTerminal(job.Status) && job.ComputeResource == models.NPUResource {
 				if len(result.JobInfo.Tasks[0].CenterID) == 1 {
 					urchin.GetBackNpuModel(job.ID, grampus.GetRemoteEndPoint(result.JobInfo.Tasks[0].CenterID[0]), grampus.BucketRemote, grampus.GetNpuModelObjectKey(job.JobName), grampus.GetCenterProxy(setting.Grampus.LocalCenterID))
 				}

routers/repo/cloudbrain.go

@@ -757,7 +757,7 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 	if ctx.Written() {
 		return
 	}
-    if task.Status==string(models.JobWaiting) || task.Status==string(models.JobRunning) {
+	if task.Status == string(models.JobWaiting) || task.Status == string(models.JobRunning) {
 		result, err := cloudbrain.GetJob(task.JobID)
 		if err != nil {
 			log.Info("error:" + err.Error())
@@ -1944,7 +1944,7 @@ func SyncCloudbrainStatus() {
 				task.CorrectCreateUnix()
 				if oldStatus != task.Status {
 					notification.NotifyChangeCloudbrainStatus(task, oldStatus)
-					if models.IsTrainJobTerminal(task.Status) {
+					if models.IsTrainJobTerminal(task.Status) && task.ComputeResource == models.NPUResource {
 						if len(result.JobInfo.Tasks[0].CenterID) == 1 {
 							urchin.GetBackNpuModel(task.ID, grampus.GetRemoteEndPoint(result.JobInfo.Tasks[0].CenterID[0]), grampus.BucketRemote, grampus.GetNpuModelObjectKey(task.JobName), grampus.GetCenterProxy(setting.Grampus.LocalCenterID))
 						}

routers/repo/grampus.go

@@ -879,7 +879,7 @@ func GrampusTrainJobShow(ctx *context.Context) {
 				task.CorrectCreateUnix()
 				if oldStatus != task.Status {
 					notification.NotifyChangeCloudbrainStatus(task, oldStatus)
-					if models.IsTrainJobTerminal(task.Status) {
+					if models.IsTrainJobTerminal(task.Status) && task.ComputeResource == models.NPUResource {
 						if len(result.JobInfo.Tasks[0].CenterID) == 1 {
 							urchin.GetBackNpuModel(task.ID, grampus.GetRemoteEndPoint(result.JobInfo.Tasks[0].CenterID[0]), grampus.BucketRemote, grampus.GetNpuModelObjectKey(task.JobName), grampus.GetCenterProxy(setting.Grampus.LocalCenterID))
 						}