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

Browse Source
This commit is contained in:
Newnius 2020-05-03 11:38:59 +08:00
parent da754b3336
commit 87bb4a9b39
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/scheduler_fair.go
View File

@ -127,13 +127,8 @@ func (scheduler *SchedulerFair) Start() {
} }
scheduler.queuesUsingGPUMu.Unlock() scheduler.queuesUsingGPUMu.Unlock()
if cnt == 1 { log.Info(cnt, reserved, pool.TotalGPU, scheduler.UsingGPU, scheduler.allocatingGPU)
cnt *= 10 if cnt*10+(scheduler.allocatingGPU)*13 > (pool.TotalGPU-scheduler.UsingGPU-reserved)*10 {
} else {
cnt *= 13
}
log.Info(cnt, reserved, pool.TotalGPU, scheduler.UsingGPU)
if cnt+(scheduler.allocatingGPU)*13 > (pool.TotalGPU-scheduler.UsingGPU-reserved)*10 {
scheduler.schedulingMu.Lock() scheduler.schedulingMu.Lock()
scheduler.schedulingJobsCnt-- scheduler.schedulingJobsCnt--
scheduler.schedulingMu.Unlock() scheduler.schedulingMu.Unlock()
@ -751,7 +746,7 @@ func (scheduler *SchedulerFair) UpdateNextQueue() {
} }
scheduler.nextQueue = next scheduler.nextQueue = next
scheduler.queueMu.Unlock() scheduler.queueMu.Unlock()
log.Info("updateNextQueue ->", next) log.Debug("updateNextQueue ->", next)
} }
func (scheduler *SchedulerFair) Attach(GPU string, job string) { func (scheduler *SchedulerFair) Attach(GPU string, job string) {