From 38ef30bdd000f56a56166062fc3216f555245a5c Mon Sep 17 00:00:00 2001 From: Marius Brehler Date: Mon, 16 Dec 2024 15:00:45 +0100 Subject: [PATCH] Skip `test_sharded_resnet_block_with_iree` --- .../models/punet/sharded_resnet_block_with_iree_test.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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 index c24dc149e..7c23e1d8c 100644 --- a/sharktank/tests/models/punet/sharded_resnet_block_with_iree_test.py +++ b/sharktank/tests/models/punet/sharded_resnet_block_with_iree_test.py @@ -4,6 +4,9 @@ # See https://llvm.org/LICENSE.txt for license information. # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +import pytest +import sys + from pathlib import Path import tempfile @@ -19,7 +22,6 @@ import iree.runtime from typing import List, Optional import os -import pytest vm_context: iree.runtime.VmContext = None @@ -231,6 +233,9 @@ def run_test_sharded_resnet_block_with_iree( @pytest.mark.xfail( torch.__version__ >= (2, 5), reason="https://github.com/nod-ai/shark-ai/issues/683" ) +@pytest.mark.skipif( + sys.platform == "win32", reason="https://github.com/nod-ai/shark-ai/issues/698" +) def test_sharded_resnet_block_with_iree( mlir_path: Optional[Path], module_path: Optional[Path],