Skip to content

Commit 366a65d

Browse files
enhance GPU info (via rocm-smi): report card series in addition
1 parent 4f96bfb commit 366a65d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

easybuild/tools/systemtools.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -625,8 +625,9 @@ def get_gpu_info():
625625
out, ec = run_cmd(cmd, force_in_dry_run=True, trace=False, stream_output=False)
626626
if ec == 0:
627627
for line in out.strip().split('\n')[1:]:
628+
amd_card_series = line.split(',')[1]
628629
amd_card_model = line.split(',')[2]
629-
amd_gpu = ', '.join([amd_card_model, amd_driver])
630+
amd_gpu = "%s (model: %s, driver: %s)" % (amd_card_series, amd_card_model, amd_driver)
630631
amd_gpu_info = gpu_info.setdefault('AMD', {})
631632
amd_gpu_info.setdefault(amd_gpu, 0)
632633
amd_gpu_info[amd_gpu] += 1

0 commit comments

Comments
 (0)