models/cloudbrain.go

@@ -3046,10 +3046,25 @@ func GetDatasetInfo(uuidStr string, grampusType ...string) (map[string]DatasetIn
 			}
 		}
 		var dataLocalPath string
-		if attach.Type == TypeCloudBrainOne {
-			dataLocalPath = setting.Attachment.Minio.BasePath + path.Join(attach.UUID[0:1], attach.UUID[1:2]) + "/" + attach.UUID
+		if len(grampusType) > 0 {
+			if grampusType[0] == GPU {
+				dataLocalPath = setting.Attachment.Minio.BasePath + path.Join(attach.UUID[0:1], attach.UUID[1:2]) + "/" + attach.UUID
+			} else if grampusType[0] == NPU {
+				dataLocalPath = setting.BasePath + path.Join(attach.UUID[0:1], attach.UUID[1:2]) + "/" + attach.UUID + "/"
+			} else if grampusType[0] == GCU || grampusType[0] == MLU {
+				if attach.Type == TypeCloudBrainOne {
+					dataLocalPath = setting.Attachment.Minio.BasePath + path.Join(attach.UUID[0:1], attach.UUID[1:2]) + "/" + attach.UUID
+				} else {
+					dataLocalPath = setting.BasePath + path.Join(attach.UUID[0:1], attach.UUID[1:2]) + "/" + attach.UUID + "/"
+				}
+			}
+
 		} else {
-			dataLocalPath = setting.BasePath + path.Join(attach.UUID[0:1], attach.UUID[1:2]) + "/" + attach.UUID + "/"
+			dataLocalPath = setting.Attachment.Minio.RealPath +
+				setting.Attachment.Minio.Bucket + "/" +
+				setting.Attachment.Minio.BasePath +
+				AttachmentRelativePath(attach.UUID) +
+				attach.UUID
 		}
 
 		datasetInfos[attach.UUID] = DatasetInfo{