qrm cpu & memory plugin support set numa binding numa id to allocatio…

…n state for shared or dedicated numa binding without exclusive pods

pkg/agent/qrm-plugins/cpu/dynamicpolicy/policy_allocation_handlers.go

@@ -386,6 +386,14 @@ func (p *DynamicPolicy) dedicatedCoresWithNUMABindingAllocationHandler(ctx conte
 		RequestQuantity:                  reqFloat64,
 	}
 
+	if !qosutil.AnnotationsIndicateNUMAExclusive(req.Annotations) {
+		if result.Size() != 1 {
+			return nil, fmt.Errorf("numa binding without numa exclusive allocation result numa node size is %d, "+
+				"not equal to 1", result.Size())
+		}
+		allocationInfo.SetSpecifiedNUMABindingNUMAID(result.ToSliceNoSortUInt64()[0])
+	}
+
 	// update pod entries directly.
 	// if one of subsequent steps is failed, we will delete current allocationInfo from podEntries in defer function of allocation function.
 	p.state.SetAllocationInfo(allocationInfo.PodUid, allocationInfo.ContainerName, allocationInfo)

pkg/agent/qrm-plugins/memory/dynamicpolicy/policy_allocation_handlers.go

@@ -208,6 +208,15 @@ func (p *DynamicPolicy) numaBindingAllocationHandler(ctx context.Context,
 		NumaAllocationResult:     result.Clone(),
 		TopologyAwareAllocations: topologyAwareAllocations,
 	}
+
+	if !qosutil.AnnotationsIndicateNUMAExclusive(req.Annotations) {
+		if result.Size() != 1 {
+			return nil, fmt.Errorf("numa binding without numa exclusive allocation result numa node size is %d, "+
+				"not equal to 1", result.Size())
+		}
+		allocationInfo.SetSpecifiedNUMABindingNUMAID(result.ToSliceNoSortUInt64()[0])
+	}
+
 	p.state.SetAllocationInfo(v1.ResourceMemory, req.PodUid, req.ContainerName, allocationInfo)
 
 	podResourceEntries = p.state.GetPodResourceEntries()