YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-06 05:51:54 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius 2020-07-01 11:44:09 +08:00
parent e5109db709
commit f2768984ae
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/job_manager.go
View File

@ -67,6 +67,8 @@ func (jm *JobManager) start() {
return
}
isShare := false
isScheduleAhead := false
if !jm.killFlag {
/* switch to Running state */
jm.scheduler.UpdateProgress(jm.job, Running)
@ -85,6 +87,9 @@ func (jm *JobManager) start() {
UUIDs = append(UUIDs, GPU.UUID)
if GPU.MemoryUsed == GPU.MemoryTotal {
shouldWait = "1"
isScheduleAhead = true
} else if GPU.MemoryUsed > 0 {
isShare = true
}
/* attach to GPUs */
InstanceOfResourcePool().attach(GPU.UUID, jm.job)
@ -174,7 +179,7 @@ func (jm *JobManager) start() {
}
InstanceOfOptimizer().FeedStats(jm.job, "Worker", stats)
if len(jm.job.Tasks) == 1 {
if len(jm.job.Tasks) == 1 && !isShare && !isScheduleAhead {
InstanceOfOptimizer().FeedTime(jm.job, stats)
}
log.Info("JobMaster exited ", jm.job.Name)