From a0638d052db74ba28eada4768b9bbf98720b44a4 Mon Sep 17 00:00:00 2001 From: Andreas Karatzas Date: Fri, 13 Feb 2026 22:01:42 -0600 Subject: [PATCH] [Bugfix] Fix ROCm UVA CPU weight offloading broken by #32993 (#34543) Signed-off-by: Andreas Karatzas --- vllm/utils/torch_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/utils/torch_utils.py b/vllm/utils/torch_utils.py index 1bff517fd7b..fe047e0df5b 100644 --- a/vllm/utils/torch_utils.py +++ b/vllm/utils/torch_utils.py @@ -683,7 +683,7 @@ def get_accelerator_view_from_cpu_tensor(cpu_tensor: torch.Tensor) -> torch.Tens if current_platform.is_xpu(): assert cpu_tensor.is_pinned(), "CPU tensor must be pinned" return torch.ops._C.get_xpu_view_from_cpu_tensor(cpu_tensor) - elif current_platform.is_cuda(): + elif current_platform.is_cuda() or current_platform.is_rocm(): return torch.ops._C.get_cuda_view_from_cpu_tensor(cpu_tensor) else: raise ValueError(