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/rocm_smi_lib commit: c7c2ac5559]
Этот коммит содержится в:
@@ -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):
|
||||
|
||||
Ссылка в новой задаче
Block a user