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 }