PyPI - xpk - Versions diffs - 0.10.1__py3-none-any.whl → 0.11.0__py3-none-any.whl - Mend

xpk 0.10.1py3-none-any.whl → 0.11.0py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (15) hide show

xpk/commands/cluster.py +263 -0
xpk/core/capacity.py +2 -2
xpk/core/config.py +1 -1
xpk/core/jobset.py +1 -1
xpk/core/kueue.py +6 -2
xpk/core/nap.py +16 -2
xpk/core/nodepool.py +52 -19
xpk/core/scheduling.py +1 -1
xpk/core/system_characteristics.py +266 -1080
{xpk-0.10.1.dist-info → xpk-0.11.0.dist-info}/METADATA +1 -1
{xpk-0.10.1.dist-info → xpk-0.11.0.dist-info}/RECORD +15 -15
{xpk-0.10.1.dist-info → xpk-0.11.0.dist-info}/WHEEL +0 -0
{xpk-0.10.1.dist-info → xpk-0.11.0.dist-info}/entry_points.txt +0 -0
{xpk-0.10.1.dist-info → xpk-0.11.0.dist-info}/licenses/LICENSE +0 -0
{xpk-0.10.1.dist-info → xpk-0.11.0.dist-info}/top_level.txt +0 -0

xpk/core/system_characteristics.py CHANGED Viewed

@@ -15,6 +15,8 @@ limitations under the License.
 """
 from dataclasses import dataclass
+from functools import reduce
+from operator import mul
 AcceleratorType = {'TPU': 1, 'GPU': 2, 'CPU': 3}
@@ -91,6 +93,34 @@ def get_system_characteristics_by_device_type(
     return None, 1
+def get_tpu_system_characteristics_map(
+    prefix: str,
+    tensorcores_per_chip: int,
+    gke_accelerator: str,
+    machine_type: str,
+    supported_topologies: list[str],
+) -> dict[str, SystemCharacteristics]:
+  system_characteristics_map = {}
+  for topology in supported_topologies:
+    total_chips = reduce(mul, (int(x) for x in topology.split('x')), 1)
+    num_tensorcores = total_chips * tensorcores_per_chip
+    chips_per_vm = 1 if total_chips == 1 else 4
+    vms_per_slice = total_chips // chips_per_vm
+    system = SystemCharacteristics(
+        topology,
+        vms_per_slice,
+        gke_accelerator,
+        machine_type,
+        chips_per_vm,
+        AcceleratorType['TPU'],
+        f'{prefix}-{num_tensorcores}',
+    )
+    system_characteristics_map[f'{prefix}-{topology}'] = system
+    system_characteristics_map[f'{prefix}-{num_tensorcores}'] = system
+  return system_characteristics_map
 ################### Subcommand Helper Functions #############################
 """ !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 IF YOU MODIFY THE BELOW UserFacingNameToSystemCharacteristics MAP YOU SHOULD
@@ -212,1098 +242,254 @@ UserFacingNameToSystemCharacteristics = {
         'h100-mega-80gb-8',
     ),
     # TPU system characteristics
-    # v6e
-    'v6e-1': SystemCharacteristics(
-        '1x1',
-        1,
-        'tpu-v6e-slice',
-        'ct6e-standard-1t',
-        1,
-        AcceleratorType['TPU'],
-        'v6e-1',
+    **get_tpu_system_characteristics_map(
+        'tpu7x', 2, 'tpu7x', 'tpu7x-standard-1t', ['1x1x1']
     ),
-    'v6e-4': SystemCharacteristics(
-        '2x2',
+    **get_tpu_system_characteristics_map(
+        'tpu7x',
+        2,
+        'tpu7x',
+        'tpu7x-standard-4t',
+        [
+            '12x12x12',
+            '12x12x16',
+            '12x12x20',
+            '12x12x24',
+            '12x12x28',
+            '12x12x36',
+            '12x16x16',
+            '12x16x20',
+            '12x16x24',
+            '12x16x28',
+            '12x20x20',
+            '12x20x24',
+            '12x24x24',
+            '16x16x16',
+            '16x16x20',
+            '16x16x24',
+            '16x16x32',
+            '16x20x28',
+            '16x24x24',
+            '2x2x1',
+            '2x2x2',
+            '2x2x4',
+            '2x4x4',
+            '4x12x116',
+            '4x12x12',
+            '4x12x124',
+            '4x12x20',
+            '4x12x28',
+            '4x12x44',
+            '4x12x52',
+            '4x12x68',
+            '4x12x76',
+            '4x12x92',
+            '4x20x20',
+            '4x20x28',
+            '4x20x44',
+            '4x20x52',
+            '4x20x68',
+            '4x20x76',
+            '4x28x28',
+            '4x28x44',
+            '4x28x52',
+            '4x4x116',
+            '4x4x12',
+            '4x4x124',
+            '4x4x148',
+            '4x4x164',
+            '4x4x172',
+            '4x4x188',
+            '4x4x20',
+            '4x4x212',
+            '4x4x236',
+            '4x4x244',
+            '4x4x28',
+            '4x4x4',
+            '4x4x44',
+            '4x4x52',
+            '4x4x68',
+            '4x4x76',
+            '4x4x8',
+            '4x4x92',
+            '4x8x116',
+            '4x8x12',
+            '4x8x124',
+            '4x8x148',
+            '4x8x164',
+            '4x8x172',
+            '4x8x188',
+            '4x8x20',
+            '4x8x28',
+            '4x8x44',
+            '4x8x52',
+            '4x8x68',
+            '4x8x76',
+            '4x8x8',
+            '4x8x92',
+            '8x12x12',
+            '8x12x16',
+            '8x12x20',
+            '8x12x28',
+            '8x12x44',
+            '8x12x52',
+            '8x16x16',
+            '8x16x20',
+            '8x16x28',
+            '8x16x44',
+            '8x20x20',
+            '8x20x28',
+            '8x8x12',
+            '8x8x16',
+            '8x8x20',
+            '8x8x28',
+            '8x8x44',
+            '8x8x52',
+            '8x8x68',
+            '8x8x76',
+            '8x8x8',
+            '8x8x92',
+        ],
+    ),
+    **get_tpu_system_characteristics_map(
+        'v6e', 1, 'tpu-v6e-slice', 'ct6e-standard-1t', ['1x1']
+    ),
+    **get_tpu_system_characteristics_map(
+        'v6e',
         1,
         'tpu-v6e-slice',
         'ct6e-standard-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v6e-4',
+        ['2x2', '2x4', '4x4', '4x8', '8x8', '8x16', '16x16'],
     ),
-    'v6e-8': SystemCharacteristics(
-        '2x4',
+    **get_tpu_system_characteristics_map(
+        'v5p',
         2,
-        'tpu-v6e-slice',
-        'ct6e-standard-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v6e-8',
-    ),
-    'v6e-16': SystemCharacteristics(
-        '4x4',
-        4,
-        'tpu-v6e-slice',
-        'ct6e-standard-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v6e-16',
-    ),
-    'v6e-32': SystemCharacteristics(
-        '4x8',
-        8,
-        'tpu-v6e-slice',
-        'ct6e-standard-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v6e-32',
-    ),
-    'v6e-64': SystemCharacteristics(
-        '8x8',
-        16,
-        'tpu-v6e-slice',
-        'ct6e-standard-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v6e-64',
-    ),
-    'v6e-128': SystemCharacteristics(
-        '8x16',
-        32,
-        'tpu-v6e-slice',
-        'ct6e-standard-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v6e-128',
-    ),
-    'v6e-256': SystemCharacteristics(
-        '16x16',
-        64,
-        'tpu-v6e-slice',
-        'ct6e-standard-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v6e-256',
-    ),
-    # v5p
-    'v5p-8': SystemCharacteristics(
-        '2x2x1',
-        1,
         'tpu-v5p-slice',
         'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-8',
+        [
+            '2x2x1',
+            '2x2x2',
+            '2x2x4',
+            '2x4x4',
+            '4x4x4',
+            '4x4x8',
+            '4x4x12',
+            '4x8x8',
+            '4x4x20',
+            '4x8x12',
+            '4x4x28',
+            '8x8x8',
+            '4x12x12',
+            '4x8x20',
+            '4x4x44',
+            '8x8x12',
+            '4x4x52',
+            '4x8x28',
+            '4x12x20',
+            '8x8x16',
+            '4x4x68',
+            '8x12x12',
+            '4x4x76',
+            '8x8x20',
+            '4x12x28',
+            '4x8x44',
+            '4x4x92',
+            '8x12x16',
+            '4x20x20',
+            '4x8x52',
+            '12x12x12',
+            '8x8x28',
+            '4x4x116',
+            '8x12x20',
+            '4x4x124',
+            '8x16x16',
+            '4x12x44',
+            '4x8x68',
+            '4x20x28',
+            '12x12x16',
+            '4x4x148',
+            '4x8x76',
+            '4x12x52',
+            '8x16x20',
+            '4x4x164',
+            '8x12x28',
+            '4x4x172',
+            '8x8x44',
+            '12x12x20',
+            '4x8x92',
+            '4x4x188',
+            '12x16x16',
+            '4x28x28',
+            '8x20x20',
+            '4x12x68',
+            '8x8x52',
+            '4x4x212',
+            '12x12x24',
+            '4x20x44',
+            '8x16x28',
+            '4x12x76',
+            '4x8x116',
+            '4x4x236',
+            '12x16x20',
+            '4x4x244',
+            '4x8x124',
+            '12x12x28',
+            '16x16x16',
+            '4x20x52',
+            '8x12x44',
+            '8x8x68',
+            '4x12x92',
+            '8x20x28',
+            '12x16x24',
+            '4x8x148',
+            '12x20x20',
+            '8x8x76',
+            '4x28x44',
+            '8x12x52',
+            '16x16x20',
+            '12x12x36',
+            '4x8x164',
+            '12x16x28',
+            '4x20x68',
+            '4x8x172',
+            '4x12x116',
+            '8x16x44',
+            '12x20x24',
+            '4x28x52',
+            '8x8x92',
+            '4x12x124',
+            '4x8x188',
+            '4x20x76',
+            '16x16x24',
+            '12x24x24',
+            '16x20x28',
+        ],
+    ),
+    **get_tpu_system_characteristics_map(
+        'v5litepod',
+        1,
+        'tpu-v5-lite-podslice',
+        'ct5lp-hightpu-4t',
+        ['2x4', '4x4', '4x8', '8x8', '8x16', '16x16'],
     ),
-    'v5p-16': SystemCharacteristics(
-        '2x2x2',
+    **get_tpu_system_characteristics_map(
+        'v4',
         2,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-16',
-    ),
-    'v5p-32': SystemCharacteristics(
-        '2x2x4',
-        4,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-32',
-    ),
-    'v5p-64': SystemCharacteristics(
-        '2x4x4',
-        8,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-64',
-    ),
-    'v5p-128': SystemCharacteristics(
-        '4x4x4',
-        16,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-128',
-    ),
-    'v5p-256': SystemCharacteristics(
-        '4x4x8',
-        32,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-256',
-    ),
-    'v5p-384': SystemCharacteristics(
-        '4x4x12',
-        48,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-384',
-    ),
-    'v5p-512': SystemCharacteristics(
-        '4x8x8',
-        64,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-512',
-    ),
-    'v5p-640': SystemCharacteristics(
-        '4x4x20',
-        80,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-640',
-    ),
-    'v5p-768': SystemCharacteristics(
-        '4x8x12',
-        96,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-768',
-    ),
-    'v5p-896': SystemCharacteristics(
-        '4x4x28',
-        112,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-896',
-    ),
-    'v5p-1024': SystemCharacteristics(
-        '8x8x8',
-        128,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-1024',
-    ),
-    'v5p-1152': SystemCharacteristics(
-        '4x12x12',
-        144,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-1152',
-    ),
-    'v5p-1280': SystemCharacteristics(
-        '4x8x20',
-        160,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-1280',
-    ),
-    'v5p-1408': SystemCharacteristics(
-        '4x4x44',
-        176,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-1408',
-    ),
-    'v5p-1536': SystemCharacteristics(
-        '8x8x12',
-        192,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-1536',
-    ),
-    'v5p-1664': SystemCharacteristics(
-        '4x4x52',
-        208,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-1664',
-    ),
-    'v5p-1792': SystemCharacteristics(
-        '4x8x28',
-        224,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-1792',
-    ),
-    'v5p-1920': SystemCharacteristics(
-        '4x12x20',
-        240,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-1920',
-    ),
-    'v5p-2048': SystemCharacteristics(
-        '8x8x16',
-        256,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-2048',
-    ),
-    'v5p-2176': SystemCharacteristics(
-        '4x4x68',
-        272,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-2176',
-    ),
-    'v5p-2304': SystemCharacteristics(
-        '8x12x12',
-        288,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-2304',
-    ),
-    'v5p-2432': SystemCharacteristics(
-        '4x4x76',
-        304,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-2432',
-    ),
-    'v5p-2560': SystemCharacteristics(
-        '8x8x20',
-        320,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-2560',
-    ),
-    'v5p-2688': SystemCharacteristics(
-        '4x12x28',
-        336,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-2688',
-    ),
-    'v5p-2816': SystemCharacteristics(
-        '4x8x44',
-        352,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-2816',
-    ),
-    'v5p-2944': SystemCharacteristics(
-        '4x4x92',
-        368,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-2944',
-    ),
-    'v5p-3072': SystemCharacteristics(
-        '8x12x16',
-        384,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-3072',
-    ),
-    'v5p-3200': SystemCharacteristics(
-        '4x20x20',
-        400,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-3200',
-    ),
-    'v5p-3328': SystemCharacteristics(
-        '4x8x52',
-        416,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-3328',
-    ),
-    'v5p-3456': SystemCharacteristics(
-        '12x12x12',
-        432,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-3456',
-    ),
-    'v5p-3584': SystemCharacteristics(
-        '8x8x28',
-        448,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-3584',
-    ),
-    'v5p-3712': SystemCharacteristics(
-        '4x4x116',
-        464,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-3712',
-    ),
-    'v5p-3840': SystemCharacteristics(
-        '8x12x20',
-        480,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-3840',
-    ),
-    'v5p-3968': SystemCharacteristics(
-        '4x4x124',
-        496,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-3968',
-    ),
-    'v5p-4096': SystemCharacteristics(
-        '8x16x16',
-        512,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-4096',
-    ),
-    'v5p-4224': SystemCharacteristics(
-        '4x12x44',
-        528,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-4224',
-    ),
-    'v5p-4352': SystemCharacteristics(
-        '4x8x68',
-        544,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-4352',
-    ),
-    'v5p-4480': SystemCharacteristics(
-        '4x20x28',
-        560,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-4480',
-    ),
-    'v5p-4608': SystemCharacteristics(
-        '12x12x16',
-        576,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-4608',
-    ),
-    'v5p-4736': SystemCharacteristics(
-        '4x4x148',
-        592,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-4736',
-    ),
-    'v5p-4864': SystemCharacteristics(
-        '4x8x76',
-        608,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-4864',
-    ),
-    'v5p-4992': SystemCharacteristics(
-        '4x12x52',
-        624,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-4992',
-    ),
-    'v5p-5120': SystemCharacteristics(
-        '8x16x20',
-        640,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-5120',
-    ),
-    'v5p-5248': SystemCharacteristics(
-        '4x4x164',
-        656,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-5248',
-    ),
-    'v5p-5376': SystemCharacteristics(
-        '8x12x28',
-        672,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-5376',
-    ),
-    'v5p-5504': SystemCharacteristics(
-        '4x4x172',
-        688,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-5504',
-    ),
-    'v5p-5632': SystemCharacteristics(
-        '8x8x44',
-        704,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-5632',
-    ),
-    'v5p-5760': SystemCharacteristics(
-        '12x12x20',
-        720,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-5760',
-    ),
-    'v5p-5888': SystemCharacteristics(
-        '4x8x92',
-        736,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-5888',
-    ),
-    'v5p-6016': SystemCharacteristics(
-        '4x4x188',
-        752,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-6016',
-    ),
-    'v5p-6144': SystemCharacteristics(
-        '12x16x16',
-        768,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-6144',
-    ),
-    'v5p-6272': SystemCharacteristics(
-        '4x28x28',
-        784,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-6272',
-    ),
-    'v5p-6400': SystemCharacteristics(
-        '8x20x20',
-        800,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-6400',
-    ),
-    'v5p-6528': SystemCharacteristics(
-        '4x12x68',
-        816,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-6528',
-    ),
-    'v5p-6656': SystemCharacteristics(
-        '8x8x52',
-        832,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-6656',
-    ),
-    'v5p-6784': SystemCharacteristics(
-        '4x4x212',
-        848,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-6784',
-    ),
-    'v5p-6912': SystemCharacteristics(
-        '12x12x24',
-        864,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-6912',
-    ),
-    'v5p-7040': SystemCharacteristics(
-        '4x20x44',
-        880,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-7040',
-    ),
-    'v5p-7168': SystemCharacteristics(
-        '8x16x28',
-        896,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-7168',
-    ),
-    'v5p-7296': SystemCharacteristics(
-        '4x12x76',
-        912,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-7296',
-    ),
-    'v5p-7424': SystemCharacteristics(
-        '4x8x116',
-        928,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-7424',
-    ),
-    'v5p-7552': SystemCharacteristics(
-        '4x4x236',
-        944,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-7552',
-    ),
-    'v5p-7680': SystemCharacteristics(
-        '12x16x20',
-        960,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-7680',
-    ),
-    'v5p-7808': SystemCharacteristics(
-        '4x4x244',
-        976,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-7808',
-    ),
-    'v5p-7936': SystemCharacteristics(
-        '4x8x124',
-        992,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-7936',
-    ),
-    'v5p-8064': SystemCharacteristics(
-        '12x12x28',
-        1008,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-8064',
-    ),
-    'v5p-8192': SystemCharacteristics(
-        '16x16x16',
-        1024,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-8192',
-    ),
-    'v5p-8320': SystemCharacteristics(
-        '4x20x52',
-        1040,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-8320',
-    ),
-    'v5p-8448': SystemCharacteristics(
-        '8x12x44',
-        1056,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-8448',
-    ),
-    'v5p-8704': SystemCharacteristics(
-        '8x8x68',
-        1088,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-8704',
-    ),
-    'v5p-8832': SystemCharacteristics(
-        '4x12x92',
-        1104,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-8832',
-    ),
-    'v5p-8960': SystemCharacteristics(
-        '8x20x28',
-        1120,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-8960',
-    ),
-    'v5p-9216': SystemCharacteristics(
-        '12x16x24',
-        1152,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-9216',
-    ),
-    'v5p-9472': SystemCharacteristics(
-        '4x8x148',
-        1184,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-9472',
-    ),
-    'v5p-9600': SystemCharacteristics(
-        '12x20x20',
-        1200,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-9600',
-    ),
-    'v5p-9728': SystemCharacteristics(
-        '8x8x76',
-        1216,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-9728',
-    ),
-    'v5p-9856': SystemCharacteristics(
-        '4x28x44',
-        1232,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-9856',
-    ),
-    'v5p-9984': SystemCharacteristics(
-        '8x12x52',
-        1248,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-9984',
-    ),
-    'v5p-10240': SystemCharacteristics(
-        '16x16x20',
-        1280,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-10240',
-    ),
-    'v5p-10368': SystemCharacteristics(
-        '12x12x36',
-        1296,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-10368',
-    ),
-    'v5p-10496': SystemCharacteristics(
-        '4x8x164',
-        1312,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-10496',
-    ),
-    'v5p-10752': SystemCharacteristics(
-        '12x16x28',
-        1344,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-10752',
-    ),
-    'v5p-10880': SystemCharacteristics(
-        '4x20x68',
-        1360,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-10880',
-    ),
-    'v5p-11008': SystemCharacteristics(
-        '4x8x172',
-        1376,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-11008',
-    ),
-    'v5p-11136': SystemCharacteristics(
-        '4x12x116',
-        1392,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-11136',
-    ),
-    'v5p-11264': SystemCharacteristics(
-        '8x16x44',
-        1408,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-11264',
-    ),
-    'v5p-11520': SystemCharacteristics(
-        '12x20x24',
-        1440,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-11520',
-    ),
-    'v5p-11648': SystemCharacteristics(
-        '4x28x52',
-        1456,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-11648',
-    ),
-    'v5p-11776': SystemCharacteristics(
-        '8x8x92',
-        1472,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-11776',
-    ),
-    'v5p-11904': SystemCharacteristics(
-        '4x12x124',
-        1488,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-11904',
-    ),
-    'v5p-12032': SystemCharacteristics(
-        '4x8x188',
-        1504,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-12032',
-    ),
-    'v5p-12160': SystemCharacteristics(
-        '4x20x76',
-        1520,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-12160',
-    ),
-    'v5p-12288': SystemCharacteristics(
-        '16x16x24',
-        1536,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-12288',
-    ),
-    'v5p-13824': SystemCharacteristics(
-        '12x24x24',
-        1728,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-13824',
-    ),
-    'v5p-17920': SystemCharacteristics(
-        '16x20x28',
-        2240,
-        'tpu-v5p-slice',
-        'ct5p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5p-17920',
-    ),
-    # v5litepod
-    'v5litepod-8': SystemCharacteristics(
-        '2x4',
-        2,
-        'tpu-v5-lite-podslice',
-        'ct5lp-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5litepod-8',
-    ),
-    'v5litepod-16': SystemCharacteristics(
-        '4x4',
-        4,
-        'tpu-v5-lite-podslice',
-        'ct5lp-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5litepod-16',
-    ),
-    'v5litepod-32': SystemCharacteristics(
-        '4x8',
-        8,
-        'tpu-v5-lite-podslice',
-        'ct5lp-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5litepod-32',
-    ),
-    'v5litepod-64': SystemCharacteristics(
-        '8x8',
-        16,
-        'tpu-v5-lite-podslice',
-        'ct5lp-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5litepod-64',
-    ),
-    'v5litepod-128': SystemCharacteristics(
-        '8x16',
-        32,
-        'tpu-v5-lite-podslice',
-        'ct5lp-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5litepod-128',
-    ),
-    'v5litepod-256': SystemCharacteristics(
-        '16x16',
-        64,
-        'tpu-v5-lite-podslice',
-        'ct5lp-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v5litepod-256',
-    ),
-    # v4
-    'v4-8': SystemCharacteristics(
-        '2x2x1',
-        1,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-8',
-    ),
-    'v4-16': SystemCharacteristics(
-        '2x2x2',
-        2,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-16',
-    ),
-    'v4-32': SystemCharacteristics(
-        '2x2x4',
-        4,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-32',
-    ),
-    'v4-64': SystemCharacteristics(
-        '2x4x4',
-        8,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-64',
-    ),
-    'v4-128': SystemCharacteristics(
-        '4x4x4',
-        16,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-128',
-    ),
-    'v4-256': SystemCharacteristics(
-        '4x4x8',
-        32,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-256',
-    ),
-    'v4-512': SystemCharacteristics(
-        '4x8x8',
-        64,
         'tpu-v4-podslice',
         'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-512',
-    ),
-    'v4-1024': SystemCharacteristics(
-        '8x8x8',
-        128,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-1024',
-    ),
-    'v4-1536': SystemCharacteristics(
-        '8x8x12',
-        192,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-1536',
-    ),
-    'v4-2048': SystemCharacteristics(
-        '8x8x16',
-        256,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-2048',
-    ),
-    'v4-4096': SystemCharacteristics(
-        '8x16x16',
-        512,
-        'tpu-v4-podslice',
-        'ct4p-hightpu-4t',
-        4,
-        AcceleratorType['TPU'],
-        'v4-4096',
+        [
+            '2x2x1',
+            '2x2x2',
+            '2x2x4',
+            '2x4x4',
+            '4x4x4',
+            '4x4x8',
+            '4x8x8',
+            '8x8x8',
+            '8x8x12',
+            '8x8x16',
+            '8x16x16',
+        ],
     ),
     # CPU system characteristics.
     # Note that chips_per_vm is actually the number of vCPUs in that CPU.

xpk 0.10.1__py3-none-any.whl → 0.11.0__py3-none-any.whl

xpk 0.10.1py3-none-any.whl → 0.11.0py3-none-any.whl