kubeflow

Documentation

Index

• Variables
• func CreatePodSetAssignment(createdWorkload *kueue.Workload, podSetsResource []PodSetsResource) []kueue.PodSetAssignment
• func JobControllerWhenWaitForPodsReadyEnabled(ctx context.Context, k8sClient client.Client, ...)
• func ShouldReconcileJob(ctx context.Context, k8sClient client.Client, ...)
• func ShouldScheduleJobsAsTheyFitInTheirClusterQueue(ctx context.Context, k8sClient client.Client, ...)
• type PodSetsResource
• type PodsReadyTestSpec

Variables

var ReplicaTypeWorker = kftraining.ReplicaType("Worker")

Functions

func CreatePodSetAssignment

func CreatePodSetAssignment(createdWorkload *kueue.Workload, podSetsResource []PodSetsResource) []kueue.PodSetAssignment

func JobControllerWhenWaitForPodsReadyEnabled

func JobControllerWhenWaitForPodsReadyEnabled(ctx context.Context, k8sClient client.Client, job, createdJob kubeflowjob.KubeflowJob, podsReadyTestSpec PodsReadyTestSpec, podSetsResources []PodSetsResource)

func ShouldReconcileJob

func ShouldReconcileJob(ctx context.Context, k8sClient client.Client, job, createdJob kubeflowjob.KubeflowJob, podSetsResources []PodSetsResource)

func ShouldScheduleJobsAsTheyFitInTheirClusterQueue

func ShouldScheduleJobsAsTheyFitInTheirClusterQueue(ctx context.Context, k8sClient client.Client, job, createdJob kubeflowjob.KubeflowJob, clusterQueue *kueue.ClusterQueue, podSetsResources []PodSetsResource)

Types

type PodSetsResource

type PodSetsResource struct {
	RoleName    kftraining.ReplicaType
	ResourceCPU kueue.ResourceFlavorReference
}

type PodsReadyTestSpec

type PodsReadyTestSpec struct {
	BeforeJobStatus *kftraining.JobStatus
	BeforeCondition *metav1.Condition
	JobStatus       kftraining.JobStatus
	Suspended       bool
	WantCondition   *metav1.Condition
}