YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-12-12 23:36:44 +00:00
Code Issues Releases Wiki Activity

update shceduler_fair, bugfix

Browse Source
This commit is contained in:
Newnius
2019-08-01 14:16:44 +08:00
parent cfad24d26c
commit 9d3489ae91
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/job_manager.go
View File

@@ -32,6 +32,7 @@ func (jm *JobManager) start() {
if len(resource.Status) > 0 {
break
}
time.Sleep(time.Second * 1)
}
log.Info("Receive resource", resource)
jm.resources = append(jm.resources, resource)

9
src/scheduler_fair.go
View File

@@ -179,6 +179,9 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task) NodeStatus {
}
}
go func(res NodeStatus) {
if len(res.Status) == 0 {
return
}
if _, ok := scheduler.resourceAllocations[job.Group]; !ok {
scheduler.resourceAllocations[job.Group] = &ResourceCount{}
}
@@ -189,10 +192,8 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task) NodeStatus {
cnt.NumberGPU ++
cnt.MemoryGPU += v.MemoryTotal
}
log.Info(job.Group, cnt)
if len(res.Status) != 0 {
scheduler.UpdateNextQueue()
}
scheduler.UpdateNextQueue()
}(res)
return res
}