pytorch

Documentation

Index

• func New(kubeClient framework.RuntimeClientInterface) framework.JobInterface
• type KubePyTorchJob
  • func (pj *KubePyTorchJob) AddEventListener(ctx context.Context, listenerType string, ...) error
  • func (pj *KubePyTorchJob) JobStatus(obj interface{}) (api.StatusInfo, error)
  • func (pj *KubePyTorchJob) Submit(ctx context.Context, job *api.PFJob) error

Functions

func New

func New(kubeClient framework.RuntimeClientInterface) framework.JobInterface

Types

type KubePyTorchJob

type KubePyTorchJob struct {
	kuberuntime.KubeBaseJob
}

KubePyTorchJob is a struct that runs a pytorch job

func (*KubePyTorchJob) AddEventListener

func (pj *KubePyTorchJob) AddEventListener(ctx context.Context, listenerType string, jobQueue workqueue.RateLimitingInterface, listener interface{}) error

func (*KubePyTorchJob) JobStatus

func (pj *KubePyTorchJob) JobStatus(obj interface{}) (api.StatusInfo, error)

JobStatus get the statusInfo of PyTorch job, including origin status, pf status and message

func (*KubePyTorchJob) Submit

func (pj *KubePyTorchJob) Submit(ctx context.Context, job *api.PFJob) error