Show AMD GPU topology

vllm-project · Sep 20, 2024 · 70cd834 · 70cd834
1 parent 855c8ae
commit 70cd834
Showing 1 changed file with 7 additions and 2 deletions.
diff --git 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