From 5ee95559ea0cbdee58243728c45ff9c64c3e2c8c Mon Sep 17 00:00:00 2001 From: Mykhailo Bobrovskyi Date: Wed, 11 Dec 2024 11:21:58 +0200 Subject: [PATCH] Support v5litepod-8. --- .github/workflows/build_tests.yaml | 1 + src/xpk/core/system_characteristics.py | 9 +++++++++ 2 files changed, 10 insertions(+) diff --git a/.github/workflows/build_tests.yaml b/.github/workflows/build_tests.yaml index 30be9873..6394ac6d 100644 --- a/.github/workflows/build_tests.yaml +++ b/.github/workflows/build_tests.yaml @@ -24,6 +24,7 @@ on: type: choice options: - v4-8 + - v5litepod-8 push: branches: ["main"] pull_request: # By default this runs for types assigned, opened and synchronize. diff --git a/src/xpk/core/system_characteristics.py b/src/xpk/core/system_characteristics.py index 8808367b..15cd8825 100644 --- a/src/xpk/core/system_characteristics.py +++ b/src/xpk/core/system_characteristics.py @@ -1105,6 +1105,15 @@ def get_system_characteristics_by_device_type( 'v5p-17920', ), # v5litepod + 'v5litepod-8': SystemCharacteristics( + '2x4', + 1, + 'tpu-v5-lite-podslice', + 'ct5lp-hightpu-8t', + 8, + AcceleratorType['TPU'], + 'v5litepod-8', + ), 'v5litepod-16': SystemCharacteristics( '4x4', 4,