pytorch

package
v0.14.6 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Mar 23, 2023 License: Apache-2.0 Imports: 15 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source
var (
	JobGVK               = k8s.PyTorchJobGVK
	KubePyTorchFwVersion = client.KubeFrameworkVersion(JobGVK)
)

Functions

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

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL