Documentation
¶
Index ¶
- Constants
- func New(_ context.Context, client client.Client, _ client.FieldIndexer) (framework.Plugin, error)
- type MPI
- func (m *MPI) Build(ctx context.Context, info *runtime.Info, trainJob *trainer.TrainJob) ([]any, error)
- func (m *MPI) EnforceMLPolicy(info *runtime.Info, trainJob *trainer.TrainJob) error
- func (m *MPI) Name() string
- func (m *MPI) ReconcilerBuilders() []runtime.ReconcilerBuilder
- func (m *MPI) Validate(_ context.Context, runtimeInfo *runtime.Info, _, newJobObj *trainer.TrainJob) (admission.Warnings, field.ErrorList)
Constants ¶
View Source
const Name = "MPI"
Variables ¶
This section is empty.
Functions ¶
Types ¶
type MPI ¶
type MPI struct {
// contains filtered or unexported fields
}
func (*MPI) EnforceMLPolicy ¶
func (*MPI) ReconcilerBuilders ¶
func (m *MPI) ReconcilerBuilders() []runtime.ReconcilerBuilder
func (*MPI) Validate ¶
func (m *MPI) Validate(_ context.Context, runtimeInfo *runtime.Info, _, newJobObj *trainer.TrainJob) (admission.Warnings, field.ErrorList)
TODO (andreyvelich): Add validation to check that TrainJob doesn't have MPI envs. TODO (andreyvelich): We should validate that envs from different plugins don't conflict with each other. Ref: https://github.com/kubeflow/trainer/pull/2308#discussion_r1823229940
Source Files
Click to show internal directories.
Click to hide internal directories.