From 38f54098a7d1e6b874d6263767133538db4759fc Mon Sep 17 00:00:00 2001 From: dgolubovicTT Date: Thu, 14 Nov 2024 11:22:29 +0000 Subject: [PATCH] Add xfail to resnet and llama mlp --- forge/test/mlir/llama/tests/test_llama_mlp.py | 1 + forge/test/mlir/resnet/test_resnet_inference.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/forge/test/mlir/llama/tests/test_llama_mlp.py b/forge/test/mlir/llama/tests/test_llama_mlp.py index 184adf0c..6a338da1 100644 --- a/forge/test/mlir/llama/tests/test_llama_mlp.py +++ b/forge/test/mlir/llama/tests/test_llama_mlp.py @@ -10,6 +10,7 @@ @pytest.mark.parametrize("model_path", ["openlm-research/open_llama_3b", "meta-llama/Llama-3.2-1B"]) +@pytest.mark.xfail() @pytest.mark.push def test_llama_mlp(model_path): if model_path == "meta-llama/Llama-3.2-1B": diff --git a/forge/test/mlir/resnet/test_resnet_inference.py b/forge/test/mlir/resnet/test_resnet_inference.py index 4e032451..d8e2b1e6 100644 --- a/forge/test/mlir/resnet/test_resnet_inference.py +++ b/forge/test/mlir/resnet/test_resnet_inference.py @@ -11,6 +11,9 @@ @pytest.mark.push +@pytest.mark.xfail( + reason=" Metal issue: Can only tilize bfloat16 tensors. tracked on: https://github.com/tenstorrent/tt-metal/issues/14570" +) def test_resnet_inference(): # Compiler configurations compiler_cfg = forge.config._get_global_compiler_config()