Skip to content

Commit b39fc98

Browse files
DarkLight1337agt
authored andcommitted
[Misc] Show AMD GPU topology in collect_env.py (vllm-project#8649)
1 parent b594935 commit b39fc98

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

collect_env.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -285,9 +285,14 @@ def summarize_vllm_build_flags():
285285

286286

287287
def get_gpu_topo(run_lambda):
288+
output = None
289+
288290
if get_platform() == 'linux':
289-
return run_and_read_all(run_lambda, 'nvidia-smi topo -m')
290-
return None
291+
output = run_and_read_all(run_lambda, 'nvidia-smi topo -m')
292+
if output is None:
293+
output = run_and_read_all(run_lambda, 'rocm-smi --showtopo')
294+
295+
return output
291296

292297

293298
# example outputs of CPU infos

0 commit comments

Comments
 (0)