YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-12-12 23:36:44 +00:00
Code Issues Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
Newnius
2020-04-13 18:37:54 +08:00
parent cb6dabb41f
commit 3f215b0e1d
5 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/scheduler_fair.go
View File

@@ -44,6 +44,7 @@ func (scheduler *SchedulerFair) Start() {
scheduler.queues = map[string][]Job{}
scheduler.queues["default"] = []Job{}
scheduler.resourceAllocations = map[string]*ResourceCount{}
scheduler.enabled = true
go func() {
for {
@@ -386,10 +387,12 @@ func (scheduler *SchedulerFair) Detach(GPU string, job string) {
pool.detach(GPU, job)
}
func (scheduler *SchedulerFair) Enable() {
func (scheduler *SchedulerFair) Enable() bool {
scheduler.enabled = true
return true
}
func (scheduler *SchedulerFair) Disable() {
func (scheduler *SchedulerFair) Disable() bool {
scheduler.enabled = false
return true
}