spdf/elektron
Archived
1
0
Fork 0
Code Issues 4 Pull requests 1 Projects Releases Packages Wiki Activity

Used the KMeans classification to classify the tasks.

Browse source
This commit is contained in:
Pradyumna Kaushik 2017-03-13 16:38:53 -04:00
parent 668ae586fe
commit 34ab753491
2 changed files with 38 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

23
schedulers/bottomHeavy.go
View file

@ -11,7 +11,6 @@ import (
"github.com/mesos/mesos-go/mesosutil"
sched "github.com/mesos/mesos-go/scheduler"
"log"
"math"
"os"
"sort"
"time"
@ -68,7 +67,23 @@ func NewBottomHeavy(tasks []def.Task, wattsAsAResource bool, schedTracePrefix st
// Separating small tasks from large tasks.
// Classification done based on MMPU watts requirements.
mid := int(math.Floor((float64(len(tasks)) / 2.0) + 0.5))
tasksToClassify := def.TasksToClassify(tasks)
classifiedTasks := tasksToClassify.ClassifyTasks(2, func(task def.Task) float64 {
if task.ClassToWatts != nil {
// taking the aggregate
observation := 0.0
for _, watts := range task.ClassToWatts {
observation += watts
}
return observation
} else if task.Watts != 0.0 {
return task.Watts
} else {
log.Fatal("Unable to classify tasks. Missing Watts or ClassToWatts attribute in workload.")
return 0.0 // won't reach here
}
})
s := &BottomHeavy{
base: base{
wattsAsAResource: wattsAsAResource,
@ -80,8 +95,8 @@ func NewBottomHeavy(tasks []def.Task, wattsAsAResource bool, schedTracePrefix st
RecordPCP: false,
schedTrace: log.New(logFile, "", log.LstdFlags),
},
smallTasks: tasks[:mid],
largeTasks: tasks[mid+1:],
smallTasks: classifiedTasks[0].Tasks,
largeTasks: classifiedTasks[1].Tasks,
}
return s
}

23
schedulers/topHeavy.go
View file

@ -11,7 +11,6 @@ import (
"github.com/mesos/mesos-go/mesosutil"
sched "github.com/mesos/mesos-go/scheduler"
"log"
"math"
"os"
"sort"
"time"
@ -67,7 +66,23 @@ func NewTopHeavy(tasks []def.Task, wattsAsAResource bool, schedTracePrefix strin
// Separating small tasks from large tasks.
// Classification done based on MMPU watts requirements.
mid := int(math.Floor((float64(len(tasks)) / 2.0) + 0.5))
tasksToClassify := def.TasksToClassify(tasks)
classifiedTasks := tasksToClassify.ClassifyTasks(2, func(task def.Task) float64 {
if task.ClassToWatts != nil {
// taking the aggregate
observation := 0.0
for _, watts := range task.ClassToWatts {
observation += watts
}
return observation
} else if task.Watts != 0.0 {
return task.Watts
} else {
log.Fatal("Unable to classify tasks. Missing Watts or ClassToWatts attribute in workload.")
return 0.0
}
})
s := &TopHeavy{
base: base{
wattsAsAResource: wattsAsAResource,
@ -79,8 +94,8 @@ func NewTopHeavy(tasks []def.Task, wattsAsAResource bool, schedTracePrefix strin
RecordPCP: false,
schedTrace: log.New(logFile, "", log.LstdFlags),
},
smallTasks: tasks[:mid],
largeTasks: tasks[mid+1:],
smallTasks: classifiedTasks[0].Tasks,
largeTasks: classifiedTasks[1].Tasks,
}
return s
}