common

Documentation

Index

• Constants
• func ExtractCurrentCondition(jobConditions []commonOp.JobCondition) (commonOp.JobCondition, error)
• func ExtractMPICurrentCondition(jobConditions []commonOp.JobCondition) (commonOp.JobCondition, error)
• func GetLogs(pluginContext k8s.PluginContext, taskType string, ...) ([]*core.TaskLog, error)
• func GetMPIPhaseInfo(currentCondition commonOp.JobCondition, occurredAt time.Time, ...) (pluginsCore.PhaseInfo, error)
• func GetPhaseInfo(currentCondition commonOp.JobCondition, occurredAt time.Time, ...) (pluginsCore.PhaseInfo, error)
• func OverrideContainerSpec(podSpec *v1.PodSpec, containerName string, image string, ...) error
• func OverridePrimaryContainerName(podSpec *v1.PodSpec, primaryContainerName string, defaultContainerName string)
• func ParseCleanPodPolicy(flyteCleanPodPolicy kfplugins.CleanPodPolicy) commonOp.CleanPodPolicy
• func ParseRestartPolicy(flyteRestartPolicy kfplugins.RestartPolicy) commonOp.RestartPolicy
• func ParseRunPolicy(flyteRunPolicy kfplugins.RunPolicy) commonOp.RunPolicy
• type ReplicaEntry

Constants

const (
	TensorflowTaskType = "tensorflow"
	MPITaskType        = "mpi"
	PytorchTaskType    = "pytorch"
)

Functions

func ExtractCurrentCondition

func ExtractCurrentCondition(jobConditions []commonOp.JobCondition) (commonOp.JobCondition, error)

ExtractCurrentCondition will return the first job condition for tensorflow/pytorch

func ExtractMPICurrentCondition

func ExtractMPICurrentCondition(jobConditions []commonOp.JobCondition) (commonOp.JobCondition, error)

ExtractMPICurrentCondition will return the first job condition for MPI

func GetLogs

func GetLogs(pluginContext k8s.PluginContext, taskType string, objectMeta meta_v1.ObjectMeta, hasMaster bool,
	workersCount int32, psReplicasCount int32, chiefReplicasCount int32) ([]*core.TaskLog, error)

GetLogs will return the logs for kubeflow job

func GetMPIPhaseInfo

func GetMPIPhaseInfo(currentCondition commonOp.JobCondition, occurredAt time.Time,
	taskPhaseInfo pluginsCore.TaskInfo) (pluginsCore.PhaseInfo, error)

GetMPIPhaseInfo will return the phase of MPI job

func GetPhaseInfo

func GetPhaseInfo(currentCondition commonOp.JobCondition, occurredAt time.Time,
	taskPhaseInfo pluginsCore.TaskInfo) (pluginsCore.PhaseInfo, error)

GetPhaseInfo will return the phase of kubeflow job

func OverrideContainerSpec

func OverrideContainerSpec(podSpec *v1.PodSpec, containerName string, image string, resources *core.Resources, args []string) error

OverrideContainerSpec overrides the specified container's properties in the given podSpec. The function updates the image, resources and command arguments of the container that matches the given containerName.

func OverridePrimaryContainerName

func OverridePrimaryContainerName(podSpec *v1.PodSpec, primaryContainerName string, defaultContainerName string)

func ParseCleanPodPolicy

func ParseCleanPodPolicy(flyteCleanPodPolicy kfplugins.CleanPodPolicy) commonOp.CleanPodPolicy

Get k8s clean pod policy from flyte kubeflow plugins clean pod policy.

func ParseRestartPolicy

func ParseRestartPolicy(flyteRestartPolicy kfplugins.RestartPolicy) commonOp.RestartPolicy

Get k8s restart policy from flyte kubeflow plugins restart policy.

func ParseRunPolicy

func ParseRunPolicy(flyteRunPolicy kfplugins.RunPolicy) commonOp.RunPolicy

ParseRunPolicy converts a kubeflow plugin RunPolicy object to a k8s RunPolicy object.

Types

type ReplicaEntry

type ReplicaEntry struct {
	PodSpec       *v1.PodSpec
	ReplicaNum    int32
	RestartPolicy commonOp.RestartPolicy
}