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

update fair

Browse Source
This commit is contained in:
Newnius 2020-05-28 10:28:02 +08:00
parent 7d1298a0f7
commit 31e45ec001
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/scheduler_fair.go
View File

@ -320,6 +320,24 @@ func (scheduler *SchedulerFair) UpdateQuota() {
available := pool.TotalGPU - usingGPU available := pool.TotalGPU - usingGPU
log.Info("Can allocate ", available) log.Info("Can allocate ", available)
log.Info("Before ", scheduler.queuesQuota)
per := available / len(scheduler.queues)
for queue := range scheduler.queues {
if _, ok := scheduler.queuesQuota[queue]; !ok {
scheduler.queuesQuota[queue] = ResourceCount{}
}
quota := scheduler.queuesQuota[queue]
quota.NumberGPU += per
available -= per
}
if available > 0 {
for queue := range scheduler.queues {
quota := scheduler.queuesQuota[queue]
quota.NumberGPU += available
break
}
}
log.Info("After ", scheduler.queuesQuota)
} }
func (scheduler *SchedulerFair) QueryState(jobName string) MsgJobStatus { func (scheduler *SchedulerFair) QueryState(jobName string) MsgJobStatus {