diff --git a/pkg/kubelet/cm/cpumanager/cpu_assignment.go b/pkg/kubelet/cm/cpumanager/cpu_assignment.go
index fe6e1b4187d..810380fa0a8 100644
--- a/pkg/kubelet/cm/cpumanager/cpu_assignment.go
+++ b/pkg/kubelet/cm/cpumanager/cpu_assignment.go
@@ -52,10 +52,13 @@ func (m mapIntInt) Keys() []int {
 	return keys
 }
 
-func (m mapIntInt) Values() []int {
+func (m mapIntInt) Values(keys ...int) []int {
+	if keys == nil {
+		keys = m.Keys()
+	}
 	var values []int
-	for _, v := range m {
-		values = append(values, v)
+	for _, k := range keys {
+		values = append(values, m[k])
 	}
 	return values
 }