From 7dca0c90cbd221bdc2650a46ba04a288fc95fd94 Mon Sep 17 00:00:00 2001 From: Pleaplusone Date: Thu, 13 Nov 2025 13:18:56 +0800 Subject: [PATCH] [BugFix][ROCm] Fix `get_cu_count` missing variable error (#28608) Signed-off-by: ganyi --- vllm/utils/platform_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/utils/platform_utils.py b/vllm/utils/platform_utils.py index 3327fce6707b..433c6734e8a9 100644 --- a/vllm/utils/platform_utils.py +++ b/vllm/utils/platform_utils.py @@ -24,7 +24,7 @@ def xpu_is_initialized() -> bool: return torch.xpu.is_initialized() -def get_cu_count(cls, device_id: int = 0) -> int: +def get_cu_count(device_id: int = 0) -> int: """Returns the total number of compute units (CU) on single GPU.""" return torch.cuda.get_device_properties(device_id).multi_processor_count