[TPU] Rename tpu_commons to tpu_inference (#26279)

Signed-off-by: Utkarsh Sharma <utksharma@google.com>
Co-authored-by: Utkarsh Sharma <utksharma@google.com>
Co-authored-by: Chengji Yao <chengjiyao@google.com>
This commit is contained in:
Utkarsh Sharma
2025-10-08 12:00:52 +05:30
committed by GitHub
parent 5e65d6b2ad
commit 335b28f7d1
6 changed files with 22 additions and 22 deletions

View File

@@ -223,9 +223,9 @@ class DefaultModelLoader(BaseModelLoader):
)
if current_platform.is_tpu():
from vllm.platforms.tpu import USE_TPU_COMMONS
from vllm.platforms.tpu import USE_TPU_INFERENCE
if not USE_TPU_COMMONS:
if not USE_TPU_INFERENCE:
# In PyTorch XLA, we should call `torch_xla.sync`
# frequently so that not too many ops are accumulated
# in the XLA program.