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:19:51 +08:00
parent 0751608eb2
commit a4c75a7065
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/scheduler_fair.go
View File

@@ -129,6 +129,7 @@ func (scheduler *SchedulerFair) Start() {
} else { } else {
cnt *= 13 cnt *= 13
} }
log.Info(cnt, reserved, pool.TotalGPU, scheduler.UsingGPU)
if cnt+(scheduler.allocatingGPU)*13 > (pool.TotalGPU-scheduler.UsingGPU-reserved)*10 { if cnt+(scheduler.allocatingGPU)*13 > (pool.TotalGPU-scheduler.UsingGPU-reserved)*10 {
scheduler.schedulingMu.Lock() scheduler.schedulingMu.Lock()
scheduler.schedulingJobsCnt-- scheduler.schedulingJobsCnt--
@@ -178,6 +179,7 @@ func (scheduler *SchedulerFair) Start() {
if len(t) == 0 || !InstanceOfGroupManager().groups[t[0].Group].Reserved { if len(t) == 0 || !InstanceOfGroupManager().groups[t[0].Group].Reserved {
continue continue
} }
log.Info(scheduler.queueUsingGPU)
scheduler.queuesUsingGPUMu.Lock() scheduler.queuesUsingGPUMu.Lock()
if cnt, ok := scheduler.queuesSchedulingCnt[t[0].Group]; ok && cnt > 0 { if cnt, ok := scheduler.queuesSchedulingCnt[t[0].Group]; ok && cnt > 0 {
scheduler.queuesUsingGPUMu.Unlock() scheduler.queuesUsingGPUMu.Unlock()
@@ -748,7 +750,7 @@ func (scheduler *SchedulerFair) UpdateNextQueue() {
} }
scheduler.nextQueue = next scheduler.nextQueue = next
scheduler.queueMu.Unlock() scheduler.queueMu.Unlock()
log.Debug("updateNextQueue ->", next) log.Info("updateNextQueue ->", next)
} }
func (scheduler *SchedulerFair) Attach(GPU string, job string) { func (scheduler *SchedulerFair) Attach(GPU string, job string) {