build universal2 package in CI (onnx#127)

Signed-off-by: daquexian <[email protected]>

Author: daquexian

.github/workflows/build-and-test.yml

@@ -5,6 +5,7 @@ on: [push, pull_request]
 jobs:
   build_wheels:
     env:
+      CIBW_ARCHS_MACOS: x86_64 universal2
       CIBW_MANYLINUX_X86_64_IMAGE: manylinux2014
       CIBW_BEFORE_ALL_LINUX: WD=`pwd` && /opt/python/cp38-cp38/bin/python -m pip install --target tmp_cmake cmake && cp tmp_cmake/bin/cmake /usr/local/bin/cmake && rm -rf tmp_cmake && /opt/python/cp38-cp38/bin/python -m pip install cmake && cmake --version && whereis cmake
       CIBW_BEFORE_ALL_MACOS: WD=`pwd` && pip install cmake

setup.py

@@ -20,6 +20,7 @@
 import subprocess
 import sys
 import platform
+import re
 from textwrap import dedent
 import multiprocessing
 
@@ -29,6 +30,7 @@
 CMAKE_BUILD_DIR = os.path.join(TOP_DIR, '.setuptools-cmake-build')
 
 WINDOWS = (os.name == 'nt')
+MACOS = sys.platform.startswith("darwin")
 
 CMAKE = find_executable('cmake')
 
@@ -186,6 +188,11 @@ def run(self):
                     cmake_args.extend(['-A', 'x64', '-T', 'host=x64'])
                 else:
                     cmake_args.extend(['-A', 'Win32', '-T', 'host=x86'])
+            if MACOS:
+                # Cross-compile support for macOS - respect ARCHFLAGS if set
+                archs = re.findall(r"-arch (\S+)", os.environ.get("ARCHFLAGS", ""))
+                if archs:
+                    cmake_args += ["-DCMAKE_OSX_ARCHITECTURES={}".format(";".join(archs))]
             if ONNX_ML:
                 cmake_args.append('-DONNX_ML=1')
             if ONNX_VERIFY_PROTO3: