From 98cc8365ecc73b6a37c1a7b37de91bf614b0c72b Mon Sep 17 00:00:00 2001
From: Newnius <newnius.cn@gmail.com>
Date: Sun, 3 May 2020 11:54:19 +0800
Subject: [PATCH] update

---
 src/scheduler_fair.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/scheduler_fair.go b/src/scheduler_fair.go
index e1622dc..e62cf51 100644
--- a/src/scheduler_fair.go
+++ b/src/scheduler_fair.go
@@ -152,6 +152,10 @@ func (scheduler *SchedulerFair) Start() {
 				scheduler.history = append(scheduler.history, &jm.job)
 				scheduler.historyMu.Unlock()
 
+				scheduler.queuesUsingGPUMu.Lock()
+				scheduler.queuesSchedulingCnt[jm.job.Group]++
+				scheduler.queuesUsingGPUMu.Unlock()
+
 				go func() {
 					jm.start()
 				}()
@@ -252,6 +256,10 @@ func (scheduler *SchedulerFair) UpdateProgress(job Job, state State) {
 		scheduler.queuesUsingGPUMu.Lock()
 		if _, ok := scheduler.queuesSchedulingCnt[job.Group]; ok {
 			scheduler.queuesSchedulingCnt[job.Group]--
+			if scheduler.queuesSchedulingCnt[job.Group] < 0 {
+				scheduler.queuesSchedulingCnt[job.Group] = 0
+				log.Warn("scheduler.queuesSchedulingCnt less than 0", job.Group)
+			}
 		}
 		scheduler.queuesUsingGPUMu.Unlock()