From 08ff52dbb2d278cc6ae3f43a3276b264207b4fac Mon Sep 17 00:00:00 2001 From: Sankar Manoj Date: Fri, 21 Feb 2025 22:54:27 +0000 Subject: [PATCH] #0: Skip test on N300 --- tests/ttnn/unit_tests/operations/test_new_conv2d.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/ttnn/unit_tests/operations/test_new_conv2d.py b/tests/ttnn/unit_tests/operations/test_new_conv2d.py index 8bb9d1f54a0..c9e6e60576e 100644 --- a/tests/ttnn/unit_tests/operations/test_new_conv2d.py +++ b/tests/ttnn/unit_tests/operations/test_new_conv2d.py @@ -2798,6 +2798,9 @@ def test_small_in_large_out_channels_auto_shard(device, torch_tensor_map): padding = (0, 0) height = 128 width = 128 + if device.core_grid.y != 8 and is_wormhole_b0(): + pytest.skip("Needs 8x8 grid for wormhole_b0") + run_conv( device, torch_tensor_map,