common

Documentation

Index

• Constants
• func ExtractCurrentCondition(jobConditions []commonOp.JobCondition) (commonOp.JobCondition, error)
• func GetLogs(taskType string, name string, namespace string, workersCount int32, ...) ([]*core.TaskLog, error)
• func GetPhaseInfo(currentCondition commonOp.JobCondition, occurredAt time.Time, ...) (pluginsCore.PhaseInfo, error)
• func OverrideDefaultContainerName(taskCtx pluginsCore.TaskExecutionContext, podSpec *v1.PodSpec, ...)

Constants

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

Functions

func ExtractCurrentCondition

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

func GetLogs

func GetLogs(taskType string, name string, namespace string,
	workersCount int32, psReplicasCount int32, chiefReplicasCount int32) ([]*core.TaskLog, error)

func GetPhaseInfo

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

func OverrideDefaultContainerName

func OverrideDefaultContainerName(taskCtx pluginsCore.TaskExecutionContext, podSpec *v1.PodSpec,
	defaultContainerName string)