YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-07 14:21:55 +00:00
Code Issues Releases Wiki Activity

update fair

Browse Source
This commit is contained in:
Newnius 2020-05-28 11:23:27 +08:00
parent d16df29b50
commit c044d43490
3 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/job_manager.go
View File

@ -197,19 +197,22 @@ func (jm *JobManager) checkStatus(status []TaskStatus) {
jm.scheduler.UpdateProgress(jm.job, Failed)
}
jm.resourcesMu.Lock()
if jm.resources[i].ClientID != "_released_" {
jm.scheduler.ReleaseResource(jm.job, jm.resources[i])
log.Info("return resource ", jm.resources[i].ClientID)
jm.resources[i].ClientID = "_released_"
}
for _, t := range jm.resources[i].Status {
InstanceOfResourcePool().detach(t.UUID, jm.job)
}
jm.resourcesMu.Unlock()
InstanceJobHistoryLogger().submitTaskStatus(jm.job.Name, status[i])
}
}
}
if flagRunning && onlyPS && !jm.killFlag {
log.Info("Only PS is running, stop ", jm.job.Name)
jm.stop(false)

2
src/scheduler_fair.go
View File

@ -320,7 +320,7 @@ func (scheduler *SchedulerFair) UpdateQuota() {
pool := InstanceOfResourcePool()
available := pool.TotalGPU - usingGPU - allocatedGPU
if available == 0 {
if available <= 0 {
return
}
log.Info("Can allocate ", available)