AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Modify yaml configs to reflect new combined Top Stats

Browse Source 
Signed-off-by: colramos-amd <colramos@amd.com>
This commit is contained in:
colramos-amd
2024-02-13 19:56:46 -06:00
parent bd83f68412
commit b7d5d3b64b
20 changed files with 89 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/utils/specs.py
+4 -3
View File
@@ -230,9 +230,10 @@ def run(cmd,exit_on_error=False):
p = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
if exit_on_error:
if cmd[0] == "rocm-smi" and p.returncode != 0:
logging.error("ERROR: No GPU detected. Unable to load rocm-smi")
sys.exit(1)
if cmd[0] == "rocm-smi":
if p.returncode != 2 and p.returncode != 0:
logging.error("ERROR: No GPU detected. Unable to load rocm-smi")
sys.exit(1)
elif p.returncode != 0:
logging.error("ERROR: command [%s] failed with non-zero exit code" % cmd)
sys.exit(1)