xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-01-20 03:14:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

[platforms] improve rocm debugging info (#14257)

Browse Source
This commit is contained in:
youkaichao 2025-03-05 13:32:18 +08:00 committed by GitHub
parent 72c62eae5f
commit 6eaf93020d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/platforms/__init__.py
View File

@ -106,6 +106,9 @@ def rocm_platform_plugin() -> Optional[str]:
if len(amdsmi.amdsmi_get_processor_handles()) > 0:
is_rocm = True
logger.debug("Confirmed ROCm platform is available.")
else:
logger.debug("ROCm platform is not available because"
" no GPU is found.")
finally:
amdsmi.amdsmi_shut_down()
except Exception as e: