From 37f0c4ae58012e369a1fd50860a9ad80d2d42e3c Mon Sep 17 00:00:00 2001 From: youys <1272586223@qq.com> Date: Tue, 5 Aug 2025 10:49:29 +0800 Subject: [PATCH] =?UTF-8?q?fix(node):=20=E5=8F=91=E7=8E=B0=E8=8A=82?= =?UTF-8?q?=E7=82=B9=E8=BF=87=E6=BB=A4=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/internal/data/node.go | 2 +- server/internal/service/resource_pool.go | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/server/internal/data/node.go b/server/internal/data/node.go index 02415bb..4f2fb39 100644 --- a/server/internal/data/node.go +++ b/server/internal/data/node.go @@ -106,7 +106,7 @@ func (r *nodeRepo) updateLocalNodes() { var all []*biz.Node - allNodes, _ := r.nodeLister.List(labels.Set{"gpu": "on"}.AsSelector()) + allNodes, _ := r.nodeLister.List(labels.Everything()) for _, node := range allNodes { bizNode := r.fetchNodeInfo(node) gpuNode := n[k8stypes.UID(bizNode.Uid)] diff --git a/server/internal/service/resource_pool.go b/server/internal/service/resource_pool.go index b552516..cc09ea1 100644 --- a/server/internal/service/resource_pool.go +++ b/server/internal/service/resource_pool.go @@ -207,6 +207,9 @@ func (s *ResourcePoolService) getK8sNodes(ctx context.Context) map[string]*biz.N nodes, _ := s.uc.ListAllNodesV2(ctx) m := make(map[string]*biz.Node) for _, node := range nodes { + if node.Lables["gpu"] != "on" { + continue + } node.GPUCount = int64(len(node.Devices)) m[node.Name] = node }