bugfix

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() {
@@ -227,3 +228,11 @@ func (scheduler *SchedulerFCFS) Attach(GPU string, job string) {
 func (scheduler *SchedulerFCFS) Detach(GPU string, job string) {
 	pool.detach(GPU, job)
 }
+
+func (scheduler *SchedulerFCFS) Enable() {
+	scheduler.enabled = true
+}
+
+func (scheduler *SchedulerFCFS) Disable() {
+	scheduler.enabled = false
+}

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
+}