Skip to content

Commit 040a819

Browse files
yansun1996sajmera-pensando
authored andcommitted
Avoid auto cleanup node labeller labels on vgpu only node
1 parent b49d833 commit 040a819

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

internal/controllers/device_config_reconciler.go

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1063,12 +1063,19 @@ func (dcrh *deviceConfigReconcilerHelper) handleNodeLabeller(ctx context.Context
10631063
// nodes without gpu, kmm, dev-plugin
10641064
sel := []string{
10651065
"! " + utils.NodeFeatureLabelAmdGpu,
1066-
"! " + labels.GetKernelModuleReadyNodeLabel(devConfig.Namespace, devConfig.Name),
1067-
"! " + labels.GetDevicePluginNodeLabel(devConfig.Namespace, devConfig.Name),
1066+
"! " + utils.NodeFeatureLabelAmdVGpu,
1067+
}
1068+
1069+
if devConfig.Spec.Driver.Enable != nil && *devConfig.Spec.Driver.Enable {
1070+
sel = append(sel,
1071+
"! "+labels.GetKernelModuleReadyNodeLabel(devConfig.Namespace, devConfig.Name),
1072+
"! "+labels.GetDevicePluginNodeLabel(devConfig.Namespace, devConfig.Name),
1073+
)
10681074
}
10691075

10701076
for k, v := range devConfig.Spec.Selector {
1071-
if k == utils.NodeFeatureLabelAmdGpu { // skip
1077+
if k == utils.NodeFeatureLabelAmdGpu ||
1078+
k == utils.NodeFeatureLabelAmdVGpu { // skip
10721079
continue
10731080
}
10741081
sel = append(sel, fmt.Sprintf("%s=%s", k, v))

0 commit comments

Comments
 (0)