From 6dd5a5b420d57f7af2b386dbf07563ac8f03c5c8 Mon Sep 17 00:00:00 2001 From: Kent Russell Date: Thu, 3 Dec 2020 09:13:38 -0500 Subject: [PATCH] rocm-smi: Try find the librocm_smi64.so in a few locations Instead of looking solely in ../lib, try looking in any /opt folder as a backup option. This is a little more robust and hopefully leads to fewer issues trying to find the lib Change-Id: Ie0d3944b48b32d9965917e5c831388838b6d4ef7 [ROCm/rocm_smi_lib commit: c7b6b4721138656b50b8855312feb13114afb573] --- projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py b/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py index e86180be31..bbac48b5a5 100644 --- a/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py +++ b/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py @@ -11,6 +11,16 @@ import os # Use ROCm installation path if running from standard installation path_librocm = os.path.dirname(os.path.realpath(__file__)) + '/../lib/librocm_smi64.so' +if not os.path.isfile(path_librocm): + print('Unable to find %s . Trying /opt/rocm*' % path_librocm) + for root, dirs, files in os.walk('/opt', followlinks=True): + if 'librocm_smi64.so' in files: + path_librocm = os.path.join(os.path.realpath(root), 'librocm_smi64.so') + if os.path.isfile(path_librocm): + print('Using lib from %s' % path_librocm) + else: + print('Unable to find librocm_smi64.so') + # ----------> TODO: Support static libs as well as SO try: