Merge pull request #1242 from NVIDIA/mig-b200

NVIDIA · Jan 30, 2025 · 00a8b9d · 00a8b9d
2 parents b127216 + 183026b
commit 00a8b9d
Showing 1 changed file with 48 additions and 1 deletion.
diff --git a/assets/state-mig-manager/0400_configmap.yaml b/assets/state-mig-manager/0400_configmap.yaml
@@ -83,6 +83,20 @@ data:
           mig-devices:
             "1g.20gb": 4
 
+      # B200
+      all-1g.23gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.23gb": 7
+
+      # B200
+      all-1g.23gb.me:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.23gb+me": 1
+
       all-2g.20gb:
         - devices: all
           mig-enabled: true
@@ -157,12 +171,24 @@ data:
           mig-devices:
             "1g.24gb": 4
 
+      all-1g.45gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.45gb": 4
+
       all-2g.24gb:
         - devices: all
           mig-enabled: true
           mig-devices:
             "2g.24gb": 3
 
+      all-2g.45gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "2g.45gb": 3
+            
       # H100 NVL, H800 NVL
       all-3g.47gb:
         - devices: all
@@ -189,6 +215,12 @@ data:
           mig-devices:
             "3g.48gb": 2
 
+      all-3g.90gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "3g.90gb": 2
+
       all-4g.48gb:
         - devices: all
           mig-enabled: true
@@ -201,8 +233,23 @@ data:
           mig-devices:
             "7g.96gb": 1
 
-      # GH200 144G HBM3e, H200-141GB, H200 NVL, H100-96GB, GH200, H100 NVL, H800 NVL, H100-80GB, H800-80GB, A800-40GB, A800-80GB, A100-40GB, A100-80GB, A30-24GB, PG506-96GB
+      all-7g.180gb:
+        - devices: all
+          mig-enabled: true
+          mig-devices:
+            "7g.180gb": 1
+
+      # B200, GH200 144G HBM3e, H200-141GB, H200 NVL, 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:
+        # B200
+        - device-filter: ["0x290110DE"]
+          devices: all
+          mig-enabled: true
+          mig-devices:
+            "1g.23gb": 2
+            "2g.45gb": 1
+            "3g.90gb": 1
+
         # GH200 144G HBM3e
         - device-filter: ["0x234810DE"]
           devices: all