add node_selector_enabled variable

ROCm · Apr 17, 2024 · 73d8618 · 73d8618
1 parent 0a820b8
commit 73d8618
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 2 deletions.
diff --git a/helm/amd-gpu/README.md b/helm/amd-gpu/README.md
@@ -26,6 +26,7 @@ Kubernetes: `>= 1.18.0`
 | lbl.image.tag | string | `"labeller-latest"` |  |
 | namespace | string | `"kube-system"` |  |
 | nfd.enabled | bool | `false` |  |
+| node_selector_enabled | bool | `false` |  |
 | node_selector."feature.node.kubernetes.io/pci-0300_1002.present" | string | `"true"` |  |
 | securityContext.allowPrivilegeEscalation | bool | `false` |  |
 | securityContext.capabilities.drop[0] | string | `"ALL"` |  |

diff --git a/helm/amd-gpu/templates/deviceplugin-daemonset.yaml b/helm/amd-gpu/templates/deviceplugin-daemonset.yaml
@@ -16,7 +16,7 @@ spec:
       imagePullSecrets:
         {{- toYaml . | nindent 8 }}
       {{- end }}
-      {{- if .Values.nfd.enabled }}
+      {{- if .Values.node_selector_enabled }}
       {{- with .Values.node_selector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}

diff --git a/helm/amd-gpu/templates/labeller.yaml b/helm/amd-gpu/templates/labeller.yaml
@@ -39,7 +39,7 @@ spec:
       labels:
         name: amdgpu-lr-ds
     spec:
-      {{- if .Values.nfd.enabled }}
+      {{- if .Values.node_selector_enabled }}
       {{- with .Values.node_selector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}

diff --git a/helm/amd-gpu/values.yaml b/helm/amd-gpu/values.yaml
@@ -31,6 +31,7 @@ tolerations:
   - key: CriticalAddonsOnly
     operator: Exists
 
+node_selector_enabled: false
 node_selector:
   feature.node.kubernetes.io/pci-0300_1002.present: "true"
   kubernetes.io/arch: amd64