YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-12-13 07:46:43 +00:00
Code Issues Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
Newnius
2020-04-13 18:27:57 +08:00
parent 301071fc87
commit cb6dabb41f
2 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/scheduler_FCFS.go
View File

@@ -12,7 +12,8 @@ type SchedulerFCFS struct {
mu sync.Mutex mu sync.Mutex
scheduling sync.Mutex scheduling sync.Mutex
jobs map[string]*JobManager jobs map[string]*JobManager
enabled bool
} }
func (scheduler *SchedulerFCFS) Start() { func (scheduler *SchedulerFCFS) Start() {
@@ -226,4 +227,12 @@ func (scheduler *SchedulerFCFS) Attach(GPU string, job string) {
func (scheduler *SchedulerFCFS) Detach(GPU string, job string) { func (scheduler *SchedulerFCFS) Detach(GPU string, job string) {
pool.detach(GPU, job) pool.detach(GPU, job)
} }
func (scheduler *SchedulerFCFS) Enable() {
scheduler.enabled = true
}
func (scheduler *SchedulerFCFS) Disable() {
scheduler.enabled = false
}

11
src/scheduler_priority.go
View File

@@ -12,7 +12,8 @@ type SchedulerPriority struct {
mu sync.Mutex mu sync.Mutex
scheduling sync.Mutex scheduling sync.Mutex
jobs map[string]*JobManager jobs map[string]*JobManager
enabled bool
} }
func (scheduler *SchedulerPriority) Start() { func (scheduler *SchedulerPriority) Start() {
@@ -251,3 +252,11 @@ func (scheduler *SchedulerPriority) Attach(GPU string, job string) {
func (scheduler *SchedulerPriority) Detach(GPU string, job string) { func (scheduler *SchedulerPriority) Detach(GPU string, job string) {
pool.detach(GPU, job) pool.detach(GPU, job)
} }
func (scheduler *SchedulerPriority) Enable() {
scheduler.enabled = true
}
func (scheduler *SchedulerPriority) Disable() {
scheduler.enabled = false
}