build: arm support for linux (#239)

Author: bagel897

.circleci/config.yml

@@ -84,15 +84,11 @@ commands:
             - $GITHUB_WORKSPACE
           key: v1-verified-codemod-repos-{{ checksum "tests/integration/verified_codemods/codemod_data/repo_commits.json" }}-{{.Environment.CIRCLE_NODE_INDEX}}-{{.Environment.CIRCLE_NODE_TOTAL}}
   build-wheels:
-    parameters:
-      base_packages:
-        type: boolean
-        default: false
     steps:
     - install-uv
     - restore_cache:
         keys:
-          - cibuildwheel-cache-<<pipeline.parameters.python_version>>-{{ .Environment.CIRCLE_JOB }}-{{ checksum "uv.lock" }}--{{ checksum "pyproject.toml" }}
+          - cibuildwheel-cache-<<pipeline.parameters.python_version>>-{{ .Environment.CIRCLE_JOB }}-{{ checksum "uv.lock" }}--{{ checksum "pyproject.toml" }}-{{ arch }}
     - run:
         command: |
           uv run --frozen cibuildwheel --output-dir dist
@@ -102,7 +98,7 @@ commands:
     - save_cache:
         paths:
           - /home/circleci/.cache/cibuildwheel
-        key: cibuildwheel-cache-<<pipeline.parameters.python_version>>-{{ .Environment.CIRCLE_JOB }}-{{ checksum "uv.lock" }}--{{ checksum "pyproject.toml" }}
+        key: cibuildwheel-cache-<<pipeline.parameters.python_version>>-{{ .Environment.CIRCLE_JOB }}-{{ checksum "uv.lock" }}--{{ checksum "pyproject.toml" }}-{{ arch }}
   setup-uv:
     steps:
     - install-lfs
@@ -339,18 +335,14 @@ jobs:
       resource_class:
         type: string
         default: "large"
-      base_packages:
-        type: boolean
-        default: false
     working_directory: ~/linux-wheels
     machine:
       image: ubuntu-2404:2024.05.1
       docker_layer_caching: true
     resource_class: <<parameters.resource_class>>
     steps:
       - checkout
-      - build-wheels:
-          base_packages: << parameters.base_packages >>
+      - build-wheels
       - persist_to_workspace:
           root: .
           paths:
@@ -503,7 +495,10 @@ workflows:
           filters:
             tags:
               only: /^v.*/
-          base_packages: true
+          matrix:
+            parameters:
+              resource_class: [large, arm.large]
+
       - osx-wheels:
           filters:
             tags:

README.md

@@ -46,7 +46,8 @@ We support
 
 - Running Codegen in Python 3.12 – 3.13
 - macOS and Linux
-  - We develop on macOS but our testing and production environments are Linux
+  - macOS is supported on Apple Silicon
+  - Linux is supported on x86_64 and aarch64 with glibc 2.34+
   - Windows is not supported
 - Python, Typescript, Javascript and React codebases
 

pyproject.toml

@@ -226,7 +226,15 @@ python-levenshtein = ["Levenshtein"]
 PyGithub = ["github"]
 [tool.cibuildwheel]
 build-frontend = "build[uv]"
-skip = ["*-musllinux_i686", "*-manylinux_i686", "*-musllinux_x86_64"]
+skip = [
+  "*-musllinux_i686",
+  "*-manylinux_i686",
+  "*-musllinux_x86_64",
+  "*-musllinux_aarch64",
+]
 environment = { "HATCH_BUILD_HOOKS_ENABLE" = "true" }
 manylinux-x86_64-image = "quay.io/pypa/manylinux_2_34_x86_64"
 manylinux-aarch64-image = "quay.io/pypa/manylinux_2_34_aarch64"
+[tool.cibuildwheel.linux]
+before-all = "curl -sSf https://sh.rustup.rs | sh -s -- -y"
+environment = { "PATH" = "$HOME/.cargo/bin:$PATH" }