From b7c947804ac26140bcc9c33ed41d5f7dff51424f Mon Sep 17 00:00:00 2001
From: lomtom <lomtom@qq.com>
Date: Fri, 30 Jan 2026 16:03:57 +0800
Subject: [PATCH] fix: creating multiple pods at the same time will be assigned
 to the same NPU.

Signed-off-by: lomtom <lomtom@qq.com>
---
 internal/server/server.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/internal/server/server.go b/internal/server/server.go
index 22f0927..442d39d 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -350,6 +350,14 @@ func (ps *PluginServer) Allocate(ctx context.Context, reqs *v1beta1.AllocateRequ
 	success := false
 	var pod *v1.Pod
 	defer func() {
+		if success {
+			annotations := make(map[string]string)
+			annotations[util.DeviceBindPhase] = util.DeviceBindSuccess
+			err := util.PatchPodAnnotations(pod, annotations)
+			if err != nil {
+				klog.Errorf("patch pod %s annotations error: %v", pod.Name, err)
+			}
+		}
 		lockerr := nodelock.ReleaseNodeLock(ps.nodeName, NodeLockAscend, pod, success)
 		if lockerr != nil {
 			klog.Errorf("failed to release lock:%s", lockerr.Error())