From c4eaa1d45b7a27f0a5c8bbd09cf0b3c1f9825ab7 Mon Sep 17 00:00:00 2001 From: coleramos425 Date: Tue, 21 Feb 2023 12:40:41 -0600 Subject: [PATCH] Specify returncode for rocm-smi error (#85) Signed-off-by: coleramos425 --- src/utils/specs.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/utils/specs.py b/src/utils/specs.py index 1bd8b0b5ed..2ba8484730 100644 --- a/src/utils/specs.py +++ b/src/utils/specs.py @@ -148,6 +148,9 @@ def gpuinfo(): def run(cmd): p = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE) + if cmd[0] == 'rocm-smi' and p.returncode == 8: + print("ERROR: No GPU detected. Unable to load rocm-smi") + sys.exit(1) return p.stdout.decode("ascii") @@ -216,10 +219,6 @@ 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()