kubeflow

Documentation

Index

• Constants
• func GetLabelSelectorForJob(job *types.Job) string
• func GetReplicaInstanceForPod(job *types.Job, pod v1.Pod) string
• type Backend
  • func New(config *restclientset.Config, namespace string) (*Backend, error)
  • func NewWithCM(config *restclientset.Config, namespace string) (*Backend, error)
  • func (b *Backend) ExecCode(parameter *types.Parameter) (*types.Job, error)
  • func (b *Backend) GetLogs(job *types.Job)

Constants

const (
	UserAgent = "kubeflow-kernel"
)

Functions

func GetLabelSelectorForJob

func GetLabelSelectorForJob(job *types.Job) string

GetLabelSelectorForJob gets label selector for the given job.

func GetReplicaInstanceForPod

func GetReplicaInstanceForPod(job *types.Job, pod v1.Pod) string

GetReplicaInstanceForPod gets the instance name of the given pod. e.g. kubeflow-xsadd-worker-0 will return worker-0.

Types

type Backend

type Backend struct {
	TFJobClient      tfjobclient.Interface
	PyTorchJobClient pyttorchjobclient.Interface
	K8sClient        kubeclient.Interface
	Generator        generator.Interface
	Namespace        string
}

Backend is the type for kubeflow backend.

func New

func New(config *restclientset.Config, namespace string) (*Backend, error)

New returns a new Backend.

func NewWithCM

func NewWithCM(config *restclientset.Config, namespace string) (*Backend, error)

func (*Backend) ExecCode

func (b *Backend) ExecCode(parameter *types.Parameter) (*types.Job, error)

ExecCode executes the code according to the parameter.

func (*Backend) GetLogs

func (b *Backend) GetLogs(job *types.Job)

GetLogs outputs logs for the given job.