Skip to content

Commit

Permalink
Support v5litepod-8.
Browse files Browse the repository at this point in the history
  • Loading branch information
mbobrovskyi committed Dec 17, 2024
1 parent a24a0f8 commit a006f27
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions src/xpk/core/system_characteristics.py
Original file line number Diff line number Diff line change
Expand Up @@ -1081,6 +1081,15 @@ def get_system_characteristics(
'v5p-17920',
),
# v5litepod
'v5litepod-8': SystemCharacteristics(
'2x4',
4,
'tpu-v5-lite-podslice',
'ct5lp-hightpu-8t',
8,
AcceleratorType['TPU'],
'v5litepod-8',
),
'v5litepod-16': SystemCharacteristics(
'4x4',
4,
Expand Down

0 comments on commit a006f27

Please sign in to comment.