AI/rocm-systems
2
0 Vork 0
Code Kwesties Pull-aanvragen Actions Paketten Projecten Publicaties Wiki Activiteit

Unify the amdsmi_get_pcie_info python interface

Bladeren bron 
Make the python interface consistent with the C interface.

Change-Id: Idda08f888947c757e475d5a024b0ec3d8e1d846a
This commit is contained in:
Bill(Shuzhou) Liu authored and Maisam Arif committed 2024-02-22 03:33:59 -05:00
1 bovenliggende f58613561c
commit db33cda0c1
7 files changed
+36 -22
Download Patch-bestand Download Diff-bestand Expand all files Collapse all files

No files matched your search

amdsmi_cli/amdsmi_commands.py
+9 -6
Bestand weergeven
@@ -353,7 +353,10 @@ class AMDSMICommands():
try:
link_caps = amdsmi_interface.amdsmi_get_pcie_info(args.gpu)
bus_info.update(link_caps)
bus_info['max_pcie_speed'] = link_caps['pcie_static']['max_pcie_speed']
bus_info['pcie_slot_type'] = link_caps['pcie_static']['slot_type']
bus_info['pcie_interface_version'] = link_caps['pcie_static']['pcie_interface_version']
if bus_info['max_pcie_speed'] % 1000 != 0:
pcie_speed_GTs_value = round(bus_info['max_pcie_speed'] / 1000, 1)
else:
@@ -1396,17 +1399,17 @@ class AMDSMICommands():
try:
pcie_link_status = amdsmi_interface.amdsmi_get_pcie_info(args.gpu)
if pcie_link_status['pcie_speed'] % 1000 != 0:
pcie_speed_GTs_value = round(pcie_link_status['pcie_speed'] / 1000, 1)
if pcie_link_status['pcie_metric']['pcie_speed'] % 1000 != 0:
pcie_speed_GTs_value = round(pcie_link_status['pcie_metric']['pcie_speed'] / 1000, 1)
else:
pcie_speed_GTs_value = round(pcie_link_status['pcie_speed'] / 1000)
pcie_speed_GTs_value = round(pcie_link_status['pcie_metric']['pcie_speed'] / 1000)
pcie_dict['current_speed'] = pcie_speed_GTs_value
pcie_dict['current_lanes'] = pcie_link_status['pcie_lanes']
pcie_dict['current_lanes'] = pcie_link_status['pcie_metric']['pcie_width']
if self.logger.is_human_readable_format():
unit = 'GT/s'
pcie_dict['current_lanes'] = f"{pcie_link_status['pcie_lanes']} lanes"
pcie_dict['current_lanes'] = f"{pcie_link_status['pcie_metric']['pcie_width']} lanes"
pcie_dict['current_speed'] = f"{pcie_dict['current_speed']} GT/s"
except amdsmi_exception.AmdSmiLibraryException as e: