add option for mpi kubectl delivery (#1525)

* add option for select kubectl delivery image for mpi

* unify receiver name and delete redundant conversion

cmd/training-operator.v1/main.go

@@ -85,6 +85,10 @@ func main() {
 	flag.StringVar(&config.Config.PyTorchInitContainerTemplateFile, "pytorch-init-container-template-file",
 		config.PyTorchInitContainerTemplateFileDefault, "The template file for pytorch init container")
 
+	// MPI related flags
+	flag.StringVar(&config.Config.MPIKubectlDeliveryImage, "mpi-kubectl-delivery-image",
+		config.MPIKubectlDeliveryImageDefault, "The image for mpi launcher init container")
+
 	opts := zap.Options{
 		Development:     true,
 		StacktraceLevel: zapcore.DPanicLevel,

pkg/config/config.go

@@ -18,6 +18,7 @@ package config
 var Config struct {
 	PyTorchInitContainerTemplateFile string
 	PyTorchInitContainerImage        string
+	MPIKubectlDeliveryImage          string
 }
 
 const (
@@ -27,4 +28,6 @@ const (
 	// PyTorchInitContainerTemplateFileDefault is the default template file for
 	// the pytorch init container.
 	PyTorchInitContainerTemplateFileDefault = "/etc/config/initContainer.yaml"
+	// MPIKubectlDeliveryImageDefault is the default image for launcher pod in MPIJob init container.
+	MPIKubectlDeliveryImageDefault = "mpioperator/kubectl-delivery:latest"
 )

pkg/controller.v1/mpi/mpijob.go

@@ -75,8 +75,6 @@ const (
 	// gang scheduler name.
 	gangSchedulerName = "volcano"
 
-	kubectlDeliveryImage = "mpioperator/kubectl-delivery:latest"
-
 	// podTemplateSchedulerNameReason is the warning reason when other scheduler name is set
 	// in pod templates with gang-scheduling enabled
 	podTemplateSchedulerNameReason = "SettedPodTemplateSchedulerName"