| @@ -2516,7 +2516,7 @@ func queryCloudBrainTask(start_unix int64, end_unix int64) (map[int64]int, map[s | |||||
| setMapKey("NpuTrainJob", cloudTaskRecord.UserID, 1, resultItemMap) | setMapKey("NpuTrainJob", cloudTaskRecord.UserID, 1, resultItemMap) | ||||
| } else if cloudTaskRecord.JobType == "INFERENCE" { | } else if cloudTaskRecord.JobType == "INFERENCE" { | ||||
| setMapKey("NpuInferenceJob", cloudTaskRecord.UserID, 1, resultItemMap) | setMapKey("NpuInferenceJob", cloudTaskRecord.UserID, 1, resultItemMap) | ||||
| } else if cloudTaskRecord.JobType == "BENCHMARK" { | |||||
| } else if cloudTaskRecord.JobType == "BENCHMARK" || cloudTaskRecord.JobType == "MODELSAFETY" { | |||||
| setMapKey("GpuBenchMarkJob", cloudTaskRecord.UserID, 1, resultItemMap) | setMapKey("GpuBenchMarkJob", cloudTaskRecord.UserID, 1, resultItemMap) | ||||
| } else { | } else { | ||||
| setMapKey("NpuDebugJob", cloudTaskRecord.UserID, 1, resultItemMap) | setMapKey("NpuDebugJob", cloudTaskRecord.UserID, 1, resultItemMap) | ||||
| @@ -2527,7 +2527,7 @@ func queryCloudBrainTask(start_unix int64, end_unix int64) (map[int64]int, map[s | |||||
| setMapKey("GpuTrainJob", cloudTaskRecord.UserID, 1, resultItemMap) | setMapKey("GpuTrainJob", cloudTaskRecord.UserID, 1, resultItemMap) | ||||
| } else if cloudTaskRecord.JobType == "INFERENCE" { | } else if cloudTaskRecord.JobType == "INFERENCE" { | ||||
| setMapKey("GpuInferenceJob", cloudTaskRecord.UserID, 1, resultItemMap) | setMapKey("GpuInferenceJob", cloudTaskRecord.UserID, 1, resultItemMap) | ||||
| } else if cloudTaskRecord.JobType == "BENCHMARK" { | |||||
| } else if cloudTaskRecord.JobType == "BENCHMARK" || cloudTaskRecord.JobType == "MODELSAFETY" { | |||||
| setMapKey("GpuBenchMarkJob", cloudTaskRecord.UserID, 1, resultItemMap) | setMapKey("GpuBenchMarkJob", cloudTaskRecord.UserID, 1, resultItemMap) | ||||
| } else { | } else { | ||||
| setMapKey("GpuDebugJob", cloudTaskRecord.UserID, 1, resultItemMap) | setMapKey("GpuDebugJob", cloudTaskRecord.UserID, 1, resultItemMap) | ||||