#3277

gcu:fix bug

modules/grampus/grampus.go

@@ -156,7 +156,7 @@ func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (job
 	var codeGrampus models.GrampusDataset
 	var cpCommand string
 	imageUrl := req.ImageUrl
-	if ProcessorTypeNPU == req.ProcessType || ProcessorTypeGCU == req.ProcessType {
+	if ProcessorTypeNPU == req.ProcessType {
 		datasetGrampus = getDatasetGrampus(req.DatasetInfos)
 		if len(req.ModelName) != 0 {
 			datasetGrampus = append(datasetGrampus, models.GrampusDataset{

routers/repo/grampus.go

@@ -228,7 +228,7 @@ func GrampusNotebookCreate(ctx *context.Context, form auth.CreateGrampusNotebook
 		return
 	}
 
-	if processType == grampus.ProcessorTypeGPU {
+	if processType == grampus.ProcessorTypeGPU || processType == grampus.ProcessorTypeGCU {
 		if err := uploadCodeToMinio(codeLocalPath+"/", jobName, cloudbrain.CodeMountPath+"/"); err != nil {
 			log.Error("Failed to uploadCodeToMinio: %s (%v)", repo.FullName(), err, ctx.Data["MsgID"])
 			grampusNotebookNewDataPrepare(ctx, processType)