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

update

Browse Source
This commit is contained in:
Newnius
2020-06-24 23:27:04 +08:00
parent 81d5f410cb
commit 0feccd0a51
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/optimizer.go
View File

@@ -391,7 +391,12 @@ func (optimizer *Optimizer) predict(job string, seq int) (OptimizerJobExecutionT
} }
func (optimizer *Optimizer) PredictReq(job Job, role string) MsgJobReq { func (optimizer *Optimizer) PredictReq(job Job, role string) MsgJobReq {
cmd := job.Tasks[0].Cmd var jobName string
str := strings.Split(job.Name, "-")
if len(str) == 2 {
jobName = str[0]
}
cmd := ""
params := map[string]int{} params := map[string]int{}
psNumber := 0 psNumber := 0
@@ -443,7 +448,7 @@ func (optimizer *Optimizer) PredictReq(job Job, role string) MsgJobReq {
spider := Spider{} spider := Spider{}
spider.Method = "GET" spider.Method = "GET"
spider.URL = "http://yao-optimizer:8080/predict?job=" + job.Name + "&features=" + string(features) spider.URL = "http://yao-optimizer:8080/predict?job=" + jobName + "&features=" + string(features)
err := spider.do() err := spider.do()
if err != nil { if err != nil {
@@ -474,6 +479,9 @@ func (optimizer *Optimizer) PredictReq(job Job, role string) MsgJobReq {
if v, ok := tmp["gpu_mem"]; ok { if v, ok := tmp["gpu_mem"]; ok {
req.MemGPU = int(math.Ceil(v/1024)) * 1024 req.MemGPU = int(math.Ceil(v/1024)) * 1024
} }
if v, ok := tmp["bw"]; ok {
req.BW = int(math.Ceil(v/10)) * 10
}
} }
return req return req
} }