xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 05:04:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BugFix][ROCm] Fix get_cu_count missing variable error (#28608)

Browse Source 
Signed-off-by: ganyi <ygan@amd.com>
This commit is contained in:
Pleaplusone 2025-11-13 13:18:56 +08:00 committed by GitHub
parent 1a0b157a2e
commit 7dca0c90cb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/utils/platform_utils.py
View File

@ -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