From 70b29ed8c39d2b1b65973483662641d8c3f91c01 Mon Sep 17 00:00:00 2001
From: Ranjith Ramakrishnan <Ranjith.Ramakrishnan@amd.com>
Date: Wed, 24 Jul 2024 23:10:00 -0700
Subject: [PATCH] SWDEV-469004 - Append additonal path to system path

rocm-smi is installed in /opt/rocm-ver/bin , but not as a soft link in wheel package
For rocm-smi to work from bin directory, it need the extra path to find rsmiBindings.py

Change-Id: I41388f680cb2ab9f11dc135639b0d30b66082392


[ROCm/rocm_smi_lib commit: c9201f7736a10020e7b6052ea7f5cb0e991c51c2]
---
 .../rocm-smi-lib/python_smi_tools/rocm_smi.py | 23 ++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py b/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py
index e832ab1152..841355bf4a 100755
--- a/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py
+++ b/projects/rocm-smi-lib/python_smi_tools/rocm_smi.py
@@ -24,7 +24,24 @@ import trace
 from io import StringIO
 from time import ctime
 from subprocess import check_output
-from rsmiBindings import *
+from typing import TYPE_CHECKING
+
+# only used for type checking
+# pyright trips up and cannot find rsmiBindings without it
+if TYPE_CHECKING:
+    from rsmiBindings import *
+
+try:
+    from rsmiBindings import *
+except ImportError:
+    current_path = os.path.dirname(os.path.abspath(__file__))
+    additional_path = f"{current_path}/../libexec/rocm_smi"
+    sys.path.append(additional_path)
+    try:
+        from rsmiBindings import *
+    except ImportError:
+        print(f"Still couldn't import 'rsmiBindings'. Make sure it's installed in {additional_path}")
+        sys.exit(1)
 
 # rocmSmiLib_cli version. Increment this as needed.
 # Major version - Increment when backwards-compatibility breaks
@@ -196,7 +213,7 @@ def getBus(device, silent=False):
     # BDFID = ((DOMAIN & 0xFFFFFFFF) << 32) | ((PARTITION_ID & 0xF) << 28) | ((BUS & 0xFF) << 8) |
     # ((DEVICE & 0x1F) <<3 ) | (FUNCTION & 0x7)
     # bits [63:32] = domain
-    # bits [31:28] or bits [2:0] = partition id 
+    # bits [31:28] or bits [2:0] = partition id
     # bits [27:16] = reserved
     # bits [15:8]  = Bus
     # bits [7:3] = Device
@@ -223,7 +240,7 @@ def getPartitionId(device, silent=False):
     # BDFID = ((DOMAIN & 0xFFFFFFFF) << 32) | ((PARTITION_ID & 0xF) << 28) | ((BUS & 0xFF) << 8) |
     # ((DEVICE & 0x1F) <<3 ) | (FUNCTION & 0x7)
     # bits [63:32] = domain
-    # bits [31:28] or bits [2:0] = partition id 
+    # bits [31:28] or bits [2:0] = partition id
     # bits [27:16] = reserved
     # bits [15:8]  = Bus
     # bits [7:3] = Device