.

run-ai · Aug 4, 2024 · 8be102e · 8be102e
1 parent 031a001
commit 8be102e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/deploy/fake-gpu-operator/templates/device-plugin/daemonset.yml b/deploy/fake-gpu-operator/templates/device-plugin/daemonset.yml
@@ -12,5 +12,10 @@ spec:
       {{- include "fake-gpu-operator.device-plugin.common.podTemplate.metadata" . | nindent 6 }}
     spec:
       {{- include "fake-gpu-operator.device-plugin.common.podTemplate.spec" . | nindent 6 }}
-      nodeSelector:
-        nvidia.com/gpu.deploy.device-plugin: "true"
+      affinity:
+        nodeAffinity:
+          requiredDuringSchedulingIgnoredDuringExecution:
+            nodeSelectorTerms:
+            - matchExpressions:
+              - key: run.ai/simulated-gpu-node-pool
+                operator: Exists
diff --git a/deploy/fake-gpu-operator/templates/status-exporter/daemonset.yaml b/deploy/fake-gpu-operator/templates/status-exporter/daemonset.yaml
@@ -12,5 +12,10 @@ spec:
       {{- include "fake-gpu-operator.status-exporter.common.podTemplate.metadata" . | nindent 6 }}
     spec:
       {{- include "fake-gpu-operator.status-exporter.common.podTemplate.spec" . | nindent 6 }}
-      nodeSelector:
-        nvidia.com/gpu.deploy.dcgm-exporter: "true"
+      affinity:
+        nodeAffinity:
+          requiredDuringSchedulingIgnoredDuringExecution:
+            nodeSelectorTerms:
+            - matchExpressions:
+              - key: run.ai/simulated-gpu-node-pool
+                operator: Exists