add H200 141GB to the example config file

Signed-off-by: Tariq Ibrahim <[email protected]>
NVIDIA · Jun 28, 2024 · 0b70c37 · 0b70c37
1 parent b699211
commit 0b70c37
Showing 1 changed file with 52 additions and 0 deletions.
diff --git a/deployments/container/nvidia-mig-manager-example-hopper.yaml b/deployments/container/nvidia-mig-manager-example-hopper.yaml
@@ -107,6 +107,49 @@ data:
         - devices: all
           mig-enabled: false
 
+      # H200-141GB
+      all-1g.18gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.18gb": 7
+
+      all-1g.18gb.me:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.18gb+me": 1
+
+      all-1g.35gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.35gb": 4
+
+      all-2g.35gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "2g.35gb": 3
+
+      all-3g.71gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "3g.71gb": 2
+
+      all-4g.71gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "4g.71gb": 1
+
+      all-7g.141gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "7g.141gb": 1
+
       # H100-80GB, H800-80GB
       all-1g.10gb:
         - devices: all
@@ -183,6 +226,15 @@ data:
 
       # H100-80GB, H100-NVL, H800-80GB, H800-NVL
       all-balanced:
+        # H200 141GB
+        - device-filter: ["0x233510DE"]
+          devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.18gb": 2
+            "2g.35gb": 1
+            "3g.71gb": 1
+
         # H100-80GB, H800-80GB
         - device-filter: ["0x233110DE", "0x233010DE", "0x232210DE"]
           devices: all