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 13:08:08 +08:00
parent 8eb492964e
commit 913eb47a0d
3 changed files with 42 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/optimizer.go
View File

@@ -11,6 +11,9 @@ type Optimizer struct {
killedFlag bool
predicts map[string]OptimizerJobExecutionTime
jobUtilsGPU map[string]int
versions map[string]int
}
var optimizerInstance *Optimizer
@@ -23,6 +26,8 @@ func InstanceOfOptimizer() *Optimizer {
if optimizerInstance == nil {
optimizerInstance = &Optimizer{}
optimizerInstance.predicts = map[string]OptimizerJobExecutionTime{}
optimizerInstance.jobUtilsGPU = map[string]int{}
optimizerInstance.versions = map[string]int{}
}
return optimizerInstance
}
@@ -30,11 +35,25 @@ func InstanceOfOptimizer() *Optimizer {
func (optimizer *Optimizer) feed(job string, utils []int) {
log.Info("optimizer feed")
log.Info(job, utils)
log.Info(optimizer.jobUtilsGPU)
log.Info(optimizer.predicts)
go func() {
str := strings.Split(job, "-")
if len(str) == 2 {
preCnt := 0
sum := 0
for i := 0; i < len(utils); i++ {
sum += utils[i]
}
last := 0
if t, err := optimizer.jobUtilsGPU[job]; !err {
last = t
}
optimizer.jobUtilsGPU[job] = (optimizer.versions[job]*last + sum/len(utils)) / (optimizer.versions[job] + 1)
optimizer.versions[job]++
for i := 0; i < len(utils); i++ {
if utils[i] > 15 {
break
@@ -43,7 +62,7 @@ func (optimizer *Optimizer) feed(job string, utils []int) {
}
postCnt := 0
for i := len(utils)-1; i >= 0; i-- {
for i := len(utils) - 1; i >= 0; i-- {
if utils[i] > 15 {
break
}
@@ -57,7 +76,25 @@ func (optimizer *Optimizer) feed(job string, utils []int) {
predict.Pre = ((predict.Pre * predict.Version) + preCnt) / (predict.Version + 1)
predict.Post = ((predict.Post * predict.Version) + postCnt) / (predict.Version + 1)
predict.Total = ((predict.Total * predict.Version) + len(utils)) / (predict.Version + 1)
predict.Main = predict.Total - predict.Pre - predict.Post
predict.Version++
}
}()
}
func (optimizer *Optimizer) predictTime(job string, utils []int) (int, bool) {
if _, err := optimizer.jobUtilsGPU[job]; err {
return 100, false
}
if optimizer.versions[job] > 5 {
return optimizer.jobUtilsGPU[job], true
}
return optimizer.jobUtilsGPU[job], false
}
func (optimizer *Optimizer) predictUtilGPU(job string) (OptimizerJobExecutionTime, bool) {
if _, err := optimizer.predicts[job]; err {
return OptimizerJobExecutionTime{}, false
}
return optimizer.predicts[job], optimizer.predicts[job].Version > 5
}