From 7b7c3088f9552d61b78bc37ea31eda04bc961456 Mon Sep 17 00:00:00 2001 From: kkeerthana0573 Date: Tue, 21 Jan 2025 10:29:29 +0000 Subject: [PATCH] #0: Test Mnist and ConvMnist integration tests --- models/demos/convnet_mnist/tests/test_performance.py | 1 - models/demos/mnist/tests/test_perf_mnist.py | 1 - tests/ttnn/integration_tests/convnet_mnist/test_convnet_mnist.py | 1 - tests/ttnn/integration_tests/mnist/test_mnist.py | 1 - 4 files changed, 4 deletions(-) diff --git a/models/demos/convnet_mnist/tests/test_performance.py b/models/demos/convnet_mnist/tests/test_performance.py index cd899342609e..f3e2c6571563 100644 --- 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 index 45de49507795..511afa20f79f 100644 --- 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 index 4036c7b3ad84..f09722d5a655 100644 --- 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 index be79057f7d6c..e4ea5143a08a 100644 --- 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",