From b0bcc846aa0a8b0cc9f555348e953718cdf64944 Mon Sep 17 00:00:00 2001 From: coleramos425 Date: Tue, 21 Feb 2023 09:47:45 -0600 Subject: [PATCH] Catch exceptions from rocm-smi (#85) Signed-off-by: coleramos425 [ROCm/rocprofiler-compute commit: f77173a2f1c1761829f91295e76d838c24be6685] --- projects/rocprofiler-compute/src/utils/specs.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/projects/rocprofiler-compute/src/utils/specs.py b/projects/rocprofiler-compute/src/utils/specs.py index a040fcbf68..1bd8b0b5ed 100644 --- a/projects/rocprofiler-compute/src/utils/specs.py +++ b/projects/rocprofiler-compute/src/utils/specs.py @@ -216,6 +216,10 @@ def get_machine_specs(devicenum): ) = gpuinfo() rocm_smi = run(["rocm-smi"]) + if search(r"ERROR(\S*)", rocm_smi) != None: + print("ERROR: No GPU detected. Unable to load rocm-smi") + sys.exit(1) + device = rf"^\s*{devicenum}(.*)" hostname = socket.gethostname()