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: c9201f7736]
This commit is contained in:
Ranjith Ramakrishnan
2024-07-24 23:10:00 -07:00
کامیت شده توسط Dmitrii Galantsev
والد 1998b57059
کامیت 70b29ed8c3
@@ -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