From 33894f4eac8cc94c03b8cfa4aa40fdd144651034 Mon Sep 17 00:00:00 2001 From: Newnius Date: Mon, 4 May 2020 00:45:50 +0800 Subject: [PATCH] update --- src/scheduler_fair.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/scheduler_fair.go b/src/scheduler_fair.go index 66abd46..2014817 100644 --- a/src/scheduler_fair.go +++ b/src/scheduler_fair.go @@ -338,7 +338,7 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node start = start.Next } - locks := map[int]sync.Mutex{} + locks := map[int]*sync.Mutex{} allocationType := 0 availableGPUs := map[string][]GPUStatus{} @@ -355,7 +355,7 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node if _, ok := locks[cur.ID]; !ok { log.Info("lock ", cur.ID) cur.Lock.Lock() - locks[cur.ID] = cur.Lock + locks[cur.ID] = &cur.Lock } 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 { log.Info("lock ", cur.ID) cur.Lock.Lock() - locks[cur.ID] = cur.Lock + locks[cur.ID] = &cur.Lock } for _, node := range cur.Nodes { var available []GPUStatus @@ -447,7 +447,7 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node if _, ok := locks[cur.ID]; !ok { log.Info("lock ", cur.ID) cur.Lock.Lock() - locks[cur.ID] = cur.Lock + locks[cur.ID] = &cur.Lock } for _, node := range cur.Nodes { var available []GPUStatus