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

update

Browse Source
This commit is contained in:
Newnius
2020-05-04 00:45:50 +08:00
parent be2840f861
commit 33894f4eac
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/scheduler_fair.go
View File

@@ -338,7 +338,7 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node
start = start.Next start = start.Next
} }
locks := map[int]sync.Mutex{} locks := map[int]*sync.Mutex{}
allocationType := 0 allocationType := 0
availableGPUs := map[string][]GPUStatus{} availableGPUs := map[string][]GPUStatus{}
@@ -355,7 +355,7 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node
if _, ok := locks[cur.ID]; !ok { if _, ok := locks[cur.ID]; !ok {
log.Info("lock ", cur.ID) log.Info("lock ", cur.ID)
cur.Lock.Lock() cur.Lock.Lock()
locks[cur.ID] = cur.Lock locks[cur.ID] = &cur.Lock
} }
for _, node := range cur.Nodes { for _, node := range cur.Nodes {
@@ -405,7 +405,7 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node
if _, ok := locks[cur.ID]; !ok { if _, ok := locks[cur.ID]; !ok {
log.Info("lock ", cur.ID) log.Info("lock ", cur.ID)
cur.Lock.Lock() cur.Lock.Lock()
locks[cur.ID] = cur.Lock locks[cur.ID] = &cur.Lock
} }
for _, node := range cur.Nodes { for _, node := range cur.Nodes {
var available []GPUStatus var available []GPUStatus
@@ -447,7 +447,7 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node
if _, ok := locks[cur.ID]; !ok { if _, ok := locks[cur.ID]; !ok {
log.Info("lock ", cur.ID) log.Info("lock ", cur.ID)
cur.Lock.Lock() cur.Lock.Lock()
locks[cur.ID] = cur.Lock locks[cur.ID] = &cur.Lock
} }
for _, node := range cur.Nodes { for _, node := range cur.Nodes {
var available []GPUStatus var available []GPUStatus