YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-07 22:31:55 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius 2020-06-30 16:37:14 +08:00
parent e6c9891460
commit f75b079cb4
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/scheduler_fair.go
View File

@ -271,6 +271,19 @@ func (scheduler *SchedulerFair) Start() {
} }
} }
/* support schedule ahead & share */
if bestQueue != "" && len(scheduler.schedulingJobs) == 0 {
maxQuota := 0
for queue, jobs := range scheduler.queues {
if len(jobs) > 0 && len(jobs[0].Tasks) == 1 && jobs[0].Tasks[0].NumberGPU == 1 {
if quota, ok := scheduler.queuesQuota[queue]; ok && (bestQueue == "" || quota.NumberGPU > maxQuota) {
maxQuota = quota.NumberGPU
bestQueue = queue
}
}
}
}
/* launch that job */ /* launch that job */
if bestQueue != "" { if bestQueue != "" {
numberGPUtmp := 0 numberGPUtmp := 0