feat: produce ABI-agnostic wheels for python (tensorchord#1324)

* feat: producce abi3 wheels for cpython

Signed-off-by: Frost Ming <[email protected]>

* remove file added unexpectedly

Signed-off-by: Frost Ming <[email protected]>

* Forcely write py2.py3-none wheels

Signed-off-by: Frost Ming <[email protected]>

* Apply suggestions from code review

Co-authored-by: Jinjing Zhou <[email protected]>
Signed-off-by: Frost Ming <[email protected]>

Signed-off-by: Frost Ming <[email protected]>
Co-authored-by: Jinjing Zhou <[email protected]>

Author: frostming

.github/workflows/release.yml

@@ -78,9 +78,11 @@ jobs:
       env:
         CIBW_ARCHS_MACOS: arm64, x86_64
         CIBW_ARCHS_LINUX: auto64
+        CIBW_SKIP: pp*
     - name: Build source distribution
       if: runner.os == 'Linux' # Only release source under linux to avoid conflict
       run: |
+        python3 -m pip install wheel
         python3 setup.py sdist
         mv dist/*.tar.gz wheelhouse/
     - name: Upload to PyPI

pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"

setup.py

@@ -16,6 +16,9 @@
 from setuptools.command.sdist import sdist
 from setuptools import setup, Extension, find_packages
 from setuptools.command.build_ext import build_ext
+from wheel.bdist_wheel import bdist_wheel
+
+
 import subprocess
 import logging
 
@@ -53,6 +56,12 @@ def build_extension(self, ext: Extension) -> None:
         build_envd_if_not_found()
 
 
+class bdist_wheel_universal(bdist_wheel):
+    def get_tag(self):
+        *_, plat = super().get_tag()
+        return "py2.py3", "none", plat
+
+
 class SdistCommand(sdist):
     def run(self):
         errno = subprocess.call(["make", "generate-git-tag-info"])
@@ -110,5 +119,9 @@ def get_version():
     ext_modules=[
         EnvdExtension(name="envd", sources=["cmd/*"]),
     ],
-    cmdclass=dict(build_ext=EnvdBuildExt, sdist=SdistCommand),
+    cmdclass=dict(
+        build_ext=EnvdBuildExt,
+        sdist=SdistCommand,
+        bdist_wheel=bdist_wheel_universal,
+    ),
 )