#0: unet shallow skip perf mode if GS

tenstorrent · Apr 12, 2024 · a57b5d5 · a57b5d5
1 parent 8ae4284
commit a57b5d5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/ttnn/integration_tests/unet/test_ttnn_shallow_unet.py b/tests/ttnn/integration_tests/unet/test_ttnn_shallow_unet.py
@@ -513,6 +513,9 @@ def forward(self, x):
 @pytest.mark.parametrize("loop", [0])
 @pytest.mark.parametrize("perf_mode, groups", [(False, 1), (True, 1)])  # , (True, 2)])
 def test_unet(device, loop, perf_mode, groups):
+    if perf_mode and device.arch() == ttl.device.Arch.GRAYSKULL:
+        pytest.skip("Perf mode is not supported on Grayskull")
+
     with torch.no_grad():
         torch.manual_seed(0)
         torch_model = UNet(groups=groups)