fix: CI — install mlx.metallib from Python mlx package

simba · simba · commit 4086ce939474 · 2026-03-24T09:38:00.000-07:00
The Metal shader library is required at runtime by MLX Swift.
Install via: python3 -m venv + pip install mlx + copy metallib.
Also trigger CI on feature/* branches.
diff --git a/.github/workflows/e2e-test.yml b/.github/workflows/e2e-test.yml
@@ -2,7 +2,7 @@ name: E2E Tests
 
 on:
   push:
-    branches: [main]
+    branches: [main, feature/*]
   pull_request:
     branches: [main]
 
@@ -18,9 +18,6 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - name: Install Metal Toolchain
-        run: xcodebuild -downloadComponent MetalToolchain || true
-
       - name: Cache Swift packages
         uses: actions/cache@v4
         with:
@@ -32,6 +29,12 @@ jobs:
       - name: Build (Release)
         run: swift build -c release
 
+      - name: Install MLX Metal library
+        run: |
+          python3 -m venv /tmp/mlx_venv
+          /tmp/mlx_venv/bin/pip install --quiet mlx
+          cp /tmp/mlx_venv/lib/python*/site-packages/mlx/lib/mlx.metallib .build/release/
+
       - name: Cache MLX model
         uses: actions/cache@v4
         with:
