Merge branch 'main' of https://bdgit.educoder.net/pgpc5xf4t/hami

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)]

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
 	}