File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff 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 }
You can’t perform that action at this time.
0 commit comments