diff --git a/.github/workflows/build_linux_wheel/action.yml b/.github/workflows/build_linux_wheel/action.yml index 2c4927e0..d92428e1 100644 --- a/.github/workflows/build_linux_wheel/action.yml +++ b/.github/workflows/build_linux_wheel/action.yml @@ -52,12 +52,7 @@ runs: args: ${{ inputs.args }} before-script-linux: | set -e - apt install -y unzip - if [ $(uname -m) = "x86_64" ]; then - PROTOC_ARCH="x86_64" - else - PROTOC_ARCH="aarch_64" - fi - curl -L https://github.com/protocolbuffers/protobuf/releases/download/v24.4/protoc-24.4-linux-$PROTOC_ARCH.zip > /tmp/protoc.zip \ + yum install -y openssl-devel clang \ + && curl -L https://github.com/protocolbuffers/protobuf/releases/download/v24.4/protoc-24.4-linux-aarch_64.zip > /tmp/protoc.zip \ && unzip /tmp/protoc.zip -d /usr/local \ && rm /tmp/protoc.zip diff --git a/.github/workflows/pypi-publish.yml b/.github/workflows/pypi-publish.yml index 6a725824..6dc2cdeb 100644 --- a/.github/workflows/pypi-publish.yml +++ b/.github/workflows/pypi-publish.yml @@ -15,15 +15,21 @@ jobs: - platform: x86_64 manylinux: "2_17" extra_args: "" + runner: ubuntu-22.04 - platform: x86_64 manylinux: "2_28" extra_args: "--features fp16kernels" + runner: ubuntu-22.04 - platform: aarch64 - manylinux: "2_24" + manylinux: "2_17" extra_args: "" - # We don't build fp16 kernels for aarch64, because it uses - # cross compilation image, which doesn't have a new enough compiler. - runs-on: "ubuntu-22.04" + # For successful fat LTO builds, we need a large runner to avoid OOM errors. + runner: ubuntu-2404-8x-arm64 + - platform: aarch64 + manylinux: "2_28" + extra_args: "--features fp16kernels" + runner: ubuntu-2404-8x-arm64 + runs-on: ${{ matrix.config.runner }} steps: - uses: actions/checkout@v4 with: