From 035fa895ecedea87810889aabbe50ba8a2ad7d5d Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Sat, 21 Sep 2024 04:52:19 +0800 Subject: [PATCH] [Misc] Show AMD GPU topology in `collect_env.py` (#8649) --- collect_env.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/collect_env.py b/collect_env.py index 839d5417..c5cd8c31 100644 --- 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