From 43266617e4222e6d59e9dad55a9dd2251b92eb1a Mon Sep 17 00:00:00 2001
From: Julio Chana <julio.chana@lokalise.com>
Date: Thu, 23 Jan 2025 17:09:33 +0100
Subject: [PATCH] Add resources to sidecar container

Signed-off-by: Julio Chana <julio.chana@lokalise.com>
---
 .../templates/daemonset-device-plugin.yml                     | 4 ++++
 .../helm/nvidia-device-plugin/templates/daemonset-gfd.yml     | 4 ++++
 .../templates/daemonset-mps-control-daemon.yml                | 4 ++++
 3 files changed, 12 insertions(+)

diff --git a/deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml b/deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml
index 1fdd0b6f4..e8c8ccd39 100644
--- a/deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml
+++ b/deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml
@@ -129,6 +129,10 @@ spec:
             mountPath: /config
         securityContext:
           {{- include "nvidia-device-plugin.securityContext" . | nindent 10 }}
+        {{- with .Values.resources }}
+        resources:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
       {{- end }}
       - image: {{ include "nvidia-device-plugin.fullimage" . }}
         imagePullPolicy: {{ .Values.image.pullPolicy }}
diff --git a/deployments/helm/nvidia-device-plugin/templates/daemonset-gfd.yml b/deployments/helm/nvidia-device-plugin/templates/daemonset-gfd.yml
index 4b398f3bd..1bd18ccbd 100644
--- a/deployments/helm/nvidia-device-plugin/templates/daemonset-gfd.yml
+++ b/deployments/helm/nvidia-device-plugin/templates/daemonset-gfd.yml
@@ -157,6 +157,10 @@ spec:
             mountPath: /config
         securityContext:
           {{- include "gpu-feature-discovery.securityContext" . | nindent 10 }}
+        {{- with .Values.resources }}
+        resources:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
       {{- end }}
       - image: {{ include "nvidia-device-plugin.fullimage" . }}
         imagePullPolicy: {{ .Values.image.pullPolicy }}
diff --git a/deployments/helm/nvidia-device-plugin/templates/daemonset-mps-control-daemon.yml b/deployments/helm/nvidia-device-plugin/templates/daemonset-mps-control-daemon.yml
index da37aba6d..1c236e04d 100644
--- a/deployments/helm/nvidia-device-plugin/templates/daemonset-mps-control-daemon.yml
+++ b/deployments/helm/nvidia-device-plugin/templates/daemonset-mps-control-daemon.yml
@@ -144,6 +144,10 @@ spec:
               mountPath: /available-configs
             - name: config
               mountPath: /config
+          {{- with .Values.resources }}
+          resources:
+            {{- toYaml . | nindent 12 }}
+          {{- end }}
       {{- end }}
         - image: {{ include "nvidia-device-plugin.fullimage" . }}
           imagePullPolicy: {{ .Values.image.pullPolicy }}