#0: Test Mnist and ConvMnist integration tests

tenstorrent · Jan 29, 2025 · 3646be3 · 3646be3
1 parent 4034423
commit 3646be3
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 4 deletions.
diff --git a/models/demos/convnet_mnist/tests/test_performance.py b/models/demos/convnet_mnist/tests/test_performance.py
@@ -115,7 +115,6 @@ def test_convnet_mnist(
     ],
 )
 @pytest.mark.models_device_performance_bare_metal
-@pytest.mark.skip("#16824: Failing when whole suite is run on all archs")
 def test_perf_device_bare_metal_convnet_mnist(batch_size, expected_perf):
     subdir = "ttnn_convnet_mnist"
     num_iterations = 1

diff --git a/models/demos/mnist/tests/test_perf_mnist.py b/models/demos/mnist/tests/test_perf_mnist.py
@@ -102,7 +102,6 @@ def test_performance_mnist(device, batch_size, tt_mnist, model_location_generato
     logger.info("Exit MNIST perf test")
 
 
-@pytest.mark.skip("#16824: Failing when whole suite is run on all archs")
 @pytest.mark.parametrize(
     "batch_size",
     [128],

diff --git a/tests/ttnn/integration_tests/convnet_mnist/test_convnet_mnist.py b/tests/ttnn/integration_tests/convnet_mnist/test_convnet_mnist.py
@@ -26,7 +26,6 @@ def model_location_generator(rel_path):
         return Path("/opt/tt-metal-models") / rel_path
 
 
-@pytest.mark.skip("#16824: Failing when whole suite is run on all archs")
 @pytest.mark.parametrize("device_params", [{"l1_small_size": 16384}], indirect=True)
 def test_convnet_mnist(device, reset_seeds):
     model_path = model_location_generator("tt_dnn-models/ConvNetMNIST/")

diff --git a/tests/ttnn/integration_tests/mnist/test_mnist.py b/tests/ttnn/integration_tests/mnist/test_mnist.py
@@ -15,7 +15,6 @@
 from models.demos.mnist.tt import tt_mnist
 
 
-@pytest.mark.skip("#16824: Failing when whole suite is run on all archs")
 @pytest.mark.parametrize("device_params", [{"l1_small_size": 32768}], indirect=True)
 @pytest.mark.parametrize(
     "batch_size",