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

update

Browse Source
This commit is contained in:
Newnius 2020-04-30 18:50:50 +08:00
parent ff0c10647b
commit 713fbdc4e7
3 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
README.md
View File

@ -49,3 +49,15 @@ GPU is occupied by which job(s)
``` ```
?action=debug_get_gpu_utils ?action=debug_get_gpu_utils
``` ```
**SetShareRatio**
```
?action=debug_update_enable_share_ratio&ratio=0.75
```
**SetPreScheduleRatio**
```
?action=debug_update_enable_pre_schedule_ratio&ratio=0.95
```

2
src/optimizer.go
View File

@ -108,7 +108,7 @@ func (optimizer *Optimizer) predictTime(job string) (*OptimizerJobExecutionTime,
if len(str) == 2 { if len(str) == 2 {
jobName := str[0] jobName := str[0]
if _, ok := optimizer.predicts[jobName]; ok { if _, ok := optimizer.predicts[jobName]; ok {
return optimizer.predicts[job], optimizer.predicts[jobName].Version >= 5 return optimizer.predicts[jobName], optimizer.predicts[jobName].Version >= 5
} }
} }
return &OptimizerJobExecutionTime{}, false return &OptimizerJobExecutionTime{}, false

5
src/scheduler_fair.go
View File

@ -281,6 +281,11 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task) 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 && scheduler.enablePreSchedule { if len(candidates) == 0 && len(job.Tasks) == 1 && task.NumberGPU == 1 && scheduler.enablePreSchedule {
estimate, valid := InstanceOfOptimizer().predictTime(job.Name) estimate, valid := InstanceOfOptimizer().predictTime(job.Name)
log.Info(pool.TotalGPU)
log.Info(estimate, valid)
log.Info(scheduler.UsingGPU)
if pool.TotalGPU != 0 && float64(scheduler.UsingGPU)/float64(pool.TotalGPU) >= scheduler.enablePreScheduleRatio && valid { if pool.TotalGPU != 0 && float64(scheduler.UsingGPU)/float64(pool.TotalGPU) >= scheduler.enablePreScheduleRatio && valid {
allocationType = 3 allocationType = 3
for i := 0; i < pool.poolsCount; i++ { for i := 0; i < pool.poolsCount; i++ {