YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-07 14:21:55 +00:00
Code Issues Releases Wiki Activity

update fair

Browse Source
This commit is contained in:
Newnius 2020-06-03 17:39:36 +08:00
parent 1e43396a2d
commit 422b043b8c
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/scheduler_priority.go
View File

@ -84,6 +84,7 @@ func (scheduler *SchedulerPriority) Start() {
} }
sort.Sort(JobSorter(jobs)) sort.Sort(JobSorter(jobs))
if len(jobs) > 0 { if len(jobs) > 0 {
before := InstanceOfResourcePool().UsingGPU
preempted := jobs[0] preempted := jobs[0]
log.Info("Start preempt ", preempted.Name) log.Info("Start preempt ", preempted.Name)
scheduler.Stop(preempted.Name) scheduler.Stop(preempted.Name)
@ -118,6 +119,14 @@ func (scheduler *SchedulerPriority) Start() {
log.Info(scheduler.queue) log.Info(scheduler.queue)
delete(scheduler.jobs, preempted.Name) delete(scheduler.jobs, preempted.Name)
for {
after := InstanceOfResourcePool().UsingGPU
if after != before {
break
}
time.Sleep(time.Millisecond * 100)
}
} }
scheduler.historyMu.Unlock() scheduler.historyMu.Unlock()
} }