Skip to content

Commit 6415fa7

Browse files
committed
Remove nodes slice in loop of takeByTopology
1 parent c58b632 commit 6415fa7

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

pkg/kubelet/cm/devicemanager/manager.go

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -688,21 +688,21 @@ func (m *ManagerImpl) takeByTopology(resource string, available sets.String, aff
688688
// available device does not have any NUMA Nodes associated with it, add it
689689
// to a list of NUMA Nodes for the fake NUMANode -1.
690690
perNodeDevices := make(map[int]sets.String)
691+
nodeWithoutTopology := -1
691692
for d := range available {
692-
var nodes []int
693-
if m.allDevices[resource][d].Topology != nil {
694-
for _, node := range m.allDevices[resource][d].Topology.Nodes {
695-
nodes = append(nodes, int(node.ID))
693+
if m.allDevices[resource][d].Topology == nil || len(m.allDevices[resource][d].Topology.Nodes) == 0 {
694+
if _, ok := perNodeDevices[nodeWithoutTopology]; !ok {
695+
perNodeDevices[nodeWithoutTopology] = sets.NewString()
696696
}
697+
perNodeDevices[nodeWithoutTopology].Insert(d)
698+
continue
697699
}
698-
if len(nodes) == 0 {
699-
nodes = []int{-1}
700-
}
701-
for _, node := range nodes {
702-
if _, ok := perNodeDevices[node]; !ok {
703-
perNodeDevices[node] = sets.NewString()
700+
701+
for _, node := range m.allDevices[resource][d].Topology.Nodes {
702+
if _, ok := perNodeDevices[int(node.ID)]; !ok {
703+
perNodeDevices[int(node.ID)] = sets.NewString()
704704
}
705-
perNodeDevices[node].Insert(d)
705+
perNodeDevices[int(node.ID)].Insert(d)
706706
}
707707
}
708708

@@ -734,7 +734,7 @@ func (m *ManagerImpl) takeByTopology(resource string, available sets.String, aff
734734
// has the device is encountered.
735735
for _, n := range nodes {
736736
if perNodeDevices[n].Has(d) {
737-
if n == -1 {
737+
if n == nodeWithoutTopology {
738738
withoutTopology = append(withoutTopology, d)
739739
} else if affinity.IsSet(n) {
740740
fromAffinity = append(fromAffinity, d)

0 commit comments

Comments
 (0)