From b7df80c34b015c4c166a35730b46b314f38bbf43 Mon Sep 17 00:00:00 2001 From: Chris Freehill Date: Wed, 28 Oct 2020 16:54:58 -0500 Subject: [PATCH] Use relative path to find librocm_smi Change-Id: Ifca3f54d680a802c1c5fa360d17e64338b9ac9a8 [ROCm/rocm_smi_lib commit: 438d28612f7ab6349cec37321de0a3ae66a5b1da] --- projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py b/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py index 87f0bb951a..91b6751634 100644 --- a/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py +++ b/projects/rocm-smi-lib/python_smi_tools/rsmiBindings.py @@ -7,7 +7,10 @@ import ctypes.util from ctypes import * from enum import Enum -path_librocm = 'librocm_smi64.so' +import os + +# Use ROCm installation path if running from standard installation +path_librocm = os.path.dirname(os.path.realpath(__file__)) + '/../lib/librocm_smi64.so' # ----------> TODO: Support static libs as well as SO try: