[sharktank] Mark test expected to fail

Marks `test_sharded_resnet_block_with_iree` as expected to fail if running with torch>=2.5.0, see nod-ai#683.
marbre · Dec 12, 2024 · 4a7ff11 · 4a7ff11
1 parent beb0d00
commit 4a7ff11
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sharktank/tests/models/punet/sharded_resnet_block_with_iree_test.py b/sharktank/tests/models/punet/sharded_resnet_block_with_iree_test.py
@@ -228,6 +228,7 @@ def run_test_sharded_resnet_block_with_iree(
     strict=True,
     raises=AssertionError,
 )
+@pytest.mark.xfail(torch.__version__ >= (2,5), reason="https://github.com/nod-ai/shark-ai/issues/683")
 def test_sharded_resnet_block_with_iree(
     mlir_path: Optional[Path],
     module_path: Optional[Path],