#0: Skip test on N300

tenstorrent · Feb 21, 2025 · 08ff52d · 08ff52d
1 parent b8f6c7f
commit 08ff52d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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,