#15833: Skip tests in case of 1x1 MeshDevice due to infra issue

tenstorrent · Dec 9, 2024 · 2a31ac7 · 2a31ac7
1 parent 1526c1f
commit 2a31ac7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/ttnn/unit_tests/test_sub_device.py b/tests/ttnn/unit_tests/test_sub_device.py
@@ -142,6 +142,8 @@ def test_sub_devices(device, enable_async_mode):
 @pytest.mark.parametrize("enable_async_mode", (False, True), indirect=True)
 @pytest.mark.parametrize("replicate_sub_devices", (False, True))
 def test_sub_devices_mesh(mesh_device, replicate_sub_devices, enable_async_mode):
+    if mesh_device.get_num_devices() == 1:
+        pytest.skip("#15833: Skipping test for single device mesh")
     run_sub_devices(mesh_device, replicate_sub_devices)
 
 
@@ -153,4 +155,6 @@ def test_sub_device_program(device, enable_async_mode):
 @pytest.mark.parametrize("enable_async_mode", (False, True), indirect=True)
 @pytest.mark.parametrize("replicate_sub_devices", (False, True))
 def test_sub_device_program_mesh(mesh_device, replicate_sub_devices, enable_async_mode):
+    if mesh_device.get_num_devices() == 1:
+        pytest.skip("#15833: Skipping test for single device mesh")
     run_sub_devices_program(mesh_device, replicate_sub_devices)