Adding MPI python sdk

hack/python-sdk/post_gen.py

@@ -54,6 +54,7 @@ def add_imports() -> None:
         init_file.write("from kubeflow.training.api.tf_job_client import TFJobClient\n")
         init_file.write("from kubeflow.training.api.py_torch_job_client import PyTorchJobClient\n")
         init_file.write("from kubeflow.training.api.xgboost_job_client import XGBoostJobClient\n")
+        init_file.write("from kubeflow.training.api.mpi_job_client import MPIJobClient\n")
     with open(os.path.join(sdk_dir, "kubeflow/__init__.py"), "a") as init_file:
         init_file.write("__path__ = __import__('pkgutil').extend_path(__path__, __name__)")
 

pkg/controller.v1/mpi/mpijob_controller.go

@@ -291,7 +291,7 @@ func (jc *MPIJobReconciler) GetDefaultContainerPortName() string {
 
 func (jc *MPIJobReconciler) IsMasterRole(replicas map[commonv1.ReplicaType]*commonv1.ReplicaSpec,
 	rtype commonv1.ReplicaType, index int) bool {
-	return false
+	return string(rtype) == string(mpiv1.MPIReplicaTypeLauncher)
 }
 
 func (jc *MPIJobReconciler) GetJobFromInformerCache(namespace, name string) (metav1.Object, error) {
@@ -1039,6 +1039,10 @@ func (jc *MPIJobReconciler) newLauncher(mpiJob *mpiv1.MPIJob, kubectlDeliveryIma
 	genericLabels := jc.GenLabels(mpiJob.GetName())
 	labels := defaultLauncherLabels(genericLabels)
 
+	masterRole := jc.IsMasterRole(mpiJob.Spec.MPIReplicaSpecs, mpiv1.MPIReplicaTypeLauncher, 0)
+	if masterRole {
+		labels[commonv1.JobRoleLabel] = "master"
+	}
 	podSpec := mpiJob.Spec.MPIReplicaSpecs[mpiv1.MPIReplicaTypeLauncher].Template.DeepCopy()
 	// copy the labels and annotations to pod from PodTemplate
 	if len(podSpec.Labels) == 0 {

sdk/python/kubeflow/training/__init__.py

@@ -54,3 +54,4 @@
 from kubeflow.training.api.tf_job_client import TFJobClient
 from kubeflow.training.api.py_torch_job_client import PyTorchJobClient
 from kubeflow.training.api.xgboost_job_client import XGBoostJobClient
+from kubeflow.training.api.mpi_job_client import MPIJobClient