rocm_smi.py: Don't try to reset non-AMD GPUs
This won't work for obvious reasons, so exit with an error instead of trying to access a file that doesn't exist and segfaulting Change-Id: Id1230922fa6e9a19e9394280faad88a43c7d2e34
Этот коммит содержится в:
1 родитель
b016a8269a
Коммит
c7c2ac5559
1 file changed
+4
@@ -941,6 +941,10 @@ def resetGpu(device):
|
||||
logging.error('GPU Reset can only be performed on one GPU per call')
|
||||
RETCODE = 1
|
||||
return
|
||||
if not isAmdDevice(device):
|
||||
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):
|
||||
|
||||
Reference in new issue
Block a user