|
@@ -2322,12 +2322,9 @@ func queryCloudBrainTask(start_unix int64, end_unix int64) (map[int64]int, map[s |
|
|
if _, ok := resourceItemMap[cloudTaskRecord.UserID]; !ok { |
|
|
if _, ok := resourceItemMap[cloudTaskRecord.UserID]; !ok { |
|
|
resourceItemMap[cloudTaskRecord.UserID] = make(map[string]int) |
|
|
resourceItemMap[cloudTaskRecord.UserID] = make(map[string]int) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
if cloudTaskRecord.Duration < 100000000 && cloudTaskRecord.Duration > 0 { |
|
|
if cloudTaskRecord.Duration < 100000000 && cloudTaskRecord.Duration > 0 { |
|
|
setMapKey("CloudBrainRunTime", cloudTaskRecord.UserID, int(cloudTaskRecord.Duration), resultItemMap) |
|
|
setMapKey("CloudBrainRunTime", cloudTaskRecord.UserID, int(cloudTaskRecord.Duration), resultItemMap) |
|
|
resourceItemMap[cloudTaskRecord.UserID][cloudTaskRecord.ComputeResource] = resourceItemMap[cloudTaskRecord.UserID][cloudTaskRecord.ComputeResource] + int(cloudTaskRecord.Duration) |
|
|
resourceItemMap[cloudTaskRecord.UserID][cloudTaskRecord.ComputeResource] = resourceItemMap[cloudTaskRecord.UserID][cloudTaskRecord.ComputeResource] + int(cloudTaskRecord.Duration) |
|
|
} else { |
|
|
|
|
|
resourceItemMap[cloudTaskRecord.UserID][cloudTaskRecord.ComputeResource] = 0 |
|
|
|
|
|
} |
|
|
} |
|
|
if cloudTaskRecord.Type == 1 { //npu |
|
|
if cloudTaskRecord.Type == 1 { //npu |
|
|
setMapKey("CloudBrainTwo", cloudTaskRecord.UserID, 1, resultItemMap) |
|
|
setMapKey("CloudBrainTwo", cloudTaskRecord.UserID, 1, resultItemMap) |
|
@@ -2359,7 +2356,7 @@ func queryCloudBrainTask(start_unix int64, end_unix int64) (map[int64]int, map[s |
|
|
} else { |
|
|
} else { |
|
|
setMapKey("NpuDebugJob", cloudTaskRecord.UserID, 1, resultItemMap) |
|
|
setMapKey("NpuDebugJob", cloudTaskRecord.UserID, 1, resultItemMap) |
|
|
} |
|
|
} |
|
|
} else if cloudTaskRecord.ComputeResource == GPUResource { |
|
|
|
|
|
|
|
|
} else { |
|
|
if cloudTaskRecord.JobType == "TRAIN" { |
|
|
if cloudTaskRecord.JobType == "TRAIN" { |
|
|
setMapKey("GpuTrainJob", cloudTaskRecord.UserID, 1, resultItemMap) |
|
|
setMapKey("GpuTrainJob", cloudTaskRecord.UserID, 1, resultItemMap) |
|
|
} else if cloudTaskRecord.JobType == "ONLINEINFERENCE" { |
|
|
} else if cloudTaskRecord.JobType == "ONLINEINFERENCE" { |
|
|