Support v5litepod-8.

AI-Hypercomputer · Dec 17, 2024 · 7d4453d · 7d4453d
1 parent cf24281
commit 7d4453d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git 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
@@ -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,