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

fix some concurrent problems

Browse Source
This commit is contained in:
Newnius
2020-04-13 20:49:26 +08:00
parent 490ce3f26a
commit 2705e12c52
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/scheduler_fair.go
View File

@@ -313,6 +313,7 @@ func (scheduler *SchedulerFair) Summary() MsgSummary {
FreeGPU := 0 FreeGPU := 0
UsingGPU := 0 UsingGPU := 0
pool.mu.Lock()
for _, node := range pool.nodes { for _, node := range pool.nodes {
for j := range node.Status { for j := range node.Status {
if node.Status[j].MemoryAllocated == 0 { if node.Status[j].MemoryAllocated == 0 {
@@ -322,6 +323,7 @@ func (scheduler *SchedulerFair) Summary() MsgSummary {
} }
} }
} }
pool.mu.Unlock()
summary.FreeGPU = FreeGPU summary.FreeGPU = FreeGPU
summary.UsingGPU = UsingGPU summary.UsingGPU = UsingGPU
@@ -344,6 +346,7 @@ func (scheduler *SchedulerFair) UpdateNextQueue() {
MemoryGPU := 0.00001 MemoryGPU := 0.00001
CPU := 0.00001 CPU := 0.00001
Memory := 0.0001 Memory := 0.0001
pool.mu.Lock()
for _, node := range pool.nodes { for _, node := range pool.nodes {
CPU += float64(node.NumCPU) CPU += float64(node.NumCPU)
Memory += float64(node.MemTotal) Memory += float64(node.MemTotal)
@@ -352,6 +355,7 @@ func (scheduler *SchedulerFair) UpdateNextQueue() {
MemoryGPU += float64(card.MemoryTotal) MemoryGPU += float64(card.MemoryTotal)
} }
} }
pool.mu.Unlock()
for k, t := range scheduler.queues { for k, t := range scheduler.queues {
if len(t) == 0 { if len(t) == 0 {