Skip to content

Commit 854bf0d

Browse files
committed
feat: query GpuNodes and then get corev1.node based on Gpunode
1 parent bcf45a4 commit 854bf0d

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

internal/controller/gpupool_controller.go

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -117,13 +117,17 @@ func (r *GPUPoolReconciler) startNodeDiscoverys(
117117
return fmt.Errorf("unmarshal pod template: %w", err)
118118
}
119119
// pool.Spec.NodeManagerConfig.NodeSelector
120-
nodes := &corev1.NodeList{}
120+
nodes := &tfv1.GPUNodeList{}
121121
if err := r.Client.List(ctx, nodes); err != nil {
122122
return fmt.Errorf("list nodes: %v", err)
123123
}
124124

125-
for _, node := range nodes.Items {
126-
matches, err := schedulingcorev1.MatchNodeSelectorTerms(&node, pool.Spec.NodeManagerConfig.NodeSelector)
125+
for _, gpuNode := range nodes.Items {
126+
node := &corev1.Node{}
127+
if err := r.Client.Get(ctx, client.ObjectKeyFromObject(&gpuNode), node); err != nil {
128+
return err
129+
}
130+
matches, err := schedulingcorev1.MatchNodeSelectorTerms(node, pool.Spec.NodeManagerConfig.NodeSelector)
127131
if err != nil {
128132
return err
129133
}

0 commit comments

Comments
 (0)