From 2d1e2d84ae8cf1c18d72d88666e30810b14e9fb9 Mon Sep 17 00:00:00 2001
From: Newnius <newnius.cn@gmail.com>
Date: Sun, 31 May 2020 11:14:35 +0800
Subject: [PATCH] update fair

---
 src/scheduler_fair.go | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/scheduler_fair.go b/src/scheduler_fair.go
index d496148..98ae7de 100644
--- a/src/scheduler_fair.go
+++ b/src/scheduler_fair.go
@@ -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
 					}
 				}
 			}