From 978e9c589f30fd832c6a0326910344df3cff2080 Mon Sep 17 00:00:00 2001
From: Cheng <git@zcbenz.com>
Date: Mon, 17 Nov 2025 16:05:29 +0900
Subject: [PATCH] Turn off tests for arm64 CUDA build

---
 .github/actions/build-linux/action.yml | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/.github/actions/build-linux/action.yml b/.github/actions/build-linux/action.yml
index 76dc86f6c..e93be45de 100644
--- a/.github/actions/build-linux/action.yml
+++ b/.github/actions/build-linux/action.yml
@@ -20,7 +20,12 @@ runs:
           -DCMAKE_CUDA_COMPILER=/usr/local/${{ inputs.toolkit }}/bin/nvcc
           -DMLX_BUILD_CUDA=ON
       run: |
-        echo CMAKE_ARGS="$CMAKE_ARGS_${{ startsWith(inputs.toolkit, 'cuda') && 'CUDA' || 'CPU' }}" > $GITHUB_ENV
+        CMAKE_ARGS="$CMAKE_ARGS_${{ startsWith(inputs.toolkit, 'cuda') && 'CUDA' || 'CPU' }}"
+        # FIXME: Unable to build tests without libnccl.
+        if ${{ startsWith(inputs.toolkit, 'cuda') && runner.arch == 'arm64' }} ; then
+          CMAKE_ARGS="$CMAKE_ARGS -DMLX_BUILD_TESTS=OFF"
+        fi
+        echo CMAKE_ARGS="$CMAKE_ARGS" > $GITHUB_ENV
 
     - name: Install Python package
       shell: sh