From 153d8e524c502c99fca0cf4e6793595085baeb05 Mon Sep 17 00:00:00 2001
From: Cheng <git@zcbenz.com>
Date: Sat, 22 Nov 2025 14:30:25 +0900
Subject: [PATCH] Set MLX_CUDA_ARCHITECTURES for arm64 CUDA build

---
 .github/actions/build-linux/action.yml | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/.github/actions/build-linux/action.yml b/.github/actions/build-linux/action.yml
index 85e20866d..b4538bc2a 100644
--- a/.github/actions/build-linux/action.yml
+++ b/.github/actions/build-linux/action.yml
@@ -13,15 +13,13 @@ runs:
       shell: sh
       env:
         DEBUG: 1
-        CMAKE_ARGS_CPU: >-
+        CMAKE_ARGS: >-
           -DCMAKE_COMPILE_WARNING_AS_ERROR=ON
-          -DMLX_BUILD_CUDA=OFF
-        CMAKE_ARGS_CUDA: >-
-          -DCMAKE_COMPILE_WARNING_AS_ERROR=ON
-          -DMLX_BUILD_CUDA=ON
+          -DMLX_BUILD_CUDA=${{ startsWith(inputs.toolkit, 'cuda') && 'ON' || 'OFF' }}
       run: |
-        export CMAKE_ARGS="$CMAKE_ARGS_${{ startsWith(inputs.toolkit, 'cuda') && 'CUDA' || 'CPU' }}"
         if ${{ startsWith(inputs.toolkit, 'cuda') && runner.arch == 'arm64' }} ; then
+          # There is no GPU in arm64 runner.
+          CMAKE_ARGS="$CMAKE_ARGS -DMLX_CUDA_ARCHITECTURES=90a"
           # Can not build tests when the built executables can not run.
           CMAKE_ARGS="$CMAKE_ARGS -DMLX_BUILD_TESTS=OFF"
         fi