From 22485bf114cd6b8330c8efb0a05baf1234cd3ea7 Mon Sep 17 00:00:00 2001 From: Kent Russell Date: Wed, 28 Apr 2021 07:43:07 -0400 Subject: [PATCH] rocm_smi.py: Fix gpu reset error Since device is a list, we need to pass a single item to the isAmdGpu function. Fixes: 17bdc065a19d "rocm_smi.py: Don't try to reset non-AMD GPUs" Signed-off-by: Kent Russell Change-Id: I19a74377636ff4589f11d092f41e1d35c1acb307 [ROCm/rocm_smi_lib commit: 242d94a668e40ad77d47688d6b6787aed269de41] --- projects/rocm-smi-lib/python_smi_tools/rocm_smi.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py b/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py index 05cfd82645..c96d998f85 100755 --- a/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py +++ b/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py @@ -961,11 +961,11 @@ def resetGpu(device): logging.error('GPU Reset can only be performed on one GPU per call') RETCODE = 1 return - if not isAmdDevice(device): + resetDev = int(device[0]) + if not isAmdDevice(resetDev): logging.error('GPU Reset can only be performed on an AMD GPU') RETCODE = 1 return - resetDev = int(device[0]) ret = rocmsmi.rsmi_dev_gpu_reset(resetDev) if rsmi_ret_ok(ret, resetDev): printLog(resetDev, 'Successfully reset GPU %d' % (resetDev), None)