From 12446b3c00f541850bf0217596d044035fccea17 Mon Sep 17 00:00:00 2001
From: IanNod <45800100+IanNod@users.noreply.github.com>
Date: Tue, 7 Jan 2025 15:55:06 -0800
Subject: [PATCH] Mark sharktank vae iree test as xfail (#775)

Xfails vae until iree runtime issue on the CI machine is fixed. Resolves
sharktank data-dependent tests
---
 sharktank/tests/models/vae/vae_test.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/sharktank/tests/models/vae/vae_test.py b/sharktank/tests/models/vae/vae_test.py
index 99454f6cf..9b77d835e 100644
--- a/sharktank/tests/models/vae/vae_test.py
+++ b/sharktank/tests/models/vae/vae_test.py
@@ -100,6 +100,9 @@ def testCompareF16EagerVsHuggingface(self):
 
         torch.testing.assert_close(ref_results, results)
 
+    @pytest.mark.xfail(
+        reason="Waiting on fix for https://github.com/iree-org/iree/issues/19623"
+    )
     def testVaeIreeVsHuggingFace(self):
         dtype = getattr(torch, "float32")
         inputs = get_random_inputs(dtype=dtype, device="cpu", bs=1)