set annotation automatically when EnableGangScheduling is set to true (…

…#1032)
kubeflow · Jun 24, 2019 · d746bde · d746bde
1 parent fd76dee
commit d746bde
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/controller.v1/tensorflow/pod.go b/pkg/controller.v1/tensorflow/pod.go
@@ -192,6 +192,12 @@ func (tc *TFController) createNewPod(tfjob *tfv1.TFJob, rt, index string, spec *
 		} else {
 			podTemplate.Spec.SchedulerName = gangSchedulerName
 		}
+
+		if podTemplate.Annotations == nil {
+			podTemplate.Annotations = map[string]string{}
+		}
+		// we create the podGroup with the same name as the tfjob
+		podTemplate.Annotations["scheduling.k8s.io/group-name"] = tfjob.Name
 	}
 
 	err = tc.PodControl.CreatePodsWithControllerRef(tfjob.Namespace, podTemplate, tfjob, controllerRef)