From 11b1e5fb42cebb244f82259e547dafcce1283b29 Mon Sep 17 00:00:00 2001 From: Maisam Arif Date: Thu, 2 Nov 2023 20:54:03 -0500 Subject: [PATCH] Handle GPU Not detected Signed-off-by: Maisam Arif Change-Id: I988b0dddde23b83c1a8e402233b6aa08ce55d0c7 [ROCm/amdsmi commit: 9bfebc6f1cf3cc0faecbbd769330e3655076b46e] --- projects/amdsmi/amdsmi_cli/amdsmi_commands.py | 5 +++++ projects/amdsmi/amdsmi_cli/amdsmi_helpers.py | 4 ++++ 2 files changed, 9 insertions(+) diff --git a/projects/amdsmi/amdsmi_cli/amdsmi_commands.py b/projects/amdsmi/amdsmi_cli/amdsmi_commands.py index cd88dfbd28..6c8e95398e 100644 --- a/projects/amdsmi/amdsmi_cli/amdsmi_commands.py +++ b/projects/amdsmi/amdsmi_cli/amdsmi_commands.py @@ -50,6 +50,11 @@ class AMDSMICommands(): sys.exit(-1) else: raise e + + if len(self.device_handles) == 0: + logging.error('Unable to detect any devices, check if driver is initialized (amdgpu not found in modules)') + sys.exit(-1) + self.stop = '' diff --git a/projects/amdsmi/amdsmi_cli/amdsmi_helpers.py b/projects/amdsmi/amdsmi_cli/amdsmi_helpers.py index cb5f7cf1bb..9e08bfbc70 100644 --- a/projects/amdsmi/amdsmi_cli/amdsmi_helpers.py +++ b/projects/amdsmi/amdsmi_cli/amdsmi_helpers.py @@ -154,6 +154,10 @@ class AMDSMIHelpers(): else: raise e + if len(device_handles) == 0: + logging.error('Unable to find any devices, check if driver is initialized (amdgpu not found in modules)') + sys.exit(-1) + # Handle spacing for the gpu_choices_str max_padding = int(math.log10(len(device_handles))) + 1