1
0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-07 06:11:56 +00:00

update fair

This commit is contained in:
Newnius 2020-05-31 11:14:35 +08:00
parent 14d33bb67c
commit 2d1e2d84ae

View File

@ -83,12 +83,19 @@ func (scheduler *SchedulerFair) Start() {
/* drf of yarn/kube-batch */
if scheduler.drfyarn {
least := math.MaxInt32
for queue, allocate := range scheduler.resourceAllocations {
if jobs, ok := scheduler.queues[queue]; ok && len(jobs) > 0 {
for queue, jobs := range scheduler.queues {
if len(jobs) == 0 {
continue
}
if allocate, ok := scheduler.resourceAllocations[queue]; ok {
if bestQueue == "" || allocate.NumberGPU < least {
bestQueue = queue
least = allocate.NumberGPU
}
} else {
bestQueue = queue
least = 0
break
}
}
}