skip windows test_gpu

Author: makslevental

.github/workflows/release.yml

@@ -11,23 +11,23 @@ on:
         description: 'bundle mlir python bindings'
         type: string
         default: 1
-  release:
-    types:
-      - published
-  push:
-    branches:
-      - main
-  pull_request:
-    paths:
-      - .github/workflows/release.yml
+#  release:
+#    types:
+#      - published
+#  push:
+#    branches:
+#      - main
+#  pull_request:
+#    paths:
+#      - .github/workflows/release.yml
 
 concurrency:
   group: ci-build-release-${{ github.event.number || github.sha }}
   cancel-in-progress: true
 
 env:
   SYSTEM_VERSION_COMPAT: 0
-  PIP_FIND_LINKS: "https://github.com/llvm/eudsl/releases/expanded_assets/latest https://github.com/makslevental/mlir-wheels/releases/expanded_assets/latest"
+  PIP_FIND_LINKS: "https://github.com/makslevental/mlir-wheels/releases/expanded_assets/latest"
 
 jobs:
 

.github/workflows/test.yml

@@ -21,7 +21,7 @@ concurrency:
 
 env:
   SYSTEM_VERSION_COMPAT: 0
-  PIP_FIND_LINKS: "https://github.com/llvm/eudsl/releases/expanded_assets/latest https://github.com/makslevental/mlir-wheels/releases/expanded_assets/latest"
+  PIP_FIND_LINKS: "https://github.com/makslevental/mlir-wheels/releases/expanded_assets/latest"
 
 jobs:
 
@@ -42,6 +42,13 @@ jobs:
           - os: macos-14
             py_version: "3.9"
 
+          # <frozen importlib._bootstrap>:491: Warning: Numpy built with MINGW-W64 on Windows 64 bits is experimental, and only available for testing. You are advised not to use it for production.
+          - os: windows-2022
+            py_version: "3.14"
+
+          - os: windows-2022
+            py_version: "3.14t"
+
     name: "${{ matrix.os }}-${{ matrix.py_version }}"
 
     steps:
@@ -64,7 +71,11 @@ jobs:
         shell: bash
         run: |
           
-          pytest -v tests
+          # so that loaded modules isn't reused between modules
+          for f in tests/test*.py; do
+            echo "Testing $f"
+            pytest -v $f
+          done
 
       - name: Test mwe
         shell: bash
@@ -148,6 +159,14 @@ jobs:
         os: [ ubuntu-22.04, macos-13, macos-14, windows-2022, ubuntu-22.04-arm ]
         py_version: [ "3.12", "3.13", "3.14", "3.14t" ]
 
+        # <frozen importlib._bootstrap>:491: Warning: Numpy built with MINGW-W64 on Windows 64 bits is experimental, and only available for testing. You are advised not to use it for production.
+        exclude:
+          - os: windows-2022
+            py_version: "3.14"
+
+          - os: windows-2022
+            py_version: "3.14t"
+
     steps:
       - name: Checkout
         uses: actions/checkout@v2
@@ -179,7 +198,15 @@ jobs:
       fail-fast: false
       matrix:
         os: [ ubuntu-22.04, macos-13, macos-14, windows-2022, ubuntu-22.04-arm ]
-        py_version: [ "3.12", "3.13" ]
+        py_version: [ "3.12", "3.13", "3.14", "3.14t" ]
+
+        # <frozen importlib._bootstrap>:491: Warning: Numpy built with MINGW-W64 on Windows 64 bits is experimental, and only available for testing. You are advised not to use it for production.
+        exclude:
+          - os: windows-2022
+            py_version: "3.14"
+
+          - os: windows-2022
+            py_version: "3.14t"
 
     steps:
 

mlir/extras/ast/py_type.py

@@ -34,7 +34,6 @@ def __new__(cls, *args, **kwargs):
         return pointer(*args, **kwargs)
 
     def __class_getitem__(cls, item):
-        """Return a `ctypes.POINTER` of the given type."""
         # For ptr[Self], return a special object
         if item is Self:
             return _SelfPtr
@@ -128,11 +127,9 @@ class PyObject(Structure):
     _fields_ = _py_object_fields
 
     def as_ref(self) -> _Ptr[Self]:
-        """Return a pointer to the Structure."""
         return pointer(self)  # type: ignore
 
     def into_object(self):
-        """Cast the PyObject into a Python object."""
         py_obj = cast(self.as_ref(), py_object)
         return py_obj.value
 
@@ -200,7 +197,6 @@ class PyTypeObject(Structure):
 
     @classmethod
     def from_object(cls, obj) -> Self:
-        """Create a PyObject from an object."""
         return cls.from_address(address(obj))
 
 
@@ -214,7 +210,6 @@ class PyTypeVarObject(Structure):
 
     @classmethod
     def from_object(cls, obj) -> Self:
-        """Create a PyObject from an object."""
         return cls.from_address(address(obj))
 
 

requirements-dev.txt

@@ -1,4 +1,4 @@
 black
 inflection
 pytest
-astpretty
+astpretty

tests/conftest.py

@@ -0,0 +1,4 @@
+# to prevent eg test_transformers.py from erroring because no tests are run
+def pytest_sessionfinish(session, exitstatus):
+    if exitstatus == 5:
+        session.exitstatus = 0  # Any arbitrary custom status you want to return```

tests/test_runtime.py

@@ -855,25 +855,25 @@ def tenfoo(
     # # assert np.array_equal(A + B, C)
 
 
-# def test_raii_context():
-#     def foo():
-#         ctx = RAIIMLIRContext()
-#         mod = Module.create()
-#         with InsertionPoint(mod.body):
-#
-#             @func(emit=True)
-#             def foo(x: T.i32()):
-#                 return x
-#
-#         # CHECK:  func.func @foo(%[[VAL_0:.*]]: i32) -> i32 {
-#         # CHECK:    return %[[VAL_0]] : i32
-#         # CHECK:  }
-#
-#         filecheck_with_comments(mod)
-#
-#     foo()
-#
-#     assert Context.current is None
+def test_raii_context():
+    def foo():
+        ctx = RAIIMLIRContext()
+        mod = Module.create()
+        with InsertionPoint(mod.body):
+
+            @func(emit=True)
+            def foo(x: T.i32()):
+                return x
+
+        # CHECK:  func.func @foo(%[[VAL_0:.*]]: i32) -> i32 {
+        # CHECK:    return %[[VAL_0]] : i32
+        # CHECK:  }
+
+        filecheck_with_comments(mod)
+
+    foo()
+
+    assert Context.current is None
 
 
 def test_explicit_module():

tests/test_transform.py

@@ -1,6 +1,12 @@
 from textwrap import dedent
 
 import pytest
+
+# noinspection PyUnresolvedReferences
+# has to be first to set nanobind.nb_type_0
+import mlir.extras
+
+
 from mlir.dialects import linalg as linalg_dialect, arith as arith_dialect
 from mlir.dialects import pdl
 from mlir.dialects.builtin import module