Skip to content

Commit

Permalink
Merge pull request volcano-sh#3954 from bogo-y/boostnuma
Browse files Browse the repository at this point in the history
vc-scheduler: optimize bachNodeOrderFn in numaaware plugin
  • Loading branch information
volcano-sh-bot authored Jan 6, 2025
2 parents bd915ad + a52c7d5 commit 6fa2813
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions pkg/scheduler/plugins/numaaware/numaaware.go
Original file line number Diff line number Diff line change
Expand Up @@ -169,15 +169,11 @@ func (pp *numaPlugin) OnSessionOpen(ssn *framework.Session) {
ssn.AddPredicateFn(pp.Name(), predicateFn)

batchNodeOrderFn := func(task *api.TaskInfo, nodeInfo []*api.NodeInfo) (map[string]float64, error) {
nodeScores := make(map[string]float64, len(nodeInfo))
if task.NumaInfo == nil || task.NumaInfo.Policy == "" || task.NumaInfo.Policy == "none" {
return nodeScores, nil
}

if _, found := pp.assignRes[task.UID]; !found {
return nodeScores, nil
if _, found := pp.assignRes[task.UID]; !found || task.NumaInfo == nil || task.NumaInfo.Policy == "" || task.NumaInfo.Policy == "none" {
return nil, nil
}

nodeScores := make(map[string]float64, len(nodeInfo))
scoreList := getNodeNumaNumForTask(nodeInfo, pp.assignRes[task.UID])
util.NormalizeScore(api.DefaultMaxNodeScore, true, scoreList)

Expand Down

0 comments on commit 6fa2813

Please sign in to comment.