From cb6dabb41f3b35d09f19197cefaf7863ae444d11 Mon Sep 17 00:00:00 2001 From: Newnius Date: Mon, 13 Apr 2020 18:27:57 +0800 Subject: [PATCH] bugfix --- src/scheduler_FCFS.go | 13 +++++++++++-- src/scheduler_priority.go | 11 ++++++++++- 2 files changed, 21 insertions(+), 3 deletions(-) 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 +}