From 8be102efe6dbae2b4a9b4fc8a0950a26ac883794 Mon Sep 17 00:00:00 2001 From: gshaibi Date: Sun, 4 Aug 2024 11:53:26 +0300 Subject: [PATCH] . --- .../templates/device-plugin/daemonset.yml | 9 +++++++-- .../templates/status-exporter/daemonset.yaml | 9 +++++++-- 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/deploy/fake-gpu-operator/templates/device-plugin/daemonset.yml b/deploy/fake-gpu-operator/templates/device-plugin/daemonset.yml index 9daa387..87a692e 100644 --- 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 index 916f6b6..6fd4f5e 100644 --- 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" \ No newline at end of file + affinity: + nodeAffinity: + requiredDuringSchedulingIgnoredDuringExecution: + nodeSelectorTerms: + - matchExpressions: + - key: run.ai/simulated-gpu-node-pool + operator: Exists \ No newline at end of file