From f7ef6c0d645c9e4f1995ad462d453b6a3694999b Mon Sep 17 00:00:00 2001 From: Akanksha Gupta Date: Tue, 5 May 2026 15:36:36 -0700 Subject: [PATCH] Fix the accelerator label for tpu7x PiperOrigin-RevId: 910952636 --- .../shared_pathways_service/deploy_pathways_service.py | 2 +- .../experimental/shared_pathways_service/validators.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pathwaysutils/experimental/shared_pathways_service/deploy_pathways_service.py b/pathwaysutils/experimental/shared_pathways_service/deploy_pathways_service.py index c18b120..34b154f 100644 --- a/pathwaysutils/experimental/shared_pathways_service/deploy_pathways_service.py +++ b/pathwaysutils/experimental/shared_pathways_service/deploy_pathways_service.py @@ -113,7 +113,7 @@ def get_tpu_config(tpu_type: str) -> TPUConfig: "tpu7x": TPUConfig( machine_type="tpu7x-standard-4t", chips_per_vm=4, - accelerator_label="tpu-v7-slice", + accelerator_label="tpu7x", instance_prefix="tpu7x", ), } diff --git a/pathwaysutils/experimental/shared_pathways_service/validators.py b/pathwaysutils/experimental/shared_pathways_service/validators.py index 818aaba..7b28440 100644 --- a/pathwaysutils/experimental/shared_pathways_service/validators.py +++ b/pathwaysutils/experimental/shared_pathways_service/validators.py @@ -68,7 +68,7 @@ def _validate_tpu_supported(tpu_instance_with_topology: str) -> None: # tpuv6e:2x4 -> type='tpuv6e', topology='2x4' # tpuv5:2x2x1 -> type='tpuv5', topology='2x2x1' match = re.match( - r"^(?:tpuv(?:5e|5|6e)):(?P\d+(?:x\d+){1,2})$", + r"^(?:tpu(?:v5e|v5|v6e|7x)):(?P\d+(?:x\d+){1,2})$", tpu_instance_with_topology, )