Skip to content

Commit 2deced2

Browse files
committed
feat: add tpu7x support
1 parent e3ae587 commit 2deced2

File tree

1 file changed

+109
-0
lines changed

1 file changed

+109
-0
lines changed

src/xpk/core/system_characteristics.py

Lines changed: 109 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,115 @@ def get_tpu_system_characteristics_map(
242242
'h100-mega-80gb-8',
243243
),
244244
# TPU system characteristics
245+
**get_tpu_system_characteristics_map(
246+
'tpu7x', 2, 'tpu7x', 'tpu7x-standard-1t', ['1x1x1']
247+
),
248+
**get_tpu_system_characteristics_map(
249+
'tpu7x',
250+
2,
251+
'tpu7x',
252+
'tpu7x-standard-4t',
253+
[
254+
'12x12x12',
255+
'12x12x16',
256+
'12x12x20',
257+
'12x12x24',
258+
'12x12x28',
259+
'12x12x36',
260+
'12x16x16',
261+
'12x16x20',
262+
'12x16x24',
263+
'12x16x28',
264+
'12x20x20',
265+
'12x20x24',
266+
'12x24x24',
267+
'16x16x16',
268+
'16x16x20',
269+
'16x16x24',
270+
'16x16x32',
271+
'16x20x28',
272+
'16x24x24',
273+
'2x2x1',
274+
'2x2x2',
275+
'2x2x4',
276+
'2x4x4',
277+
'4x12x116',
278+
'4x12x12',
279+
'4x12x124',
280+
'4x12x20',
281+
'4x12x28',
282+
'4x12x44',
283+
'4x12x52',
284+
'4x12x68',
285+
'4x12x76',
286+
'4x12x92',
287+
'4x20x20',
288+
'4x20x28',
289+
'4x20x44',
290+
'4x20x52',
291+
'4x20x68',
292+
'4x20x76',
293+
'4x28x28',
294+
'4x28x44',
295+
'4x28x52',
296+
'4x4x116',
297+
'4x4x12',
298+
'4x4x124',
299+
'4x4x148',
300+
'4x4x164',
301+
'4x4x172',
302+
'4x4x188',
303+
'4x4x20',
304+
'4x4x212',
305+
'4x4x236',
306+
'4x4x244',
307+
'4x4x28',
308+
'4x4x4',
309+
'4x4x44',
310+
'4x4x52',
311+
'4x4x68',
312+
'4x4x76',
313+
'4x4x8',
314+
'4x4x92',
315+
'4x8x116',
316+
'4x8x12',
317+
'4x8x124',
318+
'4x8x148',
319+
'4x8x164',
320+
'4x8x172',
321+
'4x8x188',
322+
'4x8x20',
323+
'4x8x28',
324+
'4x8x44',
325+
'4x8x52',
326+
'4x8x68',
327+
'4x8x76',
328+
'4x8x8',
329+
'4x8x92',
330+
'8x12x12',
331+
'8x12x16',
332+
'8x12x20',
333+
'8x12x28',
334+
'8x12x44',
335+
'8x12x52',
336+
'8x16x16',
337+
'8x16x20',
338+
'8x16x28',
339+
'8x16x44',
340+
'8x20x20',
341+
'8x20x28',
342+
'8x8x12',
343+
'8x8x16',
344+
'8x8x20',
345+
'8x8x28',
346+
'8x8x44',
347+
'8x8x52',
348+
'8x8x68',
349+
'8x8x76',
350+
'8x8x8',
351+
'8x8x92',
352+
],
353+
),
245354
**get_tpu_system_characteristics_map(
246355
'v6e', 1, 'tpu-v6e-slice', 'ct6e-standard-1t', ['1x1']
247356
),

0 commit comments

Comments
 (0)