set annotation automatically when EnableGangScheduling is set to true[1032](kubeflow/trainer#1032) (#181)

zlcnju · k8s-ci-robot · commit 6aa39a411eb9 · 2019-06-26T20:01:19.000-07:00
diff --git a/pkg/controller.v1/pytorch/pod.go b/pkg/controller.v1/pytorch/pod.go
@@ -222,6 +222,12 @@ func (pc *PyTorchController) createNewPod(job *pyv1.PyTorchJob, rtype pyv1.PyTor
 		} else {
 			podTemplate.Spec.SchedulerName = gangSchedulerName
 		}
+
+		if podTemplate.Annotations == nil {
+			podTemplate.Annotations = map[string]string{}
+		}
+		// we create the podGroup with the same name as the pyTorch job
+		podTemplate.Annotations["scheduling.k8s.io/group-name"] = job.Name
 	}
 
 	err = pc.PodControl.CreatePodsWithControllerRef(job.Namespace, podTemplate, job, controllerRef)

Original file line number	Diff line number	Diff line change
`@@ -222,6 +222,12 @@ func (pc PyTorchController) createNewPod(job pyv1.PyTorchJob, rtype pyv1.PyTor`
`222`	`222`	`} else {`
`223`	`223`	`podTemplate.Spec.SchedulerName = gangSchedulerName`
`224`	`224`	`}`
	`225`	`+`
	`226`	`+ if podTemplate.Annotations == nil {`
	`227`	`+ podTemplate.Annotations = map[string]string{}`
	`228`	`+ }`
	`229`	`+ // we create the podGroup with the same name as the pyTorch job`
	`230`	`+ podTemplate.Annotations["scheduling.k8s.io/group-name"] = job.Name`
`225`	`231`	`}`
`226`	`232`
`227`	`233`	`err = pc.PodControl.CreatePodsWithControllerRef(job.Namespace, podTemplate, job, controllerRef)`