Skip to content

Commit 42f329d

Browse files
committed
[MIG] add support for HGX GB200
Signed-off-by: Tariq Ibrahim <tibrahim@nvidia.com>
1 parent 1132ab5 commit 42f329d

File tree

1 file changed

+55
-2
lines changed

1 file changed

+55
-2
lines changed

assets/state-mig-manager/0400_configmap.yaml

Lines changed: 55 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,13 @@ data:
9797
mig-devices:
9898
"1g.23gb+me": 1
9999
100+
# GB200 HGX
101+
all-1g.24gb.me:
102+
- devices: all
103+
mig-enabled: true
104+
mig-devices:
105+
"1g.24gb+me": 1
106+
100107
all-2g.20gb:
101108
- devices: all
102109
mig-enabled: true
@@ -166,6 +173,13 @@ data:
166173
"1g.12gb+me": 1
167174
168175
all-1g.24gb:
176+
# GB200 HGX
177+
- device-filter: ["0x294110DE"]
178+
devices: all
179+
mig-enabled: true
180+
mig-devices:
181+
"1g.24gb": 7
182+
169183
- devices: all
170184
mig-enabled: true
171185
mig-devices:
@@ -177,6 +191,12 @@ data:
177191
mig-devices:
178192
"1g.45gb": 4
179193
194+
all-1g.47gb:
195+
- devices: all
196+
mig-enabled: true
197+
mig-devices:
198+
"1g.47gb": 4
199+
180200
all-2g.24gb:
181201
- devices: all
182202
mig-enabled: true
@@ -188,7 +208,13 @@ data:
188208
mig-enabled: true
189209
mig-devices:
190210
"2g.45gb": 3
191-
211+
212+
all-2g.47gb:
213+
- devices: all
214+
mig-enabled: true
215+
mig-devices:
216+
"2g.47gb": 3
217+
192218
# H100 NVL, H800 NVL
193219
all-3g.47gb:
194220
- devices: all
@@ -221,12 +247,24 @@ data:
221247
mig-devices:
222248
"3g.90gb": 2
223249
250+
all-3g.95gb:
251+
- devices: all
252+
mig-enabled: true
253+
mig-devices:
254+
"3g.95gb": 2
255+
224256
all-4g.48gb:
225257
- devices: all
226258
mig-enabled: true
227259
mig-devices:
228260
"4g.48gb": 1
229261
262+
all-4g.95gb:
263+
- devices: all
264+
mig-enabled: true
265+
mig-devices:
266+
"4g.95gb": 1
267+
230268
all-7g.96gb:
231269
- devices: all
232270
mig-enabled: true
@@ -239,8 +277,23 @@ data:
239277
mig-devices:
240278
"7g.180gb": 1
241279
242-
# 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
280+
all-7g.189gb:
281+
- devices: all
282+
mig-enabled: true
283+
mig-devices:
284+
"7g.189gb": 1
285+
286+
# GB200 HGX, 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
243287
all-balanced:
288+
# GB200 HGX
289+
- device-filter: ["0x294110DE"]
290+
devices: all
291+
mig-enabled: true
292+
mig-devices:
293+
"1g.24gb": 2
294+
"2g.47gb": 1
295+
"3g.95gb": 1
296+
244297
# B200
245298
- device-filter: ["0x290110DE"]
246299
devices: all

0 commit comments

Comments
 (0)