1
0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-07 14:21:55 +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 */ /* drf of yarn/kube-batch */
if scheduler.drfyarn { if scheduler.drfyarn {
least := math.MaxInt32 least := math.MaxInt32
for queue, allocate := range scheduler.resourceAllocations { for queue, jobs := range scheduler.queues {
if jobs, ok := scheduler.queues[queue]; ok && len(jobs) > 0 { if len(jobs) == 0 {
continue
}
if allocate, ok := scheduler.resourceAllocations[queue]; ok {
if bestQueue == "" || allocate.NumberGPU < least { if bestQueue == "" || allocate.NumberGPU < least {
bestQueue = queue bestQueue = queue
least = allocate.NumberGPU least = allocate.NumberGPU
} }
} else {
bestQueue = queue
least = 0
break
} }
} }
} }