Skip to content

GitLab

Unverified Commit 6eaf9302 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[platforms] improve rocm debugging info (#14257)

parent 72c62eae
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
vllm/platforms/__init__.py
View file @ 6eaf9302
...@@ -106,6 +106,9 @@ def rocm_platform_plugin() -> Optional[str]: ...@@ -106,6 +106,9 @@ def rocm_platform_plugin() -> Optional[str]:
if len(amdsmi.amdsmi_get_processor_handles()) > 0: if len(amdsmi.amdsmi_get_processor_handles()) > 0:
is_rocm = True is_rocm = True
logger.debug("Confirmed ROCm platform is available.") logger.debug("Confirmed ROCm platform is available.")
else:
logger.debug("ROCm platform is not available because"
" no GPU is found.")
finally: finally:
amdsmi.amdsmi_shut_down() amdsmi.amdsmi_shut_down()
except Exception as e: except Exception as e:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment