Unify the amdsmi_get_pcie_info python interface

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
+2 -2
View File
@@ -791,7 +791,7 @@ Output: Dictionary with fields
Field | Description
---|---
`pcie_lanes`| pcie lanes in use
`pcie_width`| pcie lanes in use
`pcie_speed`| current pcie speed
`pcie_interface_version`| current pcie generation
@@ -811,7 +811,7 @@ try:
else:
for device in devices:
pcie_link_status = amdsmi_get_pcie_info(device)
print(pcie_link_status["pcie_lanes"])
print(pcie_link_status["pcie_width"])
print(pcie_link_status["pcie_speed"])
print(pcie_link_status["pcie_interface_version"])
except AmdSmiException as e:
+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):
+3 -3
View File
@@ -751,9 +751,9 @@ class struct_pcie_metric_(Structure):
struct_pcie_metric_._pack_ = 1 # source:False
struct_pcie_metric_._fields_ = [
('pcie_speed', ctypes.c_uint32),
('pcie_lanes', ctypes.c_uint16),
('pcie_width', ctypes.c_uint16),
('PADDING_0', ctypes.c_ubyte * 2),
('pcie_speed', ctypes.c_uint32),
('pcie_bandwidth', ctypes.c_uint32),
('PADDING_1', ctypes.c_ubyte * 4),
('pcie_replay_count', ctypes.c_uint64),
@@ -769,7 +769,7 @@ class struct_pcie_static_(Structure):
struct_pcie_static_._pack_ = 1 # source:False
struct_pcie_static_._fields_ = [
('max_pcie_lanes', ctypes.c_uint16),
('max_pcie_width', ctypes.c_uint16),
('PADDING_0', ctypes.c_ubyte * 2),
('max_pcie_speed', ctypes.c_uint32),
('pcie_interface_version', ctypes.c_uint32),