MAINT: Build abi3 wheels and use native arm64 Linux

Author: larsoner

.github/workflows/wheels.yml

@@ -59,10 +59,10 @@ jobs:
       matrix:
         buildplat:
           - [ubuntu-latest, musllinux_x86_64]
-          - [ubuntu-latest, manylinux_aarch64]
+          - [ubuntu-24.04-arm64, manylinux_aarch64]
           - [macos-15-intel, macosx_x86_64]  # native Intel hardware
           - [windows-latest, win_amd64]
-        python: ["cp310", "cp311", "cp312", "cp313", "cp314"]
+        python: ["cp310", "cp311"]
         include:
           # Manylinux and arm64 builds (on native hardware) are cheap, do all in one
           - { buildplat: ["ubuntu-latest", "manylinux_x86_64"], python: "*" }
@@ -76,13 +76,6 @@ jobs:
       - name: Install the latest version of uv
         uses: astral-sh/setup-uv@v6
 
-      # For aarch64 support
-      # https://cibuildwheel.pypa.io/en/stable/faq/#emulation
-      - uses: docker/setup-qemu-action@v3
-        with:
-          platforms: all
-        if: runner.os == 'Linux' && endsWith(matrix.buildplat[1], 'aarch64')
-
       - name: Build wheel(s)
         run: uvx cibuildwheel
         env:

pyproject.toml

@@ -64,19 +64,37 @@ test = [
 [tool.cibuildwheel]
 # Disable PyPy
 skip = "pp*"
-
+# 3.11 is abi3
+build = "cp310-* cp311-*"
 # 64-bit builds only; 32-bit builds seem pretty niche these days, so
 # don't bother unless someone asks
 archs = ["native"]
-
+before-build = "pip install auditwheel"
 test-requires = [
     "pytest",
     "nitime[full]",  # Enable all optional behavior
 ]
 test-command = "pytest -rsx --pyargs nitime"
 
 [tool.cibuildwheel.linux]
-archs = ["x86_64", "aarch64"]
+repair-wheel-command = [
+  "auditwheel repair -w {dest_dir} {wheel}",
+  "bash tools/audit_wheel.sh {wheel}",
+]
+
+[tool.cibuildwheel.macos]
+archs = ["native"]
+repair-wheel-command = [
+  "delocate-wheel --require-archs {delocate_archs} -w {dest_dir} -v {wheel}",
+  "bash tools/audit_wheel.sh {wheel}",
+]
+
+[tool.cibuildwheel.windows]
+before-build = "pip install delvewheel abi3audit"
+repair-wheel-command = [
+  "delvewheel repair -w {dest_dir} {wheel}",
+  "bash tools/audit_wheel.sh {wheel}",
+]
 
 [tool.pytest.ini_options]
 minversion = "8"

setup.py

@@ -4,26 +4,46 @@
 This file only contains cython components.
 See pyproject.toml for the remaining configuration.
 """
+import platform
+import sys
+
 from setuptools import setup
+from setuptools import Extension
+from Cython.Build import cythonize
+from numpy import get_include
+
+# add Cython extensions to the setup options
+
+
+# https://github.com/joerick/python-abi3-package-sample/blob/main/setup.py
+class bdist_wheel_abi3(bdist_wheel):  # noqa: D101
+    def get_tag(self):  # noqa: D102
+        python, abi, plat = super().get_tag()
+
+        if python.startswith("cp"):
+            return "cp311", "abi3", plat
+
+        return python, abi, plat
+
+macros = [('NPY_NO_DEPRECATED_API', 'NPY_1_7_API_VERSION')]
+ext_kwargs = {}
+setup_kwargs = {}
+if sys.version_info.minor >= 11 and platform.python_implementation() == "CPython":
+    # Can create an abi3 wheel (typed memoryviews first available in 3.11)!
+    macros.append(("Py_LIMITED_API", "0x030B0000"))
+    ext_kwargs["py_limited_api"] = True
+    setup_kwargs["cmdclass"] = {"bdist_wheel": bdist_wheel_abi3}
+
 
-try:
-    from setuptools import Extension
-    from Cython.Build import cythonize
-    from numpy import get_include
-
-    # add Cython extensions to the setup options
-    exts = [
-        Extension(
-            'nitime._utils',
-            ['nitime/_utils.pyx'],
-            include_dirs=[get_include()],
-            define_macros=[('NPY_NO_DEPRECATED_API', 'NPY_1_7_API_VERSION')],
-        )
-    ]
-    opts = {'ext_modules': cythonize(exts, language_level='3')}
-except ImportError:
-    # no loop for you!
-    opts = {}
+exts = [
+    Extension(
+        'nitime._utils',
+        ['nitime/_utils.pyx'],
+        include_dirs=[get_include()],
+        define_macros=macros,
+    )
+]
+opts = {'ext_modules': cythonize(exts, language_level='3'), **setup_kwargs}
 
 # Now call the actual setup function
 if __name__ == '__main__':

tools/audit_wheel.sh

@@ -0,0 +1,9 @@
+#!/bin/bash -eo pipefail
+set -x
+
+PY_MINOR=$(python -c "import sys; print(sys.version_info.minor)")
+if [ "$PY_MINOR" -lt 11 ]; then
+  echo "Not checking abi3audit for Python $PY_MINOR < 3.11"
+  exit 0
+fi
+abi3audit --strict --report --verbose "$1"