[SWDEV-451960] [WIP] Add Pytest
Updates: - Added pytest to shared/pytest folder - User can execute tests: [pytest] python3 -m pytest -p no:cacheprovider /opt/rocm/share/amd_smi/tests/pytest/unit_tests.py -s -v python3 -m pytest -p no:cacheprovider /opt/rocm/share/amd_smi/tests/pytest/integration_test.py -s -v [unittest] /opt/rocm/share/amd_smi/tests/pytest/unit_tests.py -v /opt/rocm/share/amd_smi/tests/pytest/integration_test.py -v - Automatically installs pytest Change-Id: Ia3281a9608aeeb803b91f8b83f87ff84b01037f4 Signed-off-by: Charis Poag <Charis.Poag@amd.com>
This commit is contained in:
Charis Poag
Ryo Ficano
@@ -4,6 +4,32 @@ Full documentation for amd_smi_lib is available at [https://rocm.docs.amd.com/pr
|
|||||||
|
|
||||||
***All information listed below is for reference and subject to change.***
|
***All information listed below is for reference and subject to change.***
|
||||||
|
|
||||||
|
## amd_smi_lib for ROCm 6.3.0
|
||||||
|
|
||||||
|
### Changes
|
||||||
|
|
||||||
|
- **Added Pytest functionality to test amdsmi API calls in Python**.
|
||||||
|
|
||||||
|
### Removals
|
||||||
|
|
||||||
|
- N/A
|
||||||
|
|
||||||
|
### Optimizations
|
||||||
|
|
||||||
|
- N/A
|
||||||
|
|
||||||
|
### Resolved issues
|
||||||
|
|
||||||
|
- N/A
|
||||||
|
|
||||||
|
### Known issues
|
||||||
|
|
||||||
|
- N/A
|
||||||
|
|
||||||
|
### Upcoming changes
|
||||||
|
|
||||||
|
- N/A
|
||||||
|
|
||||||
## amd_smi_lib for ROCm 6.2.1
|
## amd_smi_lib for ROCm 6.2.1
|
||||||
|
|
||||||
### Additions
|
### Additions
|
||||||
|
|||||||
+2
-1
@@ -187,9 +187,10 @@ if(BUILD_TESTS)
|
|||||||
add_subdirectory("tests/amd_smi_test")
|
add_subdirectory("tests/amd_smi_test")
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
# python interface and CLI depend on shared libraries
|
# python interface, CLI, and py-test depend on shared libraries
|
||||||
if(BUILD_SHARED_LIBS)
|
if(BUILD_SHARED_LIBS)
|
||||||
add_subdirectory("py-interface")
|
add_subdirectory("py-interface")
|
||||||
|
add_subdirectory("pytest")
|
||||||
if(BUILD_CLI)
|
if(BUILD_CLI)
|
||||||
add_subdirectory("amdsmi_cli")
|
add_subdirectory("amdsmi_cli")
|
||||||
endif()
|
endif()
|
||||||
|
|||||||
@@ -2,6 +2,7 @@
|
|||||||
|
|
||||||
do_updatepciids() {
|
do_updatepciids() {
|
||||||
update-pciids >/dev/null 2>&1 || true
|
update-pciids >/dev/null 2>&1 || true
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
do_configureLogrotate() {
|
do_configureLogrotate() {
|
||||||
@@ -187,9 +188,22 @@ do_install_amdsmi_python_lib() {
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
|
do_install_amdsmi_pytest() {
|
||||||
|
echo -n "Installing pytest... "
|
||||||
|
pip install -U pytest >/dev/null 2>&1
|
||||||
|
if [ $? -ne 0 ]; then
|
||||||
|
echo "[WARNING] Detected pytest could not be installed. Running pytest may not work as documented."
|
||||||
|
else
|
||||||
|
echo -n "[SUCCESS]"
|
||||||
|
echo ""
|
||||||
|
fi
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
case "$1" in
|
case "$1" in
|
||||||
( configure )
|
( configure )
|
||||||
do_install_amdsmi_python_lib
|
do_install_amdsmi_python_lib
|
||||||
|
do_install_amdsmi_pytest
|
||||||
do_ldconfig
|
do_ldconfig
|
||||||
do_updatepciids
|
do_updatepciids
|
||||||
do_configureLogrotate || exit 0
|
do_configureLogrotate || exit 0
|
||||||
|
|||||||
@@ -2,6 +2,7 @@
|
|||||||
|
|
||||||
do_updatepciids() {
|
do_updatepciids() {
|
||||||
update-pciids >/dev/null 2>&1 || true
|
update-pciids >/dev/null 2>&1 || true
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
do_configureLogrotate() {
|
do_configureLogrotate() {
|
||||||
@@ -186,9 +187,22 @@ do_install_amdsmi_python_lib() {
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
|
do_install_amdsmi_pytest() {
|
||||||
|
echo -n "Installing pytest... "
|
||||||
|
pip install -U pytest >/dev/null 2>&1
|
||||||
|
if [ $? -ne 0 ]; then
|
||||||
|
echo "[WARNING] Detected pytest could not be installed. Running pytest may not work as documented."
|
||||||
|
else
|
||||||
|
echo -n "[SUCCESS]"
|
||||||
|
echo ""
|
||||||
|
fi
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
# post install or upgrade, $i is 1 or 2 -> do these actions
|
# post install or upgrade, $i is 1 or 2 -> do these actions
|
||||||
if [ "$1" -ge 1 ]; then
|
if [ "$1" -ge 1 ]; then
|
||||||
do_install_amdsmi_python_lib
|
do_install_amdsmi_python_lib
|
||||||
|
do_install_amdsmi_pytest
|
||||||
do_ldconfig
|
do_ldconfig
|
||||||
do_updatepciids
|
do_updatepciids
|
||||||
do_configureLogrotate || exit 0
|
do_configureLogrotate || exit 0
|
||||||
|
|||||||
@@ -590,6 +590,26 @@ def _padHexValue(value, length):
|
|||||||
return '0x' + value[2:].zfill(length)
|
return '0x' + value[2:].zfill(length)
|
||||||
return value
|
return value
|
||||||
|
|
||||||
|
class UIntegerTypes(IntEnum):
|
||||||
|
UINT8_T = 0xFF
|
||||||
|
UINT16_T = 0xFFFF
|
||||||
|
UINT32_T = 0xFFFFFFFF
|
||||||
|
UINT64_T = 0xFFFFFFFFFFFFFFFF
|
||||||
|
|
||||||
|
def _validateIfMaxUint(valToCheck, uintType: UIntegerTypes):
|
||||||
|
return_val = "N/A"
|
||||||
|
if not isinstance(valToCheck, list):
|
||||||
|
if valToCheck == uintType:
|
||||||
|
return return_val
|
||||||
|
else:
|
||||||
|
return valToCheck
|
||||||
|
else:
|
||||||
|
return_val = valToCheck
|
||||||
|
for idx, v in enumerate(valToCheck):
|
||||||
|
if v == uintType:
|
||||||
|
return_val[idx] = "N/A"
|
||||||
|
return return_val
|
||||||
|
|
||||||
|
|
||||||
def amdsmi_get_socket_handles() -> List[amdsmi_wrapper.amdsmi_socket_handle]:
|
def amdsmi_get_socket_handles() -> List[amdsmi_wrapper.amdsmi_socket_handle]:
|
||||||
"""
|
"""
|
||||||
@@ -2277,31 +2297,23 @@ def amdsmi_get_pcie_info(
|
|||||||
|
|
||||||
pcie_info_dict = {
|
pcie_info_dict = {
|
||||||
"pcie_static": {
|
"pcie_static": {
|
||||||
"max_pcie_width": pcie_info.pcie_static.max_pcie_width,
|
"max_pcie_width": _validateIfMaxUint(pcie_info.pcie_static.max_pcie_width, UIntegerTypes.UINT16_T),
|
||||||
"max_pcie_speed": pcie_info.pcie_static.max_pcie_speed,
|
"max_pcie_speed": _validateIfMaxUint(pcie_info.pcie_static.max_pcie_speed, UIntegerTypes.UINT32_T),
|
||||||
"pcie_interface_version": pcie_info.pcie_static.pcie_interface_version,
|
"pcie_interface_version": _validateIfMaxUint(pcie_info.pcie_static.pcie_interface_version, UIntegerTypes.UINT32_T),
|
||||||
"slot_type": pcie_info.pcie_static.slot_type,
|
"slot_type": pcie_info.pcie_static.slot_type,
|
||||||
},
|
},
|
||||||
"pcie_metric": {
|
"pcie_metric": {
|
||||||
"pcie_width": pcie_info.pcie_metric.pcie_width,
|
"pcie_width": _validateIfMaxUint(pcie_info.pcie_metric.pcie_width, UIntegerTypes.UINT16_T),
|
||||||
"pcie_speed": pcie_info.pcie_metric.pcie_speed,
|
"pcie_speed": _validateIfMaxUint(pcie_info.pcie_metric.pcie_speed, UIntegerTypes.UINT32_T),
|
||||||
"pcie_bandwidth": pcie_info.pcie_metric.pcie_bandwidth,
|
"pcie_bandwidth": _validateIfMaxUint(pcie_info.pcie_metric.pcie_bandwidth, UIntegerTypes.UINT32_T),
|
||||||
"pcie_replay_count": pcie_info.pcie_metric.pcie_replay_count,
|
"pcie_replay_count": _validateIfMaxUint(pcie_info.pcie_metric.pcie_replay_count, UIntegerTypes.UINT64_T),
|
||||||
"pcie_l0_to_recovery_count": pcie_info.pcie_metric.pcie_l0_to_recovery_count,
|
"pcie_l0_to_recovery_count": _validateIfMaxUint(pcie_info.pcie_metric.pcie_l0_to_recovery_count, UIntegerTypes.UINT64_T),
|
||||||
"pcie_replay_roll_over_count": pcie_info.pcie_metric.pcie_replay_roll_over_count,
|
"pcie_replay_roll_over_count": _validateIfMaxUint(pcie_info.pcie_metric.pcie_replay_roll_over_count, UIntegerTypes.UINT64_T),
|
||||||
"pcie_nak_sent_count": pcie_info.pcie_metric.pcie_nak_sent_count,
|
"pcie_nak_sent_count": _validateIfMaxUint(pcie_info.pcie_metric.pcie_nak_sent_count, UIntegerTypes.UINT64_T),
|
||||||
"pcie_nak_received_count": pcie_info.pcie_metric.pcie_nak_received_count,
|
"pcie_nak_received_count": _validateIfMaxUint(pcie_info.pcie_metric.pcie_nak_received_count, UIntegerTypes.UINT64_T),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
# Check pcie static values for uint max
|
|
||||||
if pcie_info_dict['pcie_static']['max_pcie_width'] == 0xFFFF:
|
|
||||||
pcie_info_dict['pcie_static']['max_pcie_width'] = "N/A"
|
|
||||||
if pcie_info_dict['pcie_static']['max_pcie_speed'] == 0xFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_static']['max_pcie_speed'] = "N/A"
|
|
||||||
if pcie_info_dict['pcie_static']['pcie_interface_version'] == 0xFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_static']['pcie_interface_version'] = "N/A"
|
|
||||||
|
|
||||||
slot_type = pcie_info_dict['pcie_static']['slot_type']
|
slot_type = pcie_info_dict['pcie_static']['slot_type']
|
||||||
if isinstance(slot_type, int):
|
if isinstance(slot_type, int):
|
||||||
slot_types = amdsmi_wrapper.amdsmi_card_form_factor_t__enumvalues
|
slot_types = amdsmi_wrapper.amdsmi_card_form_factor_t__enumvalues
|
||||||
@@ -2312,29 +2324,6 @@ def amdsmi_get_pcie_info(
|
|||||||
else:
|
else:
|
||||||
pcie_info_dict['pcie_static']['slot_type'] = "N/A"
|
pcie_info_dict['pcie_static']['slot_type'] = "N/A"
|
||||||
|
|
||||||
# Check pcie metric values for uint max
|
|
||||||
if pcie_info_dict['pcie_metric']['pcie_width'] == 0xFFFF:
|
|
||||||
pcie_info_dict['pcie_metric']['pcie_width'] = "N/A"
|
|
||||||
if pcie_info_dict['pcie_metric']['pcie_speed'] == 0xFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_metric']['pcie_speed'] = "N/A"
|
|
||||||
if pcie_info_dict['pcie_metric']['pcie_bandwidth'] == 0xFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_metric']['pcie_bandwidth'] = "N/A"
|
|
||||||
|
|
||||||
# TODO Just Navi 21 has a different uint max size for pcie_bandwidth
|
|
||||||
# if pcie_info_dict['pcie_metric']['pcie_bandwidth'] == 0xFFFFFFFF:
|
|
||||||
# pcie_info_dict['pcie_metric']['pcie_bandwidth'] = "N/A"
|
|
||||||
|
|
||||||
if pcie_info_dict['pcie_metric']['pcie_replay_count'] == 0xFFFFFFFFFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_metric']['pcie_replay_count'] = "N/A"
|
|
||||||
if pcie_info_dict['pcie_metric']['pcie_l0_to_recovery_count'] == 0xFFFFFFFFFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_metric']['pcie_l0_to_recovery_count'] = "N/A"
|
|
||||||
if pcie_info_dict['pcie_metric']['pcie_replay_roll_over_count'] == 0xFFFFFFFFFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_metric']['pcie_replay_roll_over_count'] = "N/A"
|
|
||||||
if pcie_info_dict['pcie_metric']['pcie_nak_sent_count'] == 0xFFFFFFFFFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_metric']['pcie_nak_sent_count'] = "N/A"
|
|
||||||
if pcie_info_dict['pcie_metric']['pcie_nak_received_count'] == 0xFFFFFFFFFFFFFFFF:
|
|
||||||
pcie_info_dict['pcie_metric']['pcie_nak_received_count'] = "N/A"
|
|
||||||
|
|
||||||
return pcie_info_dict
|
return pcie_info_dict
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -29,3 +29,9 @@ packages = ["amdsmi"]
|
|||||||
# install libamd_smi.so
|
# install libamd_smi.so
|
||||||
[tool.setuptools.package-data]
|
[tool.setuptools.package-data]
|
||||||
amdsmi = ["*.so"]
|
amdsmi = ["*.so"]
|
||||||
|
|
||||||
|
[tool.pytest.ini_options]
|
||||||
|
pythonpath = "/opt/rocm/share/amd_smi"
|
||||||
|
addopts = [
|
||||||
|
"--import-mode=importlib",
|
||||||
|
]
|
||||||
|
|||||||
@@ -0,0 +1,28 @@
|
|||||||
|
message("&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&")
|
||||||
|
message(" CMake AMD SMI Pytest ")
|
||||||
|
message("&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&")
|
||||||
|
|
||||||
|
message("")
|
||||||
|
message("Build Configuration:")
|
||||||
|
# message("-----------BuildType: " ${CMAKE_BUILD_TYPE})
|
||||||
|
# message("------------Compiler: " ${CMAKE_CXX_COMPILER})
|
||||||
|
# message("-------------Version: " ${CMAKE_CXX_COMPILER_VERSION})
|
||||||
|
message("--------Proj Src Dir: " ${PROJECT_SOURCE_DIR})
|
||||||
|
# message("--------Proj Bld Dir: " ${PROJECT_BINARY_DIR})
|
||||||
|
# message("--------Proj Lib Dir: " ${PROJECT_BINARY_DIR}/lib)
|
||||||
|
# message("--------Proj Exe Dir: " ${PROJECT_BINARY_DIR}/bin)
|
||||||
|
message("--------Share Install Prefix: " ${SHARE_INSTALL_PREFIX})
|
||||||
|
message("--------Cpack_include_toplevel_directory: " ${CPACK_INCLUDE_TOPLEVEL_DIRECTORY})
|
||||||
|
message("--------CPACK_COMPONENT_INCLUDE_TOPLEVEL_DIRECTORY: " ${CPACK_COMPONENT_INCLUDE_TOPLEVEL_DIRECTORY})
|
||||||
|
|
||||||
|
# copy python test files into shared directory
|
||||||
|
install(
|
||||||
|
DIRECTORY ./
|
||||||
|
DESTINATION ${SHARE_INSTALL_PREFIX}/tests/pytest/
|
||||||
|
COMPONENT dev
|
||||||
|
USE_SOURCE_PERMISSIONS
|
||||||
|
FILES_MATCHING
|
||||||
|
PATTERN "*.py"
|
||||||
|
)
|
||||||
|
|
||||||
|
# message(FATAL_ERROR "python lib stop")
|
||||||
تفاوت فایلی نمایش داده نمی شود زیرا این فایل بسیار بزرگ است
Diff را بارگزاری کن
@@ -0,0 +1,4 @@
|
|||||||
|
import sys
|
||||||
|
sys.path.append("/opt/rocm/libexec/amdsmi_cli/")
|
||||||
|
|
||||||
|
from _version import __version__
|
||||||
Executable
+557
@@ -0,0 +1,557 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
#
|
||||||
|
# Copyright (c) 2024 Advanced Micro Devices, Inc. All rights reserved.
|
||||||
|
#
|
||||||
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
# of this software and associated documentation files (the "Software"), to deal
|
||||||
|
# in the Software without restriction, including without limitation the rights
|
||||||
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
# copies of the Software, and to permit persons to whom the Software is
|
||||||
|
# furnished to do so, subject to the following conditions:
|
||||||
|
#
|
||||||
|
# The above copyright notice and this permission notice shall be included in
|
||||||
|
# all copies or substantial portions of the Software.
|
||||||
|
#
|
||||||
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
# THE SOFTWARE.
|
||||||
|
import sys
|
||||||
|
sys.path.append("/opt/rocm/libexec/amdsmi_cli/")
|
||||||
|
|
||||||
|
try:
|
||||||
|
import amdsmi
|
||||||
|
except ImportError:
|
||||||
|
raise ImportError("Could not import /opt/rocm/libexec/amdsmi_cli/amdsmi_cli.py")
|
||||||
|
|
||||||
|
import unittest
|
||||||
|
import threading
|
||||||
|
import multiprocessing
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
def handle_exceptions(func):
|
||||||
|
"""Exposes, silences, and logs AMD SMI exceptions to users what exception was raised.
|
||||||
|
|
||||||
|
params:
|
||||||
|
func: test function(s) that use decorator to expose AMD SMI exceptions
|
||||||
|
return:
|
||||||
|
On success - original function is returned
|
||||||
|
On failure - silences error and prints to user what exception was caught
|
||||||
|
"""
|
||||||
|
def wrapper(*args, **kwargs):
|
||||||
|
try:
|
||||||
|
return func(*args, **kwargs)
|
||||||
|
except amdsmi.AmdSmiRetryException as e:
|
||||||
|
print("**** [ERROR] | Test: " + str(func.__name__) + " | Caught AmdSmiRetryException: {}".format(e))
|
||||||
|
pass
|
||||||
|
except amdsmi.AmdSmiTimeoutException as e:
|
||||||
|
print("**** [ERROR] | Test: " + str(func.__name__) + " | Caught AmdSmiTimeoutException: {}".format(e))
|
||||||
|
pass
|
||||||
|
except amdsmi.AmdSmiLibraryException as e:
|
||||||
|
print("**** [ERROR] | Test: " + str(func.__name__) + " | Caught AmdSmiLibraryException: {}".format(e))
|
||||||
|
pass
|
||||||
|
except Exception as e:
|
||||||
|
print("**** [ERROR] | Test: " + str(func.__name__) + " | Caught unknown exception: {}".format(e))
|
||||||
|
pass
|
||||||
|
return wrapper
|
||||||
|
|
||||||
|
class TestAmdSmiInit(unittest.TestCase):
|
||||||
|
@handle_exceptions
|
||||||
|
def test_init(self):
|
||||||
|
amdsmi.amdsmi_init()
|
||||||
|
amdsmi.amdsmi_shut_down()
|
||||||
|
|
||||||
|
class TestAmdSmiPythonInterface(unittest.TestCase):
|
||||||
|
@handle_exceptions
|
||||||
|
def setUp(self):
|
||||||
|
amdsmi.amdsmi_init()
|
||||||
|
@handle_exceptions
|
||||||
|
def tearDown(self):
|
||||||
|
amdsmi.amdsmi_shut_down()
|
||||||
|
|
||||||
|
# Bad page is not supported in Navi21 and Navi31
|
||||||
|
@handle_exceptions
|
||||||
|
def test_bad_page_info(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
processor = amdsmi.amdsmi_get_processor_handle_from_bdf(bdf)
|
||||||
|
print("\n###Test amdsmi_get_gpu_bad_page_info \n")
|
||||||
|
bad_page_info = amdsmi.amdsmi_get_gpu_bad_page_info(processors[i])
|
||||||
|
print("bad_page_info: " + str(bad_page_info))
|
||||||
|
print("Number of bad pages: {}".format(len(bad_page_info)))
|
||||||
|
j = 0
|
||||||
|
for table_record in bad_page_info:
|
||||||
|
print("\ntable_record[\"value\"]" + str(table_record["value"]))
|
||||||
|
print("Page: {}".format(j))
|
||||||
|
print("Page Address: " + str(table_record["page_address"]))
|
||||||
|
print("Page Size: " + str(table_record["page_size"]))
|
||||||
|
print("Status: " + str(table_record["status"]))
|
||||||
|
print()
|
||||||
|
j += 1
|
||||||
|
print()
|
||||||
|
|
||||||
|
def test_bdf_device_id(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
processor = amdsmi.amdsmi_get_processor_handle_from_bdf(bdf)
|
||||||
|
print("\n###Test amdsmi_get_gpu_vbios_info \n")
|
||||||
|
vbios_info = amdsmi.amdsmi_get_gpu_vbios_info(processor)
|
||||||
|
print(" vbios_info['part_number'] is: {}".format(
|
||||||
|
vbios_info['part_number']))
|
||||||
|
print(" vbios_info['build_date'] is: {}".format(
|
||||||
|
vbios_info['build_date']))
|
||||||
|
print(" vbios_info['version'] is: {}".format(
|
||||||
|
vbios_info['version']))
|
||||||
|
print(" vbios_info['name'] is: {}".format(
|
||||||
|
vbios_info['name']))
|
||||||
|
print("\n###Test amdsmi_get_gpu_device_uuid \n")
|
||||||
|
uuid = amdsmi.amdsmi_get_gpu_device_uuid(processor)
|
||||||
|
print(" uuid is: {}".format(uuid))
|
||||||
|
print()
|
||||||
|
|
||||||
|
def test_ecc(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_gpu_total_ecc_count \n")
|
||||||
|
ecc_info = amdsmi.amdsmi_get_gpu_total_ecc_count(processors[i])
|
||||||
|
print("Number of uncorrectable errors: {}".format(
|
||||||
|
ecc_info['uncorrectable_count']))
|
||||||
|
print("Number of correctable errors: {}".format(
|
||||||
|
ecc_info['correctable_count']))
|
||||||
|
print("Number of deferred errors: {}".format(
|
||||||
|
ecc_info['deferred_count']))
|
||||||
|
self.assertGreaterEqual(ecc_info['uncorrectable_count'], 0)
|
||||||
|
self.assertGreaterEqual(ecc_info['correctable_count'], 0)
|
||||||
|
self.assertGreaterEqual(ecc_info['deferred_count'], 0)
|
||||||
|
print()
|
||||||
|
|
||||||
|
# RAS is not supported in Navi21 and Navi31
|
||||||
|
@handle_exceptions
|
||||||
|
def test_ras(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_gpu_ras_feature_info \n")
|
||||||
|
ras_feature = amdsmi.amdsmi_get_gpu_ras_feature_info(processors[i])
|
||||||
|
print("ras_feature: " + str(ras_feature))
|
||||||
|
if ras_feature != None:
|
||||||
|
print("ras_feature: " + str(ras_feature))
|
||||||
|
print("RAS eeprom version: {}".format(ras_feature['eeprom_version']))
|
||||||
|
print("RAS parity schema: {}".format(ras_feature['parity_schema']))
|
||||||
|
print("RAS single bit schema: {}".format(ras_feature['single_bit_schema']))
|
||||||
|
print("RAS double bit schema: {}".format(ras_feature['double_bit_schema']))
|
||||||
|
print("Poisioning supported: {}".format(ras_feature['poison_schema']))
|
||||||
|
print()
|
||||||
|
|
||||||
|
def test_clock_info(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_clock_info \n")
|
||||||
|
clock_measure = amdsmi.amdsmi_get_clock_info(
|
||||||
|
processors[i], amdsmi.AmdSmiClkType.GFX)
|
||||||
|
print(" Current clock for domain GFX is: {}".format(
|
||||||
|
clock_measure['clk']))
|
||||||
|
print(" Max clock for domain GFX is: {}".format(
|
||||||
|
clock_measure['max_clk']))
|
||||||
|
print(" Min clock for domain GFX is: {}".format(
|
||||||
|
clock_measure['min_clk']))
|
||||||
|
print(" Is GFX clock locked: {}".format(
|
||||||
|
clock_measure['clk_locked']))
|
||||||
|
print(" Is GFX clock in deep sleep: {}".format(
|
||||||
|
clock_measure['clk_deep_sleep']))
|
||||||
|
clock_measure = amdsmi.amdsmi_get_clock_info(
|
||||||
|
processors[i], amdsmi.AmdSmiClkType.MEM)
|
||||||
|
print(" Current clock for domain MEM is: {}".format(
|
||||||
|
clock_measure['clk']))
|
||||||
|
print(" Max clock for domain MEM is: {}".format(
|
||||||
|
clock_measure['max_clk']))
|
||||||
|
print(" Min clock for domain MEM is: {}".format(
|
||||||
|
clock_measure['min_clk']))
|
||||||
|
print(" Is MEM clock in deep sleep: {}".format(
|
||||||
|
clock_measure['clk_deep_sleep']))
|
||||||
|
print()
|
||||||
|
|
||||||
|
# VCLK0 and DCLK0 are not supported in MI210
|
||||||
|
@handle_exceptions
|
||||||
|
def test_gpu_clock_vclk0_dclk0(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_clock_info \n")
|
||||||
|
clock_measure = amdsmi.amdsmi_get_clock_info(
|
||||||
|
processors[i], amdsmi.AmdSmiClkType.VCLK0)
|
||||||
|
print(" Current clock for domain VCLK0 is: {}".format(
|
||||||
|
clock_measure['clk']))
|
||||||
|
print(" Max clock for domain VCLK0 is: {}".format(
|
||||||
|
clock_measure['max_clk']))
|
||||||
|
print(" Min clock for domain VCLK0 is: {}".format(
|
||||||
|
clock_measure['min_clk']))
|
||||||
|
print(" Is VCLK0 clock in deep sleep: {}".format(
|
||||||
|
clock_measure['clk_deep_sleep']))
|
||||||
|
clock_measure = amdsmi.amdsmi_get_clock_info(
|
||||||
|
processors[i], amdsmi.AmdSmiClkType.DCLK0)
|
||||||
|
print(" Current clock for domain DCLK0 is: {}".format(
|
||||||
|
clock_measure['clk']))
|
||||||
|
print(" Max clock for domain DCLK0 is: {}".format(
|
||||||
|
clock_measure['max_clk']))
|
||||||
|
print(" Min clock for domain DCLK0 is: {}".format(
|
||||||
|
clock_measure['min_clk']))
|
||||||
|
print(" Is DCLK0 clock in deep sleep: {}".format(
|
||||||
|
clock_measure['clk_deep_sleep']))
|
||||||
|
print()
|
||||||
|
|
||||||
|
# VCLK1 and DCLK1 are not supported in Navi 31, MI210, and MI300
|
||||||
|
@handle_exceptions
|
||||||
|
def test_gpu_clock_vclk1_dclk1(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_clock_info \n")
|
||||||
|
clock_measure = amdsmi.amdsmi_get_clock_info(
|
||||||
|
processors[i], amdsmi.AmdSmiClkType.VCLK1)
|
||||||
|
print(" Current clock for domain VCLK1 is: {}".format(
|
||||||
|
clock_measure['clk']))
|
||||||
|
print(" Max clock for domain VCLK1 is: {}".format(
|
||||||
|
clock_measure['max_clk']))
|
||||||
|
print(" Min clock for domain VCLK1 is: {}".format(
|
||||||
|
clock_measure['min_clk']))
|
||||||
|
print(" Is VCLK1 clock in deep sleep: {}".format(
|
||||||
|
clock_measure['clk_deep_sleep']))
|
||||||
|
clock_measure = amdsmi.amdsmi_get_clock_info(
|
||||||
|
processors[i], amdsmi.AmdSmiClkType.DCLK1)
|
||||||
|
print(" Current clock for domain DCLK1 is: {}".format(
|
||||||
|
clock_measure['clk']))
|
||||||
|
print(" Max clock for domain DCLK1 is: {}".format(
|
||||||
|
clock_measure['max_clk']))
|
||||||
|
print(" Min clock for domain DCLK1 is: {}".format(
|
||||||
|
clock_measure['min_clk']))
|
||||||
|
print(" Is DCLK1 clock in deep sleep: {}".format(
|
||||||
|
clock_measure['clk_deep_sleep']))
|
||||||
|
print()
|
||||||
|
|
||||||
|
def test_gpu_activity(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_gpu_activity \n")
|
||||||
|
engine_usage = amdsmi.amdsmi_get_gpu_activity(processors[i])
|
||||||
|
print(" engine_usage['gfx_activity'] is: {} %".format(
|
||||||
|
engine_usage['gfx_activity']))
|
||||||
|
print(" engine_usage['umc_activity'] is: {} %".format(
|
||||||
|
engine_usage['umc_activity']))
|
||||||
|
print(" engine_usage['mm_activity'] is: {} %".format(
|
||||||
|
engine_usage['mm_activity']))
|
||||||
|
print()
|
||||||
|
|
||||||
|
def test_pcie(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_pcie_info \n")
|
||||||
|
pcie_info = amdsmi.amdsmi_get_pcie_info(processors[i])
|
||||||
|
print(" pcie_info['pcie_metric']['pcie_width'] is: {}".format(
|
||||||
|
pcie_info['pcie_metric']['pcie_width']))
|
||||||
|
print(" pcie_info['pcie_static']['max_pcie_width'] is: {} ".format(
|
||||||
|
pcie_info['pcie_static']['max_pcie_width']))
|
||||||
|
print(" pcie_info['pcie_metric']['pcie_speed'] is: {} MT/s".format(
|
||||||
|
pcie_info['pcie_metric']['pcie_speed']))
|
||||||
|
print(" pcie_info['pcie_static']['max_pcie_speed'] is: {} ".format(
|
||||||
|
pcie_info['pcie_static']['max_pcie_speed']))
|
||||||
|
print(" pcie_info['pcie_static']['pcie_interface_version'] is: {}".format(
|
||||||
|
pcie_info['pcie_static']['pcie_interface_version']))
|
||||||
|
print(" pcie_info['pcie_static']['slot_type'] is: {}".format(
|
||||||
|
pcie_info['pcie_static']['slot_type']))
|
||||||
|
print(" pcie_info['pcie_metric']['pcie_replay_count'] is: {}".format(
|
||||||
|
pcie_info['pcie_metric']['pcie_replay_count']))
|
||||||
|
print(" pcie_info['pcie_metric']['pcie_bandwidth'] is: {}".format(
|
||||||
|
pcie_info['pcie_metric']['pcie_bandwidth']))
|
||||||
|
print(" pcie_info['pcie_metric']['pcie_l0_to_recovery_count'] is: {}".format(
|
||||||
|
pcie_info['pcie_metric']['pcie_l0_to_recovery_count']))
|
||||||
|
print(" pcie_info['pcie_metric']['pcie_replay_roll_over_count'] is: {}".format(
|
||||||
|
pcie_info['pcie_metric']['pcie_replay_roll_over_count']))
|
||||||
|
print(" pcie_info['pcie_metric']['pcie_nak_sent_count'] is: {}".format(
|
||||||
|
pcie_info['pcie_metric']['pcie_nak_sent_count']))
|
||||||
|
print(" pcie_info['pcie_metric']['pcie_nak_received_count'] is: {}".format(
|
||||||
|
pcie_info['pcie_metric']['pcie_nak_received_count']))
|
||||||
|
print()
|
||||||
|
|
||||||
|
def test_power(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_power_info \n")
|
||||||
|
power_info = amdsmi.amdsmi_get_power_info(processors[i])
|
||||||
|
print(" power_info['current_socket_power'] is: {}".format(
|
||||||
|
power_info['current_socket_power']))
|
||||||
|
print(" power_info['average_socket_power'] is: {}".format(
|
||||||
|
power_info['average_socket_power']))
|
||||||
|
print(" power_info['gfx_voltage'] is: {}".format(
|
||||||
|
power_info['gfx_voltage']))
|
||||||
|
print(" power_info['soc_voltage'] is: {}".format(
|
||||||
|
power_info['soc_voltage']))
|
||||||
|
print(" power_info['mem_voltage'] is: {}".format(
|
||||||
|
power_info['mem_voltage']))
|
||||||
|
print(" power_info['power_limit'] is: {}".format(
|
||||||
|
power_info['power_limit']))
|
||||||
|
print("\n###Test amdsmi_is_gpu_power_management_enabled \n")
|
||||||
|
is_power_management_enabled = amdsmi.amdsmi_is_gpu_power_management_enabled(processors[i])
|
||||||
|
print(" Is power management enabled is: {}".format(
|
||||||
|
is_power_management_enabled))
|
||||||
|
print()
|
||||||
|
|
||||||
|
def test_temperature(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_temp_metric \n")
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.HOTSPOT, amdsmi.AmdSmiTemperatureMetric.CURRENT)
|
||||||
|
print(" Current temperature for HOTSPOT is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.VRAM, amdsmi.AmdSmiTemperatureMetric.CURRENT)
|
||||||
|
print(" Current temperature for VRAM is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
print("\n###Test amdsmi_get_temp_metric \n")
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.HOTSPOT, amdsmi.AmdSmiTemperatureMetric.CRITICAL)
|
||||||
|
print(" Limit (critical) temperature for HOTSPOT is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.VRAM, amdsmi.AmdSmiTemperatureMetric.CRITICAL)
|
||||||
|
print(" Limit (critical) temperature for VRAM is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
print("\n###Test amdsmi_get_temp_metric \n")
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.HOTSPOT, amdsmi.AmdSmiTemperatureMetric.EMERGENCY)
|
||||||
|
print(" Shutdown (emergency) temperature for HOTSPOT is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.VRAM, amdsmi.AmdSmiTemperatureMetric.EMERGENCY)
|
||||||
|
print(" Shutdown (emergency) temperature for VRAM is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
print()
|
||||||
|
|
||||||
|
# Edge temperature is not supported in MI300
|
||||||
|
@handle_exceptions
|
||||||
|
def test_temperature_edge(self):
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
self.assertGreaterEqual(len(processors), 1)
|
||||||
|
self.assertLessEqual(len(processors), 32)
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_temp_metric \n")
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.EDGE, amdsmi.AmdSmiTemperatureMetric.CURRENT) # current
|
||||||
|
print(" Current temperature for EDGE is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.EDGE, amdsmi.AmdSmiTemperatureMetric.CRITICAL) # slowdown/limit
|
||||||
|
print(" Limit (critical) temperature for EDGE is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
temperature_measure = amdsmi.amdsmi_get_temp_metric(
|
||||||
|
processors[i], amdsmi.AmdSmiTemperatureType.EDGE, amdsmi.AmdSmiTemperatureMetric.EMERGENCY) # shutdown
|
||||||
|
print(" Shutdown (emergency) temperature for EDGE is: {}".format(
|
||||||
|
temperature_measure))
|
||||||
|
print()
|
||||||
|
|
||||||
|
def test_walkthrough(self):
|
||||||
|
walk_through(self)
|
||||||
|
|
||||||
|
# Unstable on workstation cards
|
||||||
|
# @handle_exceptions
|
||||||
|
# def test_walkthrough_multiprocess(self):
|
||||||
|
# print("\n\n========> test_walkthrough_multiprocess start <========\n")
|
||||||
|
# processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
# self.assertGreaterEqual(len(processors), 1)
|
||||||
|
# self.assertLessEqual(len(processors), 32)
|
||||||
|
# p0 = multiprocessing.Process(target=walk_through, args=[self])
|
||||||
|
# p1 = multiprocessing.Process(target=walk_through, args=[self])
|
||||||
|
# p2 = multiprocessing.Process(target=walk_through, args=[self])
|
||||||
|
# p3 = multiprocessing.Process(target=walk_through, args=[self])
|
||||||
|
# p0.start()
|
||||||
|
# p1.start()
|
||||||
|
# p2.start()
|
||||||
|
# p3.start()
|
||||||
|
# p0.join()
|
||||||
|
# p1.join()
|
||||||
|
# p2.join()
|
||||||
|
# p3.join()
|
||||||
|
# print("\n========> test_walkthrough_multiprocess end <========\n")
|
||||||
|
|
||||||
|
# Unstable on workstation cards
|
||||||
|
# @handle_exceptions
|
||||||
|
# def test_walkthrough_multithread(self):
|
||||||
|
# print("\n\n========> test_walkthrough_multithread start <========\n")
|
||||||
|
# processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
# self.assertGreaterEqual(len(processors), 1)
|
||||||
|
# self.assertLessEqual(len(processors), 32)
|
||||||
|
# t0 = threading.Thread(target=walk_through, args=[self])
|
||||||
|
# t1 = threading.Thread(target=walk_through, args=[self])
|
||||||
|
# t2 = threading.Thread(target=walk_through, args=[self])
|
||||||
|
# t3 = threading.Thread(target=walk_through, args=[self])
|
||||||
|
# t0.start()
|
||||||
|
# t1.start()
|
||||||
|
# t2.start()
|
||||||
|
# t3.start()
|
||||||
|
# t0.join()
|
||||||
|
# t1.join()
|
||||||
|
# t2.join()
|
||||||
|
# t3.join()
|
||||||
|
# print("\n========> test_walkthrough_multithread end <========\n")
|
||||||
|
|
||||||
|
# # Unstable - do not run
|
||||||
|
# @handle_exceptions
|
||||||
|
# def test_z_gpureset_asicinfo_multithread(self):
|
||||||
|
# def get_asic_info(processor):
|
||||||
|
# print("\n###Test amdsmi_get_gpu_asic_info \n")
|
||||||
|
# asic_info = amdsmi.amdsmi_get_gpu_asic_info(processor)
|
||||||
|
# print(" asic_info['market_name'] is: {}".format(
|
||||||
|
# asic_info['market_name']))
|
||||||
|
# print(" asic_info['vendor_id'] is: {}".format(
|
||||||
|
# asic_info['vendor_id']))
|
||||||
|
# print(" asic_info['vendor_name'] is: {}".format(
|
||||||
|
# asic_info['vendor_name']))
|
||||||
|
# print(" asic_info['device_id'] is: {}".format(
|
||||||
|
# asic_info['device_id']))
|
||||||
|
# print(" asic_info['rev_id'] is: {}".format(
|
||||||
|
# asic_info['rev_id']))
|
||||||
|
# print(" asic_info['asic_serial'] is: {}".format(
|
||||||
|
# asic_info['asic_serial']))
|
||||||
|
# print(" asic_info['oam_id'] is: {}\n".format(
|
||||||
|
# asic_info['oam_id']))
|
||||||
|
# def gpu_reset(processor):
|
||||||
|
# print("\n###Test amdsmi_reset_gpu \n")
|
||||||
|
# amdsmi.amdsmi_reset_gpu(processor)
|
||||||
|
# print(" GPU reset completed.\n")
|
||||||
|
# print("\n\n========> test_z_gpureset_asicinfo_multithread start <========\n")
|
||||||
|
# processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
# self.assertGreaterEqual(len(processors), 1)
|
||||||
|
# self.assertLessEqual(len(processors), 32)
|
||||||
|
# for i in range(0, len(processors)):
|
||||||
|
# bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
# print("\n\n###Test Processor {}, bdf: {}".format(i, bdf))
|
||||||
|
# t0 = threading.Thread(target=get_asic_info, args=[processors[i]])
|
||||||
|
# t1 = threading.Thread(target=gpu_reset, args=[processors[i]])
|
||||||
|
# # t2 = threading.Thread(target=walk_through, args=[self])
|
||||||
|
# # t3 = threading.Thread(target=walk_through, args=[self])
|
||||||
|
# t0.start()
|
||||||
|
# t1.start()
|
||||||
|
# # t2.start()
|
||||||
|
# # t3.start()
|
||||||
|
# t0.join()
|
||||||
|
# t1.join()
|
||||||
|
# # t2.join()
|
||||||
|
# # t3.join()
|
||||||
|
# print("\n========> test_z_gpureset_asicinfo_multithread end <========\n")
|
||||||
|
|
||||||
|
def walk_through(self):
|
||||||
|
print("\n###Test amdsmi_get_processor_handles() \n")
|
||||||
|
processors = amdsmi.amdsmi_get_processor_handles()
|
||||||
|
for i in range(0, len(processors)):
|
||||||
|
print("\n###Test amdsmi_get_gpu_device_bdf() | START walk_through | processor i = " + str(i) + "\n")
|
||||||
|
bdf = amdsmi.amdsmi_get_gpu_device_bdf(processors[i])
|
||||||
|
print("###Test Processor {}, bdf: {} ".format(i, bdf))
|
||||||
|
print("\n###Test amdsmi_get_gpu_asic_info \n")
|
||||||
|
asic_info = amdsmi.amdsmi_get_gpu_asic_info(processors[i])
|
||||||
|
print(" asic_info['market_name'] is: {}".format(
|
||||||
|
asic_info['market_name']))
|
||||||
|
print(" asic_info['vendor_id'] is: {}".format(
|
||||||
|
asic_info['vendor_id']))
|
||||||
|
print(" asic_info['vendor_name'] is: {}".format(
|
||||||
|
asic_info['vendor_name']))
|
||||||
|
print(" asic_info['device_id'] is: {}".format(
|
||||||
|
asic_info['device_id']))
|
||||||
|
print(" asic_info['rev_id'] is: {}\n".format(
|
||||||
|
asic_info['rev_id']))
|
||||||
|
print(" asic_info['asic_serial'] is: {}\n".format(
|
||||||
|
asic_info['asic_serial']))
|
||||||
|
print(" asic_info['oam_id'] is: {}\n".format(
|
||||||
|
asic_info['oam_id']))
|
||||||
|
print("###Test amdsmi_get_power_cap_info \n")
|
||||||
|
power_info = amdsmi.amdsmi_get_power_cap_info(processors[i])
|
||||||
|
print(" power_info['dpm_cap'] is: {}".format(
|
||||||
|
power_info['dpm_cap']))
|
||||||
|
print(" power_info['power_cap'] is: {}\n".format(
|
||||||
|
power_info['power_cap']))
|
||||||
|
print("###Test amdsmi_get_gpu_vbios_info \n")
|
||||||
|
vbios_info = amdsmi.amdsmi_get_gpu_vbios_info(processors[i])
|
||||||
|
print(" vbios_info['part_number'] is: {}".format(
|
||||||
|
vbios_info['part_number']))
|
||||||
|
print(" vbios_info['build_date'] is: {}".format(
|
||||||
|
vbios_info['build_date']))
|
||||||
|
print(" vbios_info['name'] is: {}\n".format(
|
||||||
|
vbios_info['name']))
|
||||||
|
print(" vbios_info['version'] is: {}\n".format(
|
||||||
|
vbios_info['version']))
|
||||||
|
print("###Test amdsmi_get_gpu_board_info \n")
|
||||||
|
board_info = amdsmi.amdsmi_get_gpu_board_info(processors[i])
|
||||||
|
print(" board_info['model_number'] is: {}\n".format(
|
||||||
|
board_info['model_number']))
|
||||||
|
print(" board_info['product_serial'] is: {}\n".format(
|
||||||
|
board_info['product_serial']))
|
||||||
|
print(" board_info['fru_id'] is: {}\n".format(
|
||||||
|
board_info['fru_id']))
|
||||||
|
print(" board_info['manufacturer_name'] is: {}\n".format(
|
||||||
|
board_info['manufacturer_name']))
|
||||||
|
print(" board_info['product_name'] is: {}\n".format(
|
||||||
|
board_info['product_name']))
|
||||||
|
print("###Test amdsmi_get_fw_info \n")
|
||||||
|
fw_info = amdsmi.amdsmi_get_fw_info(processors[i])
|
||||||
|
fw_num = len(fw_info['fw_list'])
|
||||||
|
self.assertLessEqual(fw_num, len(amdsmi.AmdSmiFwBlock))
|
||||||
|
for j in range(0, fw_num):
|
||||||
|
fw = fw_info['fw_list'][j]
|
||||||
|
if fw['fw_version'] != 0:
|
||||||
|
print("FW name: {}".format(
|
||||||
|
fw['fw_name'].name))
|
||||||
|
print("FW version: {}".format(
|
||||||
|
fw['fw_version']))
|
||||||
|
print("\n###Test amdsmi_get_gpu_driver_info \n")
|
||||||
|
driver_info = amdsmi.amdsmi_get_gpu_driver_info(processors[i])
|
||||||
|
print("Driver info: {}".format(driver_info))
|
||||||
|
print("\n###Test amdsmi_get_gpu_driver_info() | END walk_through | processor i = " + str(i) + "\n")
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
unittest.main()
|
||||||
Executable
+163
@@ -0,0 +1,163 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
#
|
||||||
|
# Copyright (c) 2024 Advanced Micro Devices, Inc. All rights reserved.
|
||||||
|
#
|
||||||
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
# of this software and associated documentation files (the "Software"), to deal
|
||||||
|
# in the Software without restriction, including without limitation the rights
|
||||||
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
# copies of the Software, and to permit persons to whom the Software is
|
||||||
|
# furnished to do so, subject to the following conditions:
|
||||||
|
#
|
||||||
|
# The above copyright notice and this permission notice shall be included in
|
||||||
|
# all copies or substantial portions of the Software.
|
||||||
|
#
|
||||||
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
# THE SOFTWARE.
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
import sys
|
||||||
|
sys.path.append("/opt/rocm/libexec/amdsmi_cli/")
|
||||||
|
|
||||||
|
try:
|
||||||
|
import amdsmi
|
||||||
|
except ImportError:
|
||||||
|
raise ImportError("Could not import /opt/rocm/libexec/amdsmi_cli/amdsmi_cli.py")
|
||||||
|
|
||||||
|
class TestAmdSmiPythonBDF(unittest.TestCase):
|
||||||
|
valid_bdfs = {
|
||||||
|
"00:00.0": [0, 0, 0, 0],
|
||||||
|
"01:01.1": [0, 1, 1, 1],
|
||||||
|
"FF:1F.7": [0, 255, 31, 7],
|
||||||
|
"FF:00.7": [0, 255, 0, 7],
|
||||||
|
"11:01.2": [0, 17, 1, 2],
|
||||||
|
"11:0a.2": [0, 17, 10, 2],
|
||||||
|
"0000:FF:1F.7": [0, 255, 31, 7],
|
||||||
|
"0001:ff:1F.7": [1, 255, 31, 7],
|
||||||
|
"ffff:FF:1f.7": [65535, 255, 31, 7],
|
||||||
|
}
|
||||||
|
invalid_bdfs = {
|
||||||
|
# invalid bdf strings, expect None
|
||||||
|
None: None,
|
||||||
|
"": None,
|
||||||
|
"00:00:0": None,
|
||||||
|
"00.00:0": None,
|
||||||
|
"00:00.Z": None,
|
||||||
|
"00:0Z.0": None,
|
||||||
|
"0Z:00.0": None,
|
||||||
|
"Z00:00.0": None,
|
||||||
|
"A00:00.0": None,
|
||||||
|
"0A00:00.0": None,
|
||||||
|
"00:00.07": None,
|
||||||
|
"00:00.8": None,
|
||||||
|
"00:00.10": None,
|
||||||
|
"00:00.11": None,
|
||||||
|
"00:00.-1": None,
|
||||||
|
"00:00.*-1": None,
|
||||||
|
"00:00.123": None,
|
||||||
|
"00:20.0": None,
|
||||||
|
"00:45.0": None,
|
||||||
|
"00:200.0": None,
|
||||||
|
"00:002.0": None,
|
||||||
|
"100:00.0": None,
|
||||||
|
"0100:00.0": None,
|
||||||
|
"00100:00.0": None,
|
||||||
|
"0101:00.0": None,
|
||||||
|
"00001:00.0": None,
|
||||||
|
"10001:00.0": None,
|
||||||
|
"45:0.0": None,
|
||||||
|
".00:00.0": None,
|
||||||
|
"00.00.0": None,
|
||||||
|
"00.0.0": None,
|
||||||
|
"0.00.0": None,
|
||||||
|
"000.00.0": None,
|
||||||
|
"00 00 0": None,
|
||||||
|
" 00:00.0": None,
|
||||||
|
"00:00.0 ": None,
|
||||||
|
"0000:00.00.0": None,
|
||||||
|
"000:00:00.0": None,
|
||||||
|
"00:00:00.1": None,
|
||||||
|
"0:00:00.1": None,
|
||||||
|
"0000 00 00 0": None,
|
||||||
|
"-1-1:00:00.0": None,
|
||||||
|
"AAAA:00:AA.0": None,
|
||||||
|
"*1*1:00:00.0": None,
|
||||||
|
"0000:00:00.07": None,
|
||||||
|
"0000:00:00.8": None,
|
||||||
|
"0000:00:00.10": None,
|
||||||
|
"0000:00:00.11": None,
|
||||||
|
"0000:00:00.-1": None,
|
||||||
|
"0000:00:00.*-1": None,
|
||||||
|
"0000:00:00.123": None,
|
||||||
|
"0000:00:20.0": None,
|
||||||
|
"0000:00:45.0": None,
|
||||||
|
"0000:00:200.0": None,
|
||||||
|
"0000:00:002.0": None,
|
||||||
|
"0000:100:00.0": None,
|
||||||
|
"0000:0100:00.0": None,
|
||||||
|
"0000:00100:00.0": None,
|
||||||
|
"0000:0101:00.0": None,
|
||||||
|
"0000:00001:00.0": None,
|
||||||
|
"0000:10001:00.0": None,
|
||||||
|
"0000:45:0.0": None,
|
||||||
|
".0000.00:00.0": None,
|
||||||
|
"0000.00.0.0": None,
|
||||||
|
" 0000:00:00.0": None,
|
||||||
|
"0000:00:00.0 ": None,
|
||||||
|
}
|
||||||
|
def test_parse_bdf(self):
|
||||||
|
# go through all bdfs
|
||||||
|
expectations = self.valid_bdfs.copy()
|
||||||
|
expectations.update(self.invalid_bdfs)
|
||||||
|
for bdf in expectations:
|
||||||
|
expected = expectations[bdf]
|
||||||
|
result = amdsmi.amdsmi_interface._parse_bdf(bdf)
|
||||||
|
self.assertEqual(result, expected,
|
||||||
|
"Expected {} for bdf {}, but got {}".format(
|
||||||
|
expected, bdf, result))
|
||||||
|
@classmethod
|
||||||
|
def _convert_bdf_to_long(clz, bdf):
|
||||||
|
if len(bdf) == 12:
|
||||||
|
return bdf
|
||||||
|
if len(bdf) == 7:
|
||||||
|
return "0000:" + bdf
|
||||||
|
return None
|
||||||
|
def test_format_bdf(self):
|
||||||
|
# go through valid bdfs
|
||||||
|
expectations = self.valid_bdfs.copy()
|
||||||
|
for bdf_string in expectations:
|
||||||
|
# use key as result and value as input
|
||||||
|
bdf_list = expectations[bdf_string]
|
||||||
|
smi_bdf = amdsmi.amdsmi_interface._make_amdsmi_bdf_from_list(bdf_list)
|
||||||
|
expected = TestAmdSmiPythonBDF._convert_bdf_to_long(bdf_string)
|
||||||
|
expected = expected.lower()
|
||||||
|
result = amdsmi.amdsmi_interface._format_bdf(smi_bdf)
|
||||||
|
self.assertEqual(result, expected,
|
||||||
|
"Expected {} for bdf {}, but got {}".format(
|
||||||
|
expected, bdf_string, result))
|
||||||
|
def test_check_res(self):
|
||||||
|
# expect retry error to raise SmiRetryException
|
||||||
|
with self.assertRaises(amdsmi.AmdSmiRetryException) as retry_test:
|
||||||
|
amdsmi.amdsmi_interface._check_res(
|
||||||
|
(lambda: amdsmi.amdsmi_interface.amdsmi_wrapper.AMDSMI_STATUS_RETRY)())
|
||||||
|
# except retry error to have AMDSMI_STATUS_RETRY error code
|
||||||
|
self.assertEqual(retry_test.exception.get_error_code(),
|
||||||
|
amdsmi.amdsmi_interface.amdsmi_wrapper.AMDSMI_STATUS_RETRY)
|
||||||
|
# expect invalid args error to raise AmdSmiLibraryException
|
||||||
|
with self.assertRaises(amdsmi.AmdSmiLibraryException) as inval_test:
|
||||||
|
amdsmi.amdsmi_interface._check_res(
|
||||||
|
(lambda: amdsmi.amdsmi_interface.amdsmi_wrapper.AMDSMI_STATUS_INVAL)())
|
||||||
|
# expect invalid args error to have AMDSMI_STATUS_INVAL error code
|
||||||
|
self.assertEqual(inval_test.exception.get_error_code(),
|
||||||
|
amdsmi.amdsmi_interface.amdsmi_wrapper.AMDSMI_STATUS_INVAL)
|
||||||
|
# for successfull call, expect no error is given
|
||||||
|
result = amdsmi.amdsmi_interface._check_res(
|
||||||
|
(lambda: amdsmi.amdsmi_interface.amdsmi_wrapper.AMDSMI_STATUS_SUCCESS)())
|
||||||
|
self.assertEqual(None, result)
|
||||||
|
if __name__ == '__main__':
|
||||||
|
unittest.main()
|
||||||
@@ -963,6 +963,7 @@ rsmi_dev_oam_id_get(uint32_t dv_ind, uint16_t *id) {
|
|||||||
ss << __PRETTY_FUNCTION__ << "| ======= start =======";
|
ss << __PRETTY_FUNCTION__ << "| ======= start =======";
|
||||||
LOG_TRACE(ss);
|
LOG_TRACE(ss);
|
||||||
CHK_SUPPORT_NAME_ONLY(id)
|
CHK_SUPPORT_NAME_ONLY(id)
|
||||||
|
*id = std::numeric_limits<uint16_t>::max();
|
||||||
|
|
||||||
ret = get_id(dv_ind, amd::smi::kDevXGMIPhysicalID, id);
|
ret = get_id(dv_ind, amd::smi::kDevXGMIPhysicalID, id);
|
||||||
ss << __PRETTY_FUNCTION__ << " | ======= end ======="
|
ss << __PRETTY_FUNCTION__ << " | ======= end ======="
|
||||||
|
|||||||
@@ -470,7 +470,7 @@ amdsmi_status_t amdsmi_get_gpu_board_info(amdsmi_processor_handle processor_hand
|
|||||||
LOG_INFO(ss);
|
LOG_INFO(ss);
|
||||||
}
|
}
|
||||||
|
|
||||||
ss << __PRETTY_FUNCTION__ << "[After rocm smi correction] "
|
ss << __PRETTY_FUNCTION__ << " | [After rocm smi correction] "
|
||||||
<< "Returning status = AMDSMI_STATUS_SUCCESS"
|
<< "Returning status = AMDSMI_STATUS_SUCCESS"
|
||||||
<< "\n; info->model_number: |" << board_info->model_number << "|"
|
<< "\n; info->model_number: |" << board_info->model_number << "|"
|
||||||
<< "\n; info->product_serial: |" << board_info->product_serial << "|"
|
<< "\n; info->product_serial: |" << board_info->product_serial << "|"
|
||||||
@@ -751,10 +751,9 @@ amdsmi_get_gpu_asic_info(amdsmi_processor_handle processor_handle, amdsmi_asic_i
|
|||||||
}
|
}
|
||||||
|
|
||||||
// default to 0xffff as not supported
|
// default to 0xffff as not supported
|
||||||
info->oam_id = std::numeric_limits<uint32_t>::max();
|
info->oam_id = std::numeric_limits<uint16_t>::max();
|
||||||
uint16_t tmp_oam_id = 0;
|
uint16_t tmp_oam_id = 0;
|
||||||
status = rsmi_wrapper(rsmi_dev_oam_id_get, processor_handle,
|
status = rsmi_wrapper(rsmi_dev_oam_id_get, processor_handle, &(tmp_oam_id));
|
||||||
&(tmp_oam_id));
|
|
||||||
info->oam_id = tmp_oam_id;
|
info->oam_id = tmp_oam_id;
|
||||||
|
|
||||||
// default to 0xffffffff as not supported
|
// default to 0xffffffff as not supported
|
||||||
|
|||||||
مرجع در شماره جدید
Block a user