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,