#0: disable complex pytests

tenstorrent · Nov 30, 2023 · d2db6c1 · d2db6c1
1 parent 8387301
commit d2db6c1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/tt_eager/python_api_testing/unit_testing/test_complex.py b/tests/tt_eager/python_api_testing/unit_testing/test_complex.py
@@ -698,6 +698,7 @@ def test_level2_sub(bs, memcfg, dtype, device, function_level_defaults):
 )
 @pytest.mark.parametrize("dtype", ((ttl.tensor.DataType.BFLOAT16,)))
 @pytest.mark.parametrize("bs", ((1, 1), (1, 2), (2, 2)))
+@skip_for_wormhole_b0("Watcher error, see issue#4083")
 def test_level2_mul(bs, memcfg, dtype, device, function_level_defaults):
     input_shape = torch.Size([bs[0], bs[1], 32, 64])
     # check add
@@ -735,6 +736,7 @@ def test_level2_mul(bs, memcfg, dtype, device, function_level_defaults):
 )
 @pytest.mark.parametrize("dtype", ((ttl.tensor.DataType.BFLOAT16,)))
 @pytest.mark.parametrize("bs", ((1, 1), (1, 2), (2, 2)))
+@skip_for_wormhole_b0("Watcher error, see issue#4083")
 def test_level2_div(bs, memcfg, dtype, device, function_level_defaults):
     input_shape = torch.Size([bs[0], bs[1], 32, 64])
     # check add
@@ -772,6 +774,7 @@ def test_level2_div(bs, memcfg, dtype, device, function_level_defaults):
 )
 @pytest.mark.parametrize("dtype", ((ttl.tensor.DataType.BFLOAT16,)))
 @pytest.mark.parametrize("bs", ((1, 1), (1, 2), (2, 2)))
+@skip_for_wormhole_b0("Watcher error, see issue#4083")
 def test_level2_is_real(bs, memcfg, dtype, device, function_level_defaults):
     input_shape = torch.Size([bs[0], bs[1], 32, 64])
     # check abs
@@ -802,6 +805,7 @@ def test_level2_is_real(bs, memcfg, dtype, device, function_level_defaults):
 )
 @pytest.mark.parametrize("dtype", ((ttl.tensor.DataType.BFLOAT16,)))
 @pytest.mark.parametrize("bs", ((1, 1), (1, 2), (2, 2)))
+@skip_for_wormhole_b0("Watcher error, see issue#4083")
 def test_level2_is_imag(bs, memcfg, dtype, device, function_level_defaults):
     input_shape = torch.Size([bs[0], bs[1], 32, 64])
     # check abs