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: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

4
src/main.go
View File

@@ -168,14 +168,14 @@ func serverAPI(w http.ResponseWriter, r *http.Request) {
case "debug_enable": case "debug_enable":
log.Debug("enable schedule") log.Debug("enable schedule")
js, _ := json.Marshal(scheduler.Enable) js, _ := json.Marshal(scheduler.Enable())
w.Header().Set("Content-Type", "application/json") w.Header().Set("Content-Type", "application/json")
w.Write(js) w.Write(js)
break break
case "debug_disable": case "debug_disable":
log.Debug("disable schedule") log.Debug("disable schedule")
js, _ := json.Marshal(scheduler.Disable) js, _ := json.Marshal(scheduler.Disable())
w.Header().Set("Content-Type", "application/json") w.Header().Set("Content-Type", "application/json")
w.Write(js) w.Write(js)
break break

4
src/scheduler.go
View File

@@ -29,7 +29,7 @@ type Scheduler interface {
Detach(GPU string, job string) Detach(GPU string, job string)
Enable() Enable() bool
Disable() Disable() bool
} }

7
src/scheduler_FCFS.go
View File

@@ -19,6 +19,7 @@ type SchedulerFCFS struct {
func (scheduler *SchedulerFCFS) Start() { func (scheduler *SchedulerFCFS) Start() {
scheduler.jobs = map[string]*JobManager{} scheduler.jobs = map[string]*JobManager{}
scheduler.history = []*Job{} scheduler.history = []*Job{}
scheduler.enabled = true
go func() { go func() {
for { for {
@@ -229,10 +230,12 @@ func (scheduler *SchedulerFCFS) Detach(GPU string, job string) {
pool.detach(GPU, job) pool.detach(GPU, job)
} }
func (scheduler *SchedulerFCFS) Enable() { func (scheduler *SchedulerFCFS) Enable() bool {
scheduler.enabled = true scheduler.enabled = true
return true
} }
func (scheduler *SchedulerFCFS) Disable() { func (scheduler *SchedulerFCFS) Disable() bool {
scheduler.enabled = false scheduler.enabled = false
return true
} }

7
src/scheduler_fair.go
View File

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

7
src/scheduler_priority.go
View File

@@ -19,6 +19,7 @@ type SchedulerPriority struct {
func (scheduler *SchedulerPriority) Start() { func (scheduler *SchedulerPriority) Start() {
scheduler.jobs = map[string]*JobManager{} scheduler.jobs = map[string]*JobManager{}
scheduler.history = []*Job{} scheduler.history = []*Job{}
scheduler.enabled = true
go func() { go func() {
for { for {
@@ -253,10 +254,12 @@ func (scheduler *SchedulerPriority) Detach(GPU string, job string) {
pool.detach(GPU, job) pool.detach(GPU, job)
} }
func (scheduler *SchedulerPriority) Enable() { func (scheduler *SchedulerPriority) Enable() bool {
scheduler.enabled = true scheduler.enabled = true
return true
} }
func (scheduler *SchedulerPriority) Disable() { func (scheduler *SchedulerPriority) Disable() bool {
scheduler.enabled = false scheduler.enabled = false
return true
} }