YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-12-13 07:46:43 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius
2020-05-03 11:54:19 +08:00
parent 87bb4a9b39
commit 98cc8365ec
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/scheduler_fair.go
View File

@@ -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()