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