[Misc] Show AMD GPU topology in `collect_env.py` (#8649)

035fa895 · Cyrus Leung · GitHub · b28298f2 · 035fa895
Unverified Commit 035fa895 authored Sep 21, 2024 by Cyrus Leung Committed by GitHub Sep 20, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 2 deletions

collect_env.py collect_env.py +7 -2

No files found.
--- a/collect_env.py
+++ b/collect_env.py
@@ -285,9 +285,14 @@ def summarize_vllm_build_flags():


 def get_gpu_topo(run_lambda):
+    output = None
+
    if get_platform() == 'linux':
-        return run_and_read_all(run_lambda, 'nvidia-smi topo -m')
-    return None
+        output = run_and_read_all(run_lambda, 'nvidia-smi topo -m')
+        if output is None:
+            output = run_and_read_all(run_lambda, 'rocm-smi --showtopo')
+
+    return output


 # example outputs of CPU infos