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

update fair

Browse Source
This commit is contained in:
Newnius 2020-05-28 10:34:55 +08:00
parent 0e22b84471
commit d358f9b1ef
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/scheduler_fair.go
View File

@ -16,7 +16,7 @@ type SchedulerFair struct {
queues map[string][]Job queues map[string][]Job
queuesMu sync.Mutex queuesMu sync.Mutex
queuesQuota map[string]ResourceCount queuesQuota map[string]*ResourceCount
queuesQuotaMu sync.Mutex queuesQuotaMu sync.Mutex
schedulingJobs map[string]bool schedulingJobs map[string]bool
@ -43,7 +43,7 @@ func (scheduler *SchedulerFair) Start() {
scheduler.history = []*Job{} scheduler.history = []*Job{}
scheduler.queues = map[string][]Job{} scheduler.queues = map[string][]Job{}
scheduler.queues["default"] = []Job{} scheduler.queues["default"] = []Job{}
scheduler.queuesQuota = map[string]ResourceCount{} scheduler.queuesQuota = map[string]*ResourceCount{}
scheduler.resourceAllocations = map[string]*ResourceCount{} scheduler.resourceAllocations = map[string]*ResourceCount{}
scheduler.enabled = true scheduler.enabled = true
scheduler.schedulingJobs = map[string]bool{} scheduler.schedulingJobs = map[string]bool{}
@ -324,12 +324,11 @@ func (scheduler *SchedulerFair) UpdateQuota() {
per := available / len(scheduler.queues) per := available / len(scheduler.queues)
for queue := range scheduler.queues { for queue := range scheduler.queues {
if _, ok := scheduler.queuesQuota[queue]; !ok { if _, ok := scheduler.queuesQuota[queue]; !ok {
scheduler.queuesQuota[queue] = ResourceCount{} scheduler.queuesQuota[queue] = &ResourceCount{}
} }
quota := scheduler.queuesQuota[queue] quota := scheduler.queuesQuota[queue]
quota.NumberGPU += per quota.NumberGPU += per
available -= per available -= per
log.Info(quota)
} }
if available > 0 { if available > 0 {
for queue := range scheduler.queues { for queue := range scheduler.queues {