From e725ce10536504031ce7b102673f05697b3be9f0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20F=C3=A4rber?= Date: Sat, 14 Sep 2024 14:52:13 +0200 Subject: [PATCH] Suppress Wormhole devices from table of resettable devices on Arm MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Since no better interface is offered by tt-tools-common yet, query the system_compatibility() dictionary. Resolves issue #43. Signed-off-by: Andreas Färber --- tt_smi/tt_smi_backend.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tt_smi/tt_smi_backend.py b/tt_smi/tt_smi_backend.py index 8c29b9f..4b348f5 100644 --- a/tt_smi/tt_smi_backend.py +++ b/tt_smi/tt_smi_backend.py @@ -28,6 +28,7 @@ from tt_tools_common.reset_common.galaxy_reset import GalaxyReset from tt_tools_common.utils_common.system_utils import ( get_host_info, + system_compatibility, ) from tt_tools_common.utils_common.tools_utils import ( get_board_type, @@ -146,6 +147,7 @@ def print_all_available_devices(self): f"{board_id}", ) console.print(table_1) + sys_compat = system_compatibility() table_2 = Table(title="Boards that can be reset:") table_2.add_column("Pci Dev ID") table_2.add_column("Board Type") @@ -154,6 +156,7 @@ def print_all_available_devices(self): for i, device in enumerate(self.devices): if ( not device.is_remote() + and not (device.as_wh() and not sys_compat["WH Reset"][0]) and self.device_infos[i]["board_type"] != "GALAXY" ): board_id = self.device_infos[i]["board_id"]