scheduler

package
v0.2.2 Latest Latest
Warning

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

Go to latest
Published: Nov 6, 2022 License: Apache-2.0 Imports: 42 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Scheduler

type Scheduler struct {
	SchedulerID string
	TotalGpus   int

	// Waiting & running jobs
	ReadyJobsMap map[string]*trainingjob.TrainingJob
	// Completed & failed jobs
	DoneJobsMap map[string]*trainingjob.TrainingJob
	// Number of allocated GPUs of each training job
	JobNumGPU types.JobScheduleResult
	// SchedulerLock is used to protect ReadyJobsMap, DoneJobsMap and JobNumGPU
	SchedulerLock sync.RWMutex

	// Algorithm to schedule training jobs
	Algorithm string

	// SchedulerMetrics contains run-time metrics of the scheduler
	Metrics SchedulerMetrics

	Router *mux.Router

	PlacementManager *placement.PlacementManager
	// contains filtered or unexported fields
}

func NewScheduler

func NewScheduler(id string, kConfig *rest.Config, resume bool, algorithm string, placementEnabled bool, useConfigMapOpt bool) (*Scheduler, error)

NewScheduler creates a new scheduler

func (*Scheduler) CreateTrainingJob

func (s *Scheduler) CreateTrainingJob(jobName string)

func (*Scheduler) DeleteTrainingJob

func (s *Scheduler) DeleteTrainingJob(jobName string)

func (*Scheduler) GetAllTrainingJob

func (s *Scheduler) GetAllTrainingJob() string

GetAllTrainingJob lists all training jobs and their scheduler, status, and waiting/running/total time

func (*Scheduler) Run

func (s *Scheduler) Run()

func (*Scheduler) Stop

func (s *Scheduler) Stop()

func (*Scheduler) TriggerResched

func (s *Scheduler) TriggerResched()

func (*Scheduler) TriggerReschedAtTime

func (s *Scheduler) TriggerReschedAtTime(time time.Time)

type SchedulerMetrics

type SchedulerMetrics struct {
	// contains filtered or unexported fields
}

Jump to

Keyboard shortcuts

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