mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-24 12:15:52 +00:00
Add mergeFilteredHints:
- Move remaining logic from mergeProvidersHints to generic top level mergeFilteredHints function. - Add numaNodes as parameter in order to make generic. - Move single NUMA node specific check to single-numa-node Merge function.
This commit is contained in:
parent
fc300e0e7d
commit
4d76b1c8de
@ -93,6 +93,53 @@ func filterProvidersHints(providersHints []map[string][]TopologyHint) [][]Topolo
|
|||||||
return allProviderHints
|
return allProviderHints
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func mergeFilteredHints(numaNodes []int, filteredHints [][]TopologyHint) TopologyHint {
|
||||||
|
// Set the default affinity as an any-numa affinity containing the list
|
||||||
|
// of NUMA Nodes available on this machine.
|
||||||
|
defaultAffinity, _ := bitmask.NewBitMask(numaNodes...)
|
||||||
|
|
||||||
|
// Set the bestHint to return from this function as {nil false}.
|
||||||
|
// This will only be returned if no better hint can be found when
|
||||||
|
// merging hints from each hint provider.
|
||||||
|
bestHint := TopologyHint{defaultAffinity, false}
|
||||||
|
iterateAllProviderTopologyHints(filteredHints, func(permutation []TopologyHint) {
|
||||||
|
// Get the NUMANodeAffinity from each hint in the permutation and see if any
|
||||||
|
// of them encode unpreferred allocations.
|
||||||
|
mergedHint := mergePermutation(numaNodes, permutation)
|
||||||
|
// Only consider mergedHints that result in a NUMANodeAffinity > 0 to
|
||||||
|
// replace the current bestHint.
|
||||||
|
if mergedHint.NUMANodeAffinity.Count() == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the current bestHint is non-preferred and the new mergedHint is
|
||||||
|
// preferred, always choose the preferred hint over the non-preferred one.
|
||||||
|
if mergedHint.Preferred && !bestHint.Preferred {
|
||||||
|
bestHint = mergedHint
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the current bestHint is preferred and the new mergedHint is
|
||||||
|
// non-preferred, never update bestHint, regardless of mergedHint's
|
||||||
|
// narowness.
|
||||||
|
if !mergedHint.Preferred && bestHint.Preferred {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// If mergedHint and bestHint has the same preference, only consider
|
||||||
|
// mergedHints that have a narrower NUMANodeAffinity than the
|
||||||
|
// NUMANodeAffinity in the current bestHint.
|
||||||
|
if !mergedHint.NUMANodeAffinity.IsNarrowerThan(bestHint.NUMANodeAffinity) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// In all other cases, update bestHint to the current mergedHint
|
||||||
|
bestHint = mergedHint
|
||||||
|
})
|
||||||
|
|
||||||
|
return bestHint
|
||||||
|
}
|
||||||
|
|
||||||
// Iterate over all permutations of hints in 'allProviderHints [][]TopologyHint'.
|
// Iterate over all permutations of hints in 'allProviderHints [][]TopologyHint'.
|
||||||
//
|
//
|
||||||
// This procedure is implemented as a recursive function over the set of hints
|
// This procedure is implemented as a recursive function over the set of hints
|
||||||
|
@ -17,7 +17,6 @@ limitations under the License.
|
|||||||
package topologymanager
|
package topologymanager
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"k8s.io/kubernetes/pkg/kubelet/cm/topologymanager/bitmask"
|
|
||||||
"k8s.io/kubernetes/pkg/kubelet/lifecycle"
|
"k8s.io/kubernetes/pkg/kubelet/lifecycle"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -48,58 +47,7 @@ func (p *bestEffortPolicy) canAdmitPodResult(hint *TopologyHint) lifecycle.PodAd
|
|||||||
|
|
||||||
func (p *bestEffortPolicy) Merge(providersHints []map[string][]TopologyHint) (TopologyHint, lifecycle.PodAdmitResult) {
|
func (p *bestEffortPolicy) Merge(providersHints []map[string][]TopologyHint) (TopologyHint, lifecycle.PodAdmitResult) {
|
||||||
filteredProvidersHints := filterProvidersHints(providersHints)
|
filteredProvidersHints := filterProvidersHints(providersHints)
|
||||||
hint := p.mergeProvidersHints(filteredProvidersHints)
|
bestHint := mergeFilteredHints(p.numaNodes, filteredProvidersHints)
|
||||||
admit := p.canAdmitPodResult(&hint)
|
admit := p.canAdmitPodResult(&bestHint)
|
||||||
return hint, admit
|
return bestHint, admit
|
||||||
}
|
|
||||||
|
|
||||||
// Merge the hints from all hint providers to find the best one.
|
|
||||||
func (p *bestEffortPolicy) mergeProvidersHints(filteredHints [][]TopologyHint) TopologyHint {
|
|
||||||
// Set the default affinity as an any-numa affinity containing the list
|
|
||||||
// of NUMA Nodes available on this machine.
|
|
||||||
defaultAffinity, _ := bitmask.NewBitMask(p.numaNodes...)
|
|
||||||
|
|
||||||
// Iterate over all permutations of hints in 'allProviderHints'. Merge the
|
|
||||||
// hints in each permutation by taking the bitwise-and of their affinity masks.
|
|
||||||
// Return the hint with the narrowest NUMANodeAffinity of all merged
|
|
||||||
// permutations that have at least one NUMA ID set. If no merged mask can be
|
|
||||||
// found that has at least one NUMA ID set, return the 'defaultAffinity'.
|
|
||||||
bestHint := TopologyHint{defaultAffinity, false}
|
|
||||||
iterateAllProviderTopologyHints(filteredHints, func(permutation []TopologyHint) {
|
|
||||||
// Get the NUMANodeAffinity from each hint in the permutation and see if any
|
|
||||||
// of them encode unpreferred allocations.
|
|
||||||
mergedHint := mergePermutation(p.numaNodes, permutation)
|
|
||||||
|
|
||||||
// Only consider mergedHints that result in a NUMANodeAffinity > 0 to
|
|
||||||
// replace the current bestHint.
|
|
||||||
if mergedHint.NUMANodeAffinity.Count() == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// If the current bestHint is non-preferred and the new mergedHint is
|
|
||||||
// preferred, always choose the preferred hint over the non-preferred one.
|
|
||||||
if mergedHint.Preferred && !bestHint.Preferred {
|
|
||||||
bestHint = mergedHint
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// If the current bestHint is preferred and the new mergedHint is
|
|
||||||
// non-preferred, never update bestHint, regardless of mergedHint's
|
|
||||||
// narowness.
|
|
||||||
if !mergedHint.Preferred && bestHint.Preferred {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// If mergedHint and bestHint has the same preference, only consider
|
|
||||||
// mergedHints that have a narrower NUMANodeAffinity than the
|
|
||||||
// NUMANodeAffinity in the current bestHint.
|
|
||||||
if !mergedHint.NUMANodeAffinity.IsNarrowerThan(bestHint.NUMANodeAffinity) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// In all other cases, update bestHint to the current mergedHint
|
|
||||||
bestHint = mergedHint
|
|
||||||
})
|
|
||||||
|
|
||||||
return bestHint
|
|
||||||
}
|
}
|
||||||
|
@ -53,7 +53,7 @@ func (p *restrictedPolicy) canAdmitPodResult(hint *TopologyHint) lifecycle.PodAd
|
|||||||
|
|
||||||
func (p *restrictedPolicy) Merge(providersHints []map[string][]TopologyHint) (TopologyHint, lifecycle.PodAdmitResult) {
|
func (p *restrictedPolicy) Merge(providersHints []map[string][]TopologyHint) (TopologyHint, lifecycle.PodAdmitResult) {
|
||||||
filteredHints := filterProvidersHints(providersHints)
|
filteredHints := filterProvidersHints(providersHints)
|
||||||
hint := p.mergeProvidersHints(filteredHints)
|
hint := mergeFilteredHints(p.numaNodes, filteredHints)
|
||||||
admit := p.canAdmitPodResult(&hint)
|
admit := p.canAdmitPodResult(&hint)
|
||||||
return hint, admit
|
return hint, admit
|
||||||
}
|
}
|
||||||
|
@ -71,63 +71,17 @@ func filterSingleNumaHints(allResourcesHints [][]TopologyHint) [][]TopologyHint
|
|||||||
return filteredResourcesHints
|
return filteredResourcesHints
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p *singleNumaNodePolicy) mergeProvidersHints(filteredHints [][]TopologyHint) TopologyHint {
|
|
||||||
// Set the default affinity as an any-numa affinity containing the list
|
|
||||||
// of NUMA Nodes available on this machine.
|
|
||||||
defaultAffinity, _ := bitmask.NewBitMask(p.numaNodes...)
|
|
||||||
|
|
||||||
// Set the bestHint to return from this function as {nil false}.
|
|
||||||
// This will only be returned if no better hint can be found when
|
|
||||||
// merging hints from each hint provider.
|
|
||||||
bestHint := TopologyHint{defaultAffinity, false}
|
|
||||||
iterateAllProviderTopologyHints(filteredHints, func(permutation []TopologyHint) {
|
|
||||||
// Get the NUMANodeAffinity from each hint in the permutation and see if any
|
|
||||||
// of them encode unpreferred allocations.
|
|
||||||
mergedHint := mergePermutation(p.numaNodes, permutation)
|
|
||||||
|
|
||||||
// Only consider mergedHints that result in a NUMANodeAffinity > 0 to
|
|
||||||
// replace the current bestHint.
|
|
||||||
if mergedHint.NUMANodeAffinity.Count() == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// If the current bestHint is non-preferred and the new mergedHint is
|
|
||||||
// preferred, always choose the preferred hint over the non-preferred one.
|
|
||||||
if mergedHint.Preferred && !bestHint.Preferred {
|
|
||||||
bestHint = mergedHint
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// If the current bestHint is preferred and the new mergedHint is
|
|
||||||
// non-preferred, never update bestHint, regardless of mergedHint's
|
|
||||||
// narowness.
|
|
||||||
if !mergedHint.Preferred && bestHint.Preferred {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// If mergedHint and bestHint has the same preference, only consider
|
|
||||||
// mergedHints that have a narrower NUMANodeAffinity than the
|
|
||||||
// NUMANodeAffinity in the current bestHint.
|
|
||||||
if !mergedHint.NUMANodeAffinity.IsNarrowerThan(bestHint.NUMANodeAffinity) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// In all other cases, update bestHint to the current mergedHint
|
|
||||||
bestHint = mergedHint
|
|
||||||
})
|
|
||||||
|
|
||||||
if bestHint.NUMANodeAffinity.IsEqual(defaultAffinity) {
|
|
||||||
bestHint = TopologyHint{nil, bestHint.Preferred}
|
|
||||||
}
|
|
||||||
|
|
||||||
return bestHint
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *singleNumaNodePolicy) Merge(providersHints []map[string][]TopologyHint) (TopologyHint, lifecycle.PodAdmitResult) {
|
func (p *singleNumaNodePolicy) Merge(providersHints []map[string][]TopologyHint) (TopologyHint, lifecycle.PodAdmitResult) {
|
||||||
filteredHints := filterProvidersHints(providersHints)
|
filteredHints := filterProvidersHints(providersHints)
|
||||||
// Filter to only include don't cares and hints with a single NUMA node.
|
// Filter to only include don't cares and hints with a single NUMA node.
|
||||||
singleNumaHints := filterSingleNumaHints(filteredHints)
|
singleNumaHints := filterSingleNumaHints(filteredHints)
|
||||||
hint := p.mergeProvidersHints(singleNumaHints)
|
bestHint := mergeFilteredHints(p.numaNodes, singleNumaHints)
|
||||||
admit := p.canAdmitPodResult(&hint)
|
|
||||||
return hint, admit
|
defaultAffinity, _ := bitmask.NewBitMask(p.numaNodes...)
|
||||||
|
if bestHint.NUMANodeAffinity.IsEqual(defaultAffinity) {
|
||||||
|
bestHint = TopologyHint{nil, bestHint.Preferred}
|
||||||
|
}
|
||||||
|
|
||||||
|
admit := p.canAdmitPodResult(&bestHint)
|
||||||
|
return bestHint, admit
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user