From 3afcd3e1bb83896eb77b2a28d836a53c204ccd95 Mon Sep 17 00:00:00 2001
From: Cory Latschkowski <koree@redhat.com>
Date: Wed, 18 Oct 2023 10:36:20 -0500
Subject: [PATCH] update: gpu overlays

---
 .../time-slicing-2/kustomization.yaml         |  6 +--
 .../time-slicing-4/kustomization.yaml         |  2 +-
 .../time-slicing-8-a100/kustomization.yaml    | 42 +++++++++++++++++++
 3 files changed, 46 insertions(+), 4 deletions(-)
 create mode 100644 gpu-operator-certified/instance/overlays/time-slicing-8-a100/kustomization.yaml

diff --git a/gpu-operator-certified/instance/overlays/time-slicing-2/kustomization.yaml b/gpu-operator-certified/instance/overlays/time-slicing-2/kustomization.yaml
index 94f5b390..1ed4b944 100644
--- a/gpu-operator-certified/instance/overlays/time-slicing-2/kustomization.yaml
+++ b/gpu-operator-certified/instance/overlays/time-slicing-2/kustomization.yaml
@@ -14,7 +14,7 @@ patches:
         value: device-plugin-config
       - op: add
         path: /spec/devicePlugin/config/default
-        value: Tesla-T4-time-sliced
+        value: Tesla-T4
       - op: replace
         path: /spec/gfd/enabled
         value: true
@@ -25,10 +25,10 @@ patches:
       - op: add
         path: /data
         value:
-          Tesla-T4-time-sliced: |-
+          Tesla-T4: |-
             version: v1
             sharing:
               timeSlicing:
                 resources:
                   - name: nvidia.com/gpu
-                    replicas: 2
\ No newline at end of file
+                    replicas: 2
diff --git a/gpu-operator-certified/instance/overlays/time-slicing-4/kustomization.yaml b/gpu-operator-certified/instance/overlays/time-slicing-4/kustomization.yaml
index 67a61c7a..35fe72ba 100644
--- a/gpu-operator-certified/instance/overlays/time-slicing-4/kustomization.yaml
+++ b/gpu-operator-certified/instance/overlays/time-slicing-4/kustomization.yaml
@@ -12,7 +12,7 @@ patches:
       - op: add
         path: /data
         value:
-          Tesla-T4-time-sliced: |-
+          Tesla-T4: |-
             version: v1
             sharing:
               timeSlicing:
diff --git a/gpu-operator-certified/instance/overlays/time-slicing-8-a100/kustomization.yaml b/gpu-operator-certified/instance/overlays/time-slicing-8-a100/kustomization.yaml
new file mode 100644
index 00000000..802544b3
--- /dev/null
+++ b/gpu-operator-certified/instance/overlays/time-slicing-8-a100/kustomization.yaml
@@ -0,0 +1,42 @@
+apiVersion: kustomize.config.k8s.io/v1beta1
+kind: Kustomization
+
+resources:
+  - ../../base
+
+patches:
+  - target:
+      kind: ClusterPolicy
+      name: gpu-cluster-policy
+    patch: |-
+      - op: add
+        path: /spec/devicePlugin/config/name
+        value: device-plugin-config
+      - op: add
+        path: /spec/devicePlugin/config/default
+        value: A100-SXM4-40GB
+      - op: replace
+        path: /spec/gfd/enabled
+        value: true
+  - target:
+      kind: ConfigMap
+      name: device-plugin-config
+    patch: |-
+      - op: add
+        path: /data
+        value:
+          A100-SXM4-40GB: |-
+            version: v1
+            sharing:
+              timeSlicing:
+                resources:
+                  - name: nvidia.com/gpu
+                    replicas: 8
+                  - name: nvidia.com/mig-1g.5gb
+                    replicas: 1
+                  - name: nvidia.com/mig-2g.10gb
+                    replicas: 2
+                  - name: nvidia.com/mig-3g.20gb
+                    replicas: 3
+                  - name: nvidia.com/mig-7g.40gb
+                    replicas: 7
\ No newline at end of file