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


[ROCm/amdsmi commit: c7c2ac5559]
Этот коммит содержится в:
Kent Russell
2021-04-13 08:00:17 -04:00
родитель f9cd4e6093
Коммит ffbe481241
+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):