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 11:11:01 +08:00
parent 7d7d42eeb8
commit 2cc71f30f2
1 changed files with 8 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/scheduler_fair.go
View File

@ -129,6 +129,14 @@ func (scheduler *SchedulerFair) Start() {
} }
log.Info("After, ", scheduler.queuesQuota[bestQueue]) log.Info("After, ", scheduler.queuesQuota[bestQueue])
scheduler.resourceAllocationsMu.Lock()
if _, ok := scheduler.resourceAllocations[bestQueue]; !ok {
scheduler.resourceAllocations[bestQueue] = &ResourceCount{}
}
cnt, _ := scheduler.resourceAllocations[bestQueue]
cnt.NumberGPU += numberGPUtmp
scheduler.resourceAllocationsMu.Unlock()
scheduler.allocatingGPUMu.Lock() scheduler.allocatingGPUMu.Lock()
scheduler.allocatingGPU += numberGPUtmp scheduler.allocatingGPU += numberGPUtmp
scheduler.allocatingGPUMu.Unlock() scheduler.allocatingGPUMu.Unlock()
@ -258,24 +266,6 @@ func (scheduler *SchedulerFair) AcquireResource(job Job) []NodeStatus {
scheduler.allocatingGPUMu.Unlock() scheduler.allocatingGPUMu.Unlock()
} }
log.Info("allocatingGPU is ", scheduler.allocatingGPU) log.Info("allocatingGPU is ", scheduler.allocatingGPU)
go func(nodes []NodeStatus) {
for _, node := range nodes {
scheduler.resourceAllocationsMu.Lock()
if _, ok := scheduler.resourceAllocations[job.Group]; !ok {
scheduler.resourceAllocations[job.Group] = &ResourceCount{}
}
cnt, _ := scheduler.resourceAllocations[job.Group]
cnt.CPU += node.MemTotal
cnt.Memory += node.NumCPU
for _, v := range node.Status {
cnt.NumberGPU ++
cnt.MemoryGPU += v.MemoryTotal
}
scheduler.resourceAllocationsMu.Unlock()
}
}(res)
} }
go func() { go func() {
scheduler.UpdateQuota() scheduler.UpdateQuota()