[Bugfix] Fix ROCm UVA CPU weight offloading broken by #32993 (#34543)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
2026-02-13 22:01:42 -06:00
parent c027541eaf
commit a0638d052d
1 changed files with 1 additions and 1 deletions
--- 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(