Add xfail to resnet and llama mlp

tenstorrent · Nov 14, 2024 · 38f5409 · 38f5409
1 parent 4691171
commit 38f5409
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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()