fix mpijob role

D0m021ng · Jan 8, 2024 · 017b9b1 · 017b9b1
1 parent d862b96
commit 017b9b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/apiserver/controller/job/get.go b/pkg/apiserver/controller/job/get.go
@@ -496,7 +496,7 @@ func getPyTorchJobRoleAndIndex(name string, annotations map[string]string) (sche
 // getMPIJobRoleAndIndex returns the runtime info of mpi job
 func getMPIJobRoleAndIndex(name string, annotations map[string]string) (schema.MemberRole, int) {
 	taskRole, taskIndex := schema.RoleWorker, 0
-	if annotations["volcano.sh/task-spec"] == "master" {
+	if strings.HasSuffix(name, "launcher") {
 		taskRole = schema.RoleMaster
 	} else {
 		//  worker is named with format: xxxx-worker-0