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

update

Browse Source
This commit is contained in:
Newnius
2020-04-30 23:06:12 +08:00
parent 1447192a45
commit e880c21e82
4 changed files with 50 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/optimizer.go
View File

@@ -33,7 +33,7 @@ func InstanceOfOptimizer() *Optimizer {
return optimizerInstance
}
func (optimizer *Optimizer) feed(job string, utils []int) {
func (optimizer *Optimizer) feed(job string, utils []UtilGPUTimeSeries) {
log.Info("optimizer feed")
log.Info(job, utils)
@@ -44,12 +44,11 @@ func (optimizer *Optimizer) feed(job string, utils []int) {
go func() {
str := strings.Split(job, "-")
if len(str) == 2 {
preCnt := 0
jobName := str[0]
sum := 0
for i := 0; i < len(utils); i++ {
sum += utils[i]
sum += utils[i].Util
}
sum /= len(utils)
if _, ok := optimizer.jobUtilsGPU[jobName]; !ok {
@@ -59,27 +58,26 @@ func (optimizer *Optimizer) feed(job string, utils []int) {
t.Util = (t.Version*t.Util + sum) / (t.Version + 1)
t.Version++
preTime := 0
for i := 0; i < len(utils); i++ {
if utils[i] > 15 {
if utils[i].Util > 15 {
preTime = utils[i].Time - utils[0].Time
break
}
preCnt++
}
postCnt := 0
postTime := 0
for i := len(utils) - 1; i >= 0; i-- {
if utils[i] > 15 {
if utils[i].Util > 15 {
postTime = utils[len(utils)-1].Time - utils[i].Time
break
}
postCnt++
}
if _, ok := optimizer.predicts[jobName]; !ok {
optimizer.predicts[jobName] = &OptimizerJobExecutionTime{}
}
postTime := postCnt * optimizer.heartbeatInterval
preTime := preCnt * optimizer.heartbeatInterval
totalTime := len(utils) * optimizer.heartbeatInterval
totalTime := utils[len(utils)-1].Time - utils[0].Time
predict := optimizer.predicts[jobName]
predict.Pre = ((predict.Pre * predict.Version) + preTime) / (predict.Version + 1)