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

update fair

Browse Source
This commit is contained in:
Newnius 2020-05-28 21:58:09 +08:00
parent a8d0d59c10
commit 3e1c1ee4f8
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/scheduler_fair.go
View File

@ -173,7 +173,27 @@ func (scheduler *SchedulerFair) Start() {
scheduler.queuesQuota[bestQueue].NumberGPU += least
log.Info(bestQueue, " borrow ", least, " from ", queue)
}
if least == 0 {
for queue, quota := range scheduler.queuesQuota {
/* do not self borrow */
if queue == bestQueue || quota.NumberGPU < least {
continue
}
quota.NumberGPU -= minRequestGPU
if _, ok := scheduler.IOUs[bestQueue]; !ok {
scheduler.IOUs[bestQueue] = map[string]*ResourceCount{}
}
IOU, ok := scheduler.IOUs[bestQueue][queue]
if !ok {
scheduler.IOUs[bestQueue][queue] = &ResourceCount{}
IOU = scheduler.IOUs[bestQueue][queue]
}
IOU.NumberGPU += minRequestGPU
scheduler.queuesQuota[bestQueue].NumberGPU += minRequestGPU
log.Info(bestQueue, " borrow ", minRequestGPU, " from ", queue)
break
}
}
}