testing

Documentation

Index

• type PyTorchJobWrapper
  • func MakePyTorchJob(name, ns string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) Clone() *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) Image(replicaType kftraining.ReplicaType, image string, args []string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) Label(key, value string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) Obj() *kftraining.PyTorchJob
  • func (j *PyTorchJobWrapper) Parallelism(p int32) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) PodAnnotation(replicaType kftraining.ReplicaType, k, v string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) PodLabel(replicaType kftraining.ReplicaType, k, v string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) PriorityClass(pc string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) PyTorchReplicaSpecs(replicaSpecs ...PyTorchReplicaSpecRequirement) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) PyTorchReplicaSpecsDefault() *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) Queue(queue string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) Request(replicaType kftraining.ReplicaType, r corev1.ResourceName, v string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) StatusConditions(c kftraining.JobCondition) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) Suspend(s bool) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) UID(uid string) *PyTorchJobWrapper
  • func (j *PyTorchJobWrapper) WorkloadPriorityClass(wpc string) *PyTorchJobWrapper
• type PyTorchReplicaSpecRequirement

Types

type PyTorchJobWrapper

type PyTorchJobWrapper struct{ kftraining.PyTorchJob }

PyTorchJobWrapper wraps a Job.

func MakePyTorchJob

func MakePyTorchJob(name, ns string) *PyTorchJobWrapper

MakePyTorchJob creates a wrapper for a suspended job with a single container and parallelism=1.

func (*PyTorchJobWrapper) Clone

func (j *PyTorchJobWrapper) Clone() *PyTorchJobWrapper

Clone returns deep copy of the PyTorchJobWrapper.

func (*PyTorchJobWrapper) Image

func (j *PyTorchJobWrapper) Image(replicaType kftraining.ReplicaType, image string, args []string) *PyTorchJobWrapper

func (*PyTorchJobWrapper) Label

func (j *PyTorchJobWrapper) Label(key, value string) *PyTorchJobWrapper

Label sets the label key and value

func (*PyTorchJobWrapper) Obj

func (j *PyTorchJobWrapper) Obj() *kftraining.PyTorchJob

Obj returns the inner Job.

func (*PyTorchJobWrapper) Parallelism

func (j *PyTorchJobWrapper) Parallelism(p int32) *PyTorchJobWrapper

Parallelism updates job parallelism.

func (*PyTorchJobWrapper) PodAnnotation

func (j *PyTorchJobWrapper) PodAnnotation(replicaType kftraining.ReplicaType, k, v string) *PyTorchJobWrapper

PodAnnotation sets annotation at the pod template level

func (*PyTorchJobWrapper) PodLabel

func (j *PyTorchJobWrapper) PodLabel(replicaType kftraining.ReplicaType, k, v string) *PyTorchJobWrapper

PodLabel sets label at the pod template level

func (*PyTorchJobWrapper) PriorityClass

func (j *PyTorchJobWrapper) PriorityClass(pc string) *PyTorchJobWrapper

PriorityClass updates job priorityclass.

func (*PyTorchJobWrapper) PyTorchReplicaSpecs

func (j *PyTorchJobWrapper) PyTorchReplicaSpecs(replicaSpecs ...PyTorchReplicaSpecRequirement) *PyTorchJobWrapper

func (*PyTorchJobWrapper) PyTorchReplicaSpecsDefault

func (j *PyTorchJobWrapper) PyTorchReplicaSpecsDefault() *PyTorchJobWrapper

func (*PyTorchJobWrapper) Queue

func (j *PyTorchJobWrapper) Queue(queue string) *PyTorchJobWrapper

Queue updates the queue name of the job.

func (*PyTorchJobWrapper) Request

func (j *PyTorchJobWrapper) Request(replicaType kftraining.ReplicaType, r corev1.ResourceName, v string) *PyTorchJobWrapper

Request adds a resource request to the default container.

func (*PyTorchJobWrapper) StatusConditions

func (j *PyTorchJobWrapper) StatusConditions(c kftraining.JobCondition) *PyTorchJobWrapper

Condition adds a condition

func (*PyTorchJobWrapper) Suspend

func (j *PyTorchJobWrapper) Suspend(s bool) *PyTorchJobWrapper

Suspend updates the suspend status of the job.

func (*PyTorchJobWrapper) UID

func (j *PyTorchJobWrapper) UID(uid string) *PyTorchJobWrapper

UID updates the uid of the job.

func (*PyTorchJobWrapper) WorkloadPriorityClass

func (j *PyTorchJobWrapper) WorkloadPriorityClass(wpc string) *PyTorchJobWrapper

WorkloadPriorityClass updates job workloadpriorityclass.

type PyTorchReplicaSpecRequirement

type PyTorchReplicaSpecRequirement struct {
	ReplicaType   kftraining.ReplicaType
	Name          string
	ReplicaCount  int32
	Annotations   map[string]string
	RestartPolicy kftraining.RestartPolicy
}