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

update

Browse Source
This commit is contained in:
Newnius
2020-04-30 21:26:04 +08:00
parent 014592fa43
commit cbaef3bd59
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/scheduler_fair.go
View File

@@ -34,7 +34,7 @@ type SchedulerFair struct {
enablePreSchedule bool enablePreSchedule bool
enablePreScheduleRatio float64 enablePreScheduleRatio float64
UsingGPU int UsingGPU int
UsingGPUMu sync.Mutex UsingGPUMu sync.Mutex
} }
@@ -309,6 +309,7 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node
if valid2 { if valid2 {
t := s t := s
now := (int)(time.Now().Unix()) now := (int)(time.Now().Unix())
log.Info(t, now, estimate, est)
if now-t > est.Total-est.Post-estimate.Pre && status.MemoryFree > task.MemoryGPU { if now-t > est.Total-est.Post-estimate.Pre && status.MemoryFree > task.MemoryGPU {
available = append(available, status) available = append(available, status)
} }
@@ -362,9 +363,6 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []Node
} }
} }
for i := range locks { for i := range locks {
pool.poolsMu[i].Unlock() pool.poolsMu[i].Unlock()
} }