Arm64 build (#73)

G-Research · Nov 6, 2024 · b11efdd · b11efdd
1 parent 1a2b5b3
commit b11efdd
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 9 deletions.
diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml
@@ -51,12 +51,18 @@ jobs:
   build:
     needs: test
     if: github.event_name == 'schedule' || github.event_name == 'push' || github.event.pull_request.head.repo.id != github.event.pull_request.base.repo.id
+    runs-on: ${{ matrix.runner }}
     strategy:
       matrix:
-        os: [ubuntu-latest, windows-latest]
+        include:
+          - runner: ubuntu-latest
+            arch: x64
+          - runner: windows-latest
+            arch: x64
+          - runner: ubuntu-22.04-arm64
+            arch: arm64
       fail-fast: false
 
-    runs-on: ${{ matrix.os }}
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python 3.10
@@ -76,7 +82,7 @@ jobs:
     - name: Compute vcpkg triplet and root
       id: vcpkg-info
       run: |
-        triplet="x64-"
+        triplet="${{ matrix.arch }}-"
         case ${{ runner.os }} in
           Linux)
             triplet+="linux"
@@ -136,22 +142,29 @@ jobs:
         # We use manylinux_2_28 for ABI compatibility with pyarrow
         # With the default image we were getting "undefined symbol: _ZNK5arrow6Status8ToStringEv" error (e.g https://github.com/ray-project/ray/issues/24566) 
         CIBW_MANYLINUX_X86_64_IMAGE: manylinux_2_28 
+        CIBW_MANYLINUX_AARCH64_IMAGE: manylinux_2_28
         # Disable unsupported builds
         CIBW_SKIP: "pp* *_i686 *-musllinux_* *win32 cp313-*"
 
     - uses: actions/upload-artifact@v4
       with:
-        name: dist-${{ matrix.os }}
+        name: dist-${{ matrix.runner }}
         path: ./python/dist/*
 
   test-binary:
     needs: build
+    runs-on: ${{ matrix.runner }}
     strategy:
       matrix:
-        os: [ubuntu-latest, windows-latest]
+        include:
+          - runner: ubuntu-latest
+            arch: x64
+          - runner: windows-latest
+            arch: x64
+          - runner: ubuntu-22.04-arm64
+            arch: arm64
       fail-fast: false
 
-    runs-on: ${{ matrix.os }}
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python 3.10
@@ -189,12 +202,18 @@ jobs:
 
   benchmarks:
     needs: build
+    runs-on: ${{ matrix.runner }}
     strategy:
       matrix:
-        os: [ubuntu-latest, windows-latest]
+        include:
+          - runner: ubuntu-latest
+            arch: x64
+          - runner: windows-latest
+            arch: x64
+          - runner: ubuntu-22.04-arm64
+            arch: arm64
       fail-fast: false
 
-    runs-on: ${{ matrix.os }}
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python 3.10

diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -10,7 +10,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "palletjack"
-version = "2.3.0"
+version = "2.3.1"
 description = "Faster parquet metadata reading"
 readme = "README.md"
 requires-python = ">=3.9"