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
nod-ai · Jan 7, 2025 · 12446b3 · 12446b3
1 parent 2c699fd
commit 12446b3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)