Merge d3967a6 into 8d179f7

kubeflow · Apr 2, 2021 · 1c1d56c · 1c1d56c
2 parents 8d179f7 + d3967a6
commit 1c1d56c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/pkg/controller.v1/tensorflow/controller.go b/pkg/controller.v1/tensorflow/controller.go
@@ -55,6 +55,8 @@ const (
 	labelGroupName      = "group-name"
 	// Deprecated label for backwards compatibility. Has to be removed
 	labelTFJobName = "tf-job-name"
+	// volcanoTaskSpecKey task spec key used in pod annotation when EnableGangScheduling is true
+	volcanoTaskSpecKey = "volcano.sh/task-spec"
 )
 
 var (

diff --git a/pkg/controller.v1/tensorflow/pod.go b/pkg/controller.v1/tensorflow/pod.go
@@ -228,6 +228,7 @@ func (tc *TFController) createNewPod(tfjob *tfv1.TFJob, rt, index string, spec *
 			podTemplate.Annotations = map[string]string{}
 		}
 		podTemplate.Annotations[gangSchedulingPodGroupAnnotation] = tfjob.GetName()
+		podTemplate.Annotations[volcanoTaskSpecKey] = rt
 	}
 
 	err = tc.PodControl.CreatePodsWithControllerRef(tfjob.Namespace, podTemplate, tfjob, controllerRef)