kubeflow / kubeflow/trainer

Extend MPI Plugin with MPICH and Intel Backends

Open
#3,712 4 comments 4 reactions 0 assignees View on GitHub
area/mpi kind/feature
Dominant language
Go
Stars
2.2k
Forks
1.1k
Avg merge
3d 22h
Merged PRs (30d)
39

Description

### What you would like to be added?

During the latest Trainer call some users express interest of supporting additional backends in MPI plugin to run some HPC jobs with TrainJobs.
That will also ensure we have 1-1 parity with MPI Operator: https://youtu.be/SQGjGTvjObo?t=1407

The goal is to [extend the `MPIImplementation`](https://github.com/kubeflow/trainer/blob/master/pkg/apis/trainer/v1alpha1/trainingruntime_types.go#L311) to support required backends.

cc @vsoch @kannon92 @hahahannes @eero-t @kubeflow/kubeflow-trainer-team @robert-bell @eero-t @killianmuldoon @catblade

### Why is this needed?

Enhance MPI plugin with IntelMPI and MPICH backends

### Love this feature?

Give it a 👍 We prioritize the features with most 👍

Contributor guide

Open the contributing guide

Research direction

Start with the MPIImplementation definition in pkg/apis/trainer/v1alpha1/trainingruntime_types.go and inspect the MPI plugin's existing backend handling. Add IntelMPI and MPICH support so the MPI plugin supports the requested backends and reaches parity with the MPI Operator.

Written by the indexing model from the issue text.

Assessment

Tech stack
go, kubernetes
Domain
distributed-systems
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
48/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.