Use relative path to find librocm_smi

Change-Id: Ifca3f54d680a802c1c5fa360d17e64338b9ac9a8 [ROCm/rocm_smi_lib commit: 438d28612f]
2020-10-28 16:54:58 -05:00
parent c17f9e05e1
commit b7df80c34b
1 changed files with 4 additions and 1 deletions
@@ -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: