AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[SWDEV-553416] Fix amdsmi_get_gpu_reg_table_info and amdsmi_get_gpu_pm_metrics_info(#787)

Browse Source 
Signed-off-by: Narlo, Joseph <Joseph.Narlo@amd.com>
This commit is contained in:
Narlo, Joseph
2025-10-27 14:43:31 -05:00
committed by GitHub
parent 90f4b8c43d
commit ced7d12395
3 changed files with 122 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/python_unittest/integration_test.py
+8 -10
View File
@@ -27,24 +27,22 @@ import threading
import unittest
# Default path for AMDSMI_CLI_PATH is "/opt/rocm/libexec/amdsmi_cli/"
amdsmi_cli_path = os.environ.get("AMDSMI_CLI_PATH", "/opt/rocm/libexec/amdsmi_cli/")
if not os.path.exists(amdsmi_cli_path):
raise FileNotFoundError(f"AMDSMI_CLI_PATH '{amdsmi_cli_path}' does not exist. Please set the correct path in your environment.")
sys.path.append(amdsmi_cli_path)
amdsmi_path = os.environ.get("AMDSMI_PATH", "/opt/rocm/share/amd_smi")
if not os.path.exists(amdsmi_path):
raise FileNotFoundError(f"AMDSMI_PATH '{amdsmi_path}' does not exist. Please set the correct path in your environment.")
sys.path.append(amdsmi_path)
try:
import amdsmi, amdsmi.amdsmi_wrapper
except ImportError:
raise ImportError(f"Could not import the 'amdsmi' module from '{amdsmi_cli_path}'")
import amdsmi
except ImportError as exc:
raise ImportError(f'Could not import {amdsmi_path}') from exc
class TestAmdSmiInit(unittest.TestCase):
def test_init(self):
amdsmi.amdsmi_init()
amdsmi.amdsmi_shut_down()
class TestAmdSmiPythonInterface(unittest.TestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)