1
0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-07 14:21:55 +00:00
This commit is contained in:
Newnius 2020-06-30 20:43:42 +08:00
parent 7180d6d5e4
commit 526a9d8b4c

View File

@ -769,10 +769,9 @@ func (pool *ResourcePool) doAcquireResource(job Job) []NodeStatus {
utilT := InstanceOfOptimizer().PredictReq(job, "Worker").UtilGPU utilT := InstanceOfOptimizer().PredictReq(job, "Worker").UtilGPU
totalUtil += utilT totalUtil += utilT
} }
if totalUtil < 250 { if totalUtil < 100 {
available = append(available, status) available = append(available, status)
} }
log.Info(pred.UtilGPU, totalUtil)
} }
} }
} }
@ -860,6 +859,7 @@ func (pool *ResourcePool) doAcquireResource(job Job) []NodeStatus {
/* third round, find gpu to be released */ /* third round, find gpu to be released */
if len(candidates) == 0 && len(job.Tasks) == 1 && task.NumberGPU == 1 && pool.enablePreSchedule { if len(candidates) == 0 && len(job.Tasks) == 1 && task.NumberGPU == 1 && pool.enablePreSchedule {
estimate := InstanceOfOptimizer().PredictTime(job) estimate := InstanceOfOptimizer().PredictTime(job)
log.Info(estimate)
if loadRatio >= pool.enablePreScheduleRatio { if loadRatio >= pool.enablePreScheduleRatio {
allocationType = 3 allocationType = 3