From 11c684db3b5573e4893bba0bca6b2e9d8604631f Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Fri, 13 Aug 2021 17:39:25 +0800
Subject: [PATCH] fix kubelet pod annotation wrong

---
 pkg/gpu/nvidia/podmanager.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/pkg/gpu/nvidia/podmanager.go b/pkg/gpu/nvidia/podmanager.go
index fa132da..fa64268 100644
--- a/pkg/gpu/nvidia/podmanager.go
+++ b/pkg/gpu/nvidia/podmanager.go
@@ -221,7 +221,13 @@ func getCandidatePods(queryKubelet bool, client *client.KubeletClient) ([]*v1.Po
 	for _, pod := range allPods {
 		current := pod
 		if isGPUMemoryAssumedPod(&current) {
-			candidatePods = append(candidatePods, &current)
+			currentFromApi, err := clientset.CoreV1().Pods(pod.Namespace).Get(pod.Name, metav1.GetOptions{})
+			if err != nil {
+				return candidatePods, err
+			}
+			if isGPUMemoryAssumedPod(currentFromApi) {
+				candidatePods = append(candidatePods, currentFromApi)
+			}
 		}
 	}