controller_v1

Documentation

Index

• Constants
• Variables
• type EnabledSchemes
  • func (es *EnabledSchemes) Empty() bool
  • func (es *EnabledSchemes) FillAll()
  • func (es *EnabledSchemes) Set(kind string) error
  • func (es *EnabledSchemes) String() string
• type ReconcilerSetupFunc

Constants

const ErrTemplateSchemeNotSupported = "scheme %s is not supported yet"

Variables

var SupportedSchemeReconciler = map[string]ReconcilerSetupFunc{
	kubeflowv1.TFJobKind: func(mgr manager.Manager, gangSchedulingSetupFunc common.GangSchedulingSetupFunc, controllerThreads int) error {
		return tensorflowcontroller.NewReconciler(mgr, gangSchedulingSetupFunc).SetupWithManager(mgr, controllerThreads)
	},
	kubeflowv1.PyTorchJobKind: func(mgr manager.Manager, gangSchedulingSetupFunc common.GangSchedulingSetupFunc, controllerThreads int) error {
		return pytorchcontroller.NewReconciler(mgr, gangSchedulingSetupFunc).SetupWithManager(mgr, controllerThreads)
	},
	kubeflowv1.XGBoostJobKind: func(mgr manager.Manager, gangSchedulingSetupFunc common.GangSchedulingSetupFunc, controllerThreads int) error {
		return xgboostcontroller.NewReconciler(mgr, gangSchedulingSetupFunc).SetupWithManager(mgr, controllerThreads)
	},
	kubeflowv1.MPIJobKind: func(mgr manager.Manager, gangSchedulingSetupFunc common.GangSchedulingSetupFunc, controllerThreads int) error {
		return mpicontroller.NewReconciler(mgr, gangSchedulingSetupFunc).SetupWithManager(mgr, controllerThreads)
	},
	kubeflowv1.PaddleJobKind: func(mgr manager.Manager, gangSchedulingSetupFunc common.GangSchedulingSetupFunc, controllerThreads int) error {
		return paddlecontroller.NewReconciler(mgr, gangSchedulingSetupFunc).SetupWithManager(mgr, controllerThreads)
	},
	kubeflowv1.JAXJobKind: func(mgr manager.Manager, gangSchedulingSetupFunc common.GangSchedulingSetupFunc, controllerThreads int) error {
		return jaxcontroller.NewReconciler(mgr, gangSchedulingSetupFunc).SetupWithManager(mgr, controllerThreads)
	},
}

Types

type EnabledSchemes

type EnabledSchemes []string

func (*EnabledSchemes) Empty

func (es *EnabledSchemes) Empty() bool

func (*EnabledSchemes) FillAll

func (es *EnabledSchemes) FillAll()

func (*EnabledSchemes) Set

func (es *EnabledSchemes) Set(kind string) error

func (*EnabledSchemes) String

func (es *EnabledSchemes) String() string

type ReconcilerSetupFunc

type ReconcilerSetupFunc func(manager manager.Manager, gangSchedulingSetupFunc common.GangSchedulingSetupFunc, controllerThreads int) error