File tree Expand file tree Collapse file tree 1 file changed +108
-108
lines changed Expand file tree Collapse file tree 1 file changed +108
-108
lines changed Original file line number Diff line number Diff line change @@ -243,116 +243,116 @@ def get_tpu_system_characteristics_map(
243
243
),
244
244
# TPU system characteristics
245
245
** get_tpu_system_characteristics_map (
246
- prefix = " tpu7x" ,
247
- tensorcores_per_chip = 2 ,
248
- gke_accelerator = " tpu7x" ,
249
- machine_type = " tpu7x-standard-1t" ,
250
- supported_topologies = [ " 1x1x1" ]
246
+ ' tpu7x' ,
247
+ 2 ,
248
+ ' tpu7x' ,
249
+ ' tpu7x-standard-1t' ,
250
+ [ ' 1x1x1' ]
251
251
),
252
252
** get_tpu_system_characteristics_map (
253
- prefix = " tpu7x" ,
254
- tensorcores_per_chip = 2 ,
255
- gke_accelerator = " tpu7x" ,
256
- machine_type = " tpu7x-standard-4t" ,
257
- supported_topologies = [
258
- " 12x12x12" ,
259
- " 12x12x16" ,
260
- " 12x12x20" ,
261
- " 12x12x24" ,
262
- " 12x12x28" ,
263
- " 12x12x36" ,
264
- " 12x16x16" ,
265
- " 12x16x20" ,
266
- " 12x16x24" ,
267
- " 12x16x28" ,
268
- " 12x20x20" ,
269
- " 12x20x24" ,
270
- " 12x24x24" ,
271
- " 16x16x16" ,
272
- " 16x16x20" ,
273
- " 16x16x24" ,
274
- " 16x16x32" ,
275
- " 16x20x28" ,
276
- " 16x24x24" ,
277
- " 2x2x1" ,
278
- " 2x2x2" ,
279
- " 2x2x4" ,
280
- " 2x4x4" ,
281
- " 4x12x116" ,
282
- " 4x12x12" ,
283
- " 4x12x124" ,
284
- " 4x12x20" ,
285
- " 4x12x28" ,
286
- " 4x12x44" ,
287
- " 4x12x52" ,
288
- " 4x12x68" ,
289
- " 4x12x76" ,
290
- " 4x12x92" ,
291
- " 4x20x20" ,
292
- " 4x20x28" ,
293
- " 4x20x44" ,
294
- " 4x20x52" ,
295
- " 4x20x68" ,
296
- " 4x20x76" ,
297
- " 4x28x28" ,
298
- " 4x28x44" ,
299
- " 4x28x52" ,
300
- " 4x4x116" ,
301
- " 4x4x12" ,
302
- " 4x4x124" ,
303
- " 4x4x148" ,
304
- " 4x4x164" ,
305
- " 4x4x172" ,
306
- " 4x4x188" ,
307
- " 4x4x20" ,
308
- " 4x4x212" ,
309
- " 4x4x236" ,
310
- " 4x4x244" ,
311
- " 4x4x28" ,
312
- " 4x4x4" ,
313
- " 4x4x44" ,
314
- " 4x4x52" ,
315
- " 4x4x68" ,
316
- " 4x4x76" ,
317
- " 4x4x8" ,
318
- " 4x4x92" ,
319
- " 4x8x116" ,
320
- " 4x8x12" ,
321
- " 4x8x124" ,
322
- " 4x8x148" ,
323
- " 4x8x164" ,
324
- " 4x8x172" ,
325
- " 4x8x188" ,
326
- " 4x8x20" ,
327
- " 4x8x28" ,
328
- " 4x8x44" ,
329
- " 4x8x52" ,
330
- " 4x8x68" ,
331
- " 4x8x76" ,
332
- " 4x8x8" ,
333
- " 4x8x92" ,
334
- " 8x12x12" ,
335
- " 8x12x16" ,
336
- " 8x12x20" ,
337
- " 8x12x28" ,
338
- " 8x12x44" ,
339
- " 8x12x52" ,
340
- " 8x16x16" ,
341
- " 8x16x20" ,
342
- " 8x16x28" ,
343
- " 8x16x44" ,
344
- " 8x20x20" ,
345
- " 8x20x28" ,
346
- " 8x8x12" ,
347
- " 8x8x16" ,
348
- " 8x8x20" ,
349
- " 8x8x28" ,
350
- " 8x8x44" ,
351
- " 8x8x52" ,
352
- " 8x8x68" ,
353
- " 8x8x76" ,
354
- " 8x8x8" ,
355
- " 8x8x92" ,
253
+ ' tpu7x' ,
254
+ 2 ,
255
+ ' tpu7x' ,
256
+ ' tpu7x-standard-4t' ,
257
+ [
258
+ ' 12x12x12' ,
259
+ ' 12x12x16' ,
260
+ ' 12x12x20' ,
261
+ ' 12x12x24' ,
262
+ ' 12x12x28' ,
263
+ ' 12x12x36' ,
264
+ ' 12x16x16' ,
265
+ ' 12x16x20' ,
266
+ ' 12x16x24' ,
267
+ ' 12x16x28' ,
268
+ ' 12x20x20' ,
269
+ ' 12x20x24' ,
270
+ ' 12x24x24' ,
271
+ ' 16x16x16' ,
272
+ ' 16x16x20' ,
273
+ ' 16x16x24' ,
274
+ ' 16x16x32' ,
275
+ ' 16x20x28' ,
276
+ ' 16x24x24' ,
277
+ ' 2x2x1' ,
278
+ ' 2x2x2' ,
279
+ ' 2x2x4' ,
280
+ ' 2x4x4' ,
281
+ ' 4x12x116' ,
282
+ ' 4x12x12' ,
283
+ ' 4x12x124' ,
284
+ ' 4x12x20' ,
285
+ ' 4x12x28' ,
286
+ ' 4x12x44' ,
287
+ ' 4x12x52' ,
288
+ ' 4x12x68' ,
289
+ ' 4x12x76' ,
290
+ ' 4x12x92' ,
291
+ ' 4x20x20' ,
292
+ ' 4x20x28' ,
293
+ ' 4x20x44' ,
294
+ ' 4x20x52' ,
295
+ ' 4x20x68' ,
296
+ ' 4x20x76' ,
297
+ ' 4x28x28' ,
298
+ ' 4x28x44' ,
299
+ ' 4x28x52' ,
300
+ ' 4x4x116' ,
301
+ ' 4x4x12' ,
302
+ ' 4x4x124' ,
303
+ ' 4x4x148' ,
304
+ ' 4x4x164' ,
305
+ ' 4x4x172' ,
306
+ ' 4x4x188' ,
307
+ ' 4x4x20' ,
308
+ ' 4x4x212' ,
309
+ ' 4x4x236' ,
310
+ ' 4x4x244' ,
311
+ ' 4x4x28' ,
312
+ ' 4x4x4' ,
313
+ ' 4x4x44' ,
314
+ ' 4x4x52' ,
315
+ ' 4x4x68' ,
316
+ ' 4x4x76' ,
317
+ ' 4x4x8' ,
318
+ ' 4x4x92' ,
319
+ ' 4x8x116' ,
320
+ ' 4x8x12' ,
321
+ ' 4x8x124' ,
322
+ ' 4x8x148' ,
323
+ ' 4x8x164' ,
324
+ ' 4x8x172' ,
325
+ ' 4x8x188' ,
326
+ ' 4x8x20' ,
327
+ ' 4x8x28' ,
328
+ ' 4x8x44' ,
329
+ ' 4x8x52' ,
330
+ ' 4x8x68' ,
331
+ ' 4x8x76' ,
332
+ ' 4x8x8' ,
333
+ ' 4x8x92' ,
334
+ ' 8x12x12' ,
335
+ ' 8x12x16' ,
336
+ ' 8x12x20' ,
337
+ ' 8x12x28' ,
338
+ ' 8x12x44' ,
339
+ ' 8x12x52' ,
340
+ ' 8x16x16' ,
341
+ ' 8x16x20' ,
342
+ ' 8x16x28' ,
343
+ ' 8x16x44' ,
344
+ ' 8x20x20' ,
345
+ ' 8x20x28' ,
346
+ ' 8x8x12' ,
347
+ ' 8x8x16' ,
348
+ ' 8x8x20' ,
349
+ ' 8x8x28' ,
350
+ ' 8x8x44' ,
351
+ ' 8x8x52' ,
352
+ ' 8x8x68' ,
353
+ ' 8x8x76' ,
354
+ ' 8x8x8' ,
355
+ ' 8x8x92' ,
356
356
]
357
357
),
358
358
** get_tpu_system_characteristics_map (
You can’t perform that action at this time.
0 commit comments