win

Signed-off-by: Xavier Dupre <xadupre@microsoft.com>
microsoft · Jun 26, 2024 · 9ee3e08 · 9ee3e08
1 parent 5d6621d
commit 9ee3e08
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/onnxscript/backend/onnx_export_test.py b/onnxscript/backend/onnx_export_test.py
@@ -92,6 +92,16 @@ def skip(pattern: str | Pattern, reason: str, *, condition: bool = True):
     skip(r"^test_ai_onnx_ml_label_encoder", "ONNX Runtime does not support Opset 21 at 1.17"),
 )
 
+if sys.platform == "win32":
+    SKIP_TESTS = (
+        *SKIP_TESTS,
+        skip(r"^test_gemm_beta", "cannot import module, import_module does not work"),
+        skip(
+            r"^test_averagepool_2d_default",
+            "cannot import module, import_module does not work",
+        ),
+    )
+
 
 def load_function(obj):
     return ort.InferenceSession(obj.SerializeToString(), providers=("CPUExecutionProvider",))