(run_lambda)
| 299 | |
| 300 | |
| 301 | def get_gpu_topo(run_lambda): |
| 302 | output = None |
| 303 | |
| 304 | if get_platform() == "linux": |
| 305 | output = run_and_read_all(run_lambda, "nvidia-smi topo -m") |
| 306 | if output is None: |
| 307 | output = run_and_read_all(run_lambda, "rocm-smi --showtopo") |
| 308 | |
| 309 | return output |
| 310 | |
| 311 | |
| 312 | # example outputs of CPU infos |
no test coverage detected