diff --git a/src/scheduler_FCFS.go b/src/scheduler_FCFS.go index 320bc6c..9feefca 100644 --- a/src/scheduler_FCFS.go +++ b/src/scheduler_FCFS.go @@ -12,7 +12,8 @@ type SchedulerFCFS struct { mu sync.Mutex scheduling sync.Mutex - jobs map[string]*JobManager + jobs map[string]*JobManager + enabled bool } func (scheduler *SchedulerFCFS) Start() { @@ -226,4 +227,12 @@ func (scheduler *SchedulerFCFS) Attach(GPU string, job string) { func (scheduler *SchedulerFCFS) Detach(GPU string, job string) { pool.detach(GPU, job) -} \ No newline at end of file +} + +func (scheduler *SchedulerFCFS) Enable() { + scheduler.enabled = true +} + +func (scheduler *SchedulerFCFS) Disable() { + scheduler.enabled = false +} diff --git a/src/scheduler_priority.go b/src/scheduler_priority.go index 13d5b81..6bc2fcf 100644 --- a/src/scheduler_priority.go +++ b/src/scheduler_priority.go @@ -12,7 +12,8 @@ type SchedulerPriority struct { mu sync.Mutex scheduling sync.Mutex - jobs map[string]*JobManager + jobs map[string]*JobManager + enabled bool } func (scheduler *SchedulerPriority) Start() { @@ -251,3 +252,11 @@ func (scheduler *SchedulerPriority) Attach(GPU string, job string) { func (scheduler *SchedulerPriority) Detach(GPU string, job string) { pool.detach(GPU, job) } + +func (scheduler *SchedulerPriority) Enable() { + scheduler.enabled = true +} + +func (scheduler *SchedulerPriority) Disable() { + scheduler.enabled = false +}