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

Unify the amdsmi_get_pcie_info python interface

Browse Source 
Make the python interface consistent with the C interface.

Change-Id: Idda08f888947c757e475d5a024b0ec3d8e1d846a
This commit is contained in:
Bill(Shuzhou) Liu
2024-02-15 12:41:50 -06:00
committed by Maisam Arif
parent f58613561c
commit db33cda0c1
7 changed files with 36 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py-interface/amdsmi_interface.py
+16 -5
View File
@@ -2134,13 +2134,24 @@ def amdsmi_get_pcie_info(
)
)
return {"pcie_speed": pcie_info.pcie_metric.pcie_speed,
"pcie_lanes": pcie_info.pcie_metric.pcie_lanes,
"pcie_interface_version": pcie_info.pcie_static.pcie_interface_version,
return {
"pcie_static": {
"max_pcie_width": pcie_info.pcie_static.max_pcie_width,
"max_pcie_speed": pcie_info.pcie_static.max_pcie_speed,
"max_pcie_lanes": pcie_info.pcie_static.max_pcie_lanes,
"pcie_interface_version": pcie_info.pcie_static.pcie_interface_version,
"pcie_slot_type": pcie_info.pcie_static.slot_type}
"slot_type": pcie_info.pcie_static.slot_type,
},
"pcie_metric": {
"pcie_width": pcie_info.pcie_metric.pcie_width,
"pcie_speed": pcie_info.pcie_metric.pcie_speed,
"pcie_bandwidth": pcie_info.pcie_metric.pcie_bandwidth,
"pcie_replay_count": pcie_info.pcie_metric.pcie_replay_count,
"pcie_l0_to_recovery_count": pcie_info.pcie_metric.pcie_l0_to_recovery_count,
"pcie_replay_roll_over_count": pcie_info.pcie_metric.pcie_replay_roll_over_count,
"pcie_nak_sent_count": pcie_info.pcie_metric.pcie_nak_sent_count,
"pcie_nak_received_count": pcie_info.pcie_metric.pcie_nak_received_count,
}
}
def amdsmi_get_processor_handle_from_bdf(bdf):