[MIG] add support for H200 141GB

Signed-off-by: Tariq Ibrahim <[email protected]>
NVIDIA · Jun 24, 2024 · 28a15bf · 28a15bf
1 parent 5d11023
commit 28a15bf
Showing 1 changed file with 53 additions and 2 deletions.
diff --git a/assets/state-mig-manager/0400_configmap.yaml b/assets/state-mig-manager/0400_configmap.yaml
@@ -201,9 +201,17 @@ data:
           mig-devices:
             "7g.96gb": 1
 
-
-      # H100-96GB, GH200, H100 NVL, H800 NVL, H100-80GB, H800-80GB, A800-40GB, A800-80GB, A100-40GB, A100-80GB, A30-24GB, PG506-96GB
+      # H200-141GB, H100-96GB, GH200, H100 NVL, H800 NVL, H100-80GB, H800-80GB, A800-40GB, A800-80GB, A100-40GB, A100-80GB, A30-24GB, PG506-96GB
       all-balanced:
+        # H200 141GB
+        - device-filter: ["0x233510DE"]
+          devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.18gb": 2
+            "2g.35gb": 1
+            "3g.71gb": 1
+
         # H100 NVL, H800 NVL
         - device-filter: ["0x232110DE", "0x233A10DE"]
           devices: all
@@ -247,3 +255,46 @@ data:
             "1g.12gb": 2
             "2g.24gb": 1
             "3g.48gb": 1
+
+      # 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