[sharktank] Test additional version on windows (#697)

.github/workflows/ci-sharktank.yml

@@ -33,6 +33,9 @@ jobs:
           - os: windows-2022
             python-version: "3.11"
             torch-version: "2.3.0"
+          - os: windows-2022
+            python-version: "3.12"
+            torch-version: "2.4.1"
         exclude:
           - python-version: "3.12"
              # `torch.compile` requires torch>=2.4.0 for Python 3.12+

sharktank/tests/examples/main_test.py

@@ -4,11 +4,16 @@
 # See https://llvm.org/LICENSE.txt for license information.
 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
+import pytest
+import sys
 import unittest
 
 from sharktank.utils.testing import MainRunnerTestBase
 
 
+@pytest.mark.skipif(
+    sys.platform == "win32", reason="https://github.com/nod-ai/shark-ai/issues/698"
+)
 class ShardingTests(MainRunnerTestBase):
     def testExportFfnNet(self):
         from sharktank.examples.sharding.export_ffn_net import main

sharktank/tests/layers/sharded_conv2d_with_iree_test.py

@@ -7,6 +7,7 @@
 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
 import pytest
+import sys
 
 from pathlib import Path
 import tempfile
@@ -186,6 +187,9 @@ def run_test_sharded_conv2d_with_iree(
 @pytest.mark.xfail(
     torch.__version__ >= (2, 5), reason="https://github.com/nod-ai/shark-ai/issues/682"
 )
+@pytest.mark.skipif(
+    sys.platform == "win32", reason="https://github.com/nod-ai/shark-ai/issues/698"
+)
 def test_sharded_conv2d_with_iree(
     mlir_path: Optional[Path],
     module_path: Optional[Path],

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],