up

Author: metascroy

.github/workflows/trunk.yml

@@ -593,19 +593,23 @@ jobs:
       contents: read
     strategy:
       matrix:
-        model: [qwen3_4b, phi_4_mini]
-        runner: [linux.2xlarge]
-        docker-image: [executorch-ubuntu-22.04-clang12]
-        backend: [xnnpack]
         include:
           - model: qwen3_4b
+            backend: xnnpack
+            runner: linux.2xlarge
+            docker-image: executorch-ubuntu-22.04-clang12
+          - model: phi_4_mini
+            backend: xnnpack
+            runner: linux.2xlarge
+            docker-image: executorch-ubuntu-22.04-clang12
+          - model: qwen3_4b
+            backend: torchao
             runner: linux.arm64.2xlarge
             docker-image: executorch-ubuntu-22.04-gcc11-aarch64
-            backend: torchao
           - model: phi_4_mini
+            backend: torchao
             runner: linux.arm64.2xlarge
             docker-image: executorch-ubuntu-22.04-gcc11-aarch64
-            backend: torchao
       fail-fast: false
     with:
       runner: ${{ matrix.runner }}
@@ -618,17 +622,11 @@ jobs:
         CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
         conda activate "${CONDA_ENV}"
 
+        PYTHON_EXECUTABLE=python bash .ci/scripts/setup-linux.sh --build-tool cmake
+
         if [[ "${{ matrix.backend }}" == "torchao" ]]; then
-          PYTHON_EXECUTABLE=python \
-              BUILD_TORCHAO_EXPERIMENTAL=1 \
-              TORCHAO_BUILD_CPU_AARCH64=1 \
-              TORCHAO_ENABLE_ARM_NEON_DOT=1 \
-              TORCHAO_BUILD_KLEIDIAI=1 \
-              bash .ci/scripts/setup-linux.sh --build-tool cmake
+          BUILD_TORCHAO_EXPERIMENTAL=1 TORCHAO_BUILD_CPU_AARCH64=1 TORCHAO_BUILD_KLEIDIAI=1 TORCHAO_ENABLE_ARM_NEON_DOT=1 pip install third-party/ao
         else
-          PYTHON_EXECUTABLE=python \
-              bash .ci/scripts/setup-linux.sh --build-tool cmake
-        fi
 
         pip install -U "huggingface_hub[cli]"