zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Turn off tests for arm64 CUDA build

Browse Source
This commit is contained in:
Cheng
2025-11-17 16:05:29 +09:00
parent b79360d386
commit 978e9c589f
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
.github/actions/build-linux/action.yml vendored
View File

@@ -20,7 +20,12 @@ runs:
-DCMAKE_CUDA_COMPILER=/usr/local/${{ inputs.toolkit }}/bin/nvcc -DCMAKE_CUDA_COMPILER=/usr/local/${{ inputs.toolkit }}/bin/nvcc
-DMLX_BUILD_CUDA=ON -DMLX_BUILD_CUDA=ON
run: | 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 - name: Install Python package
shell: sh shell: sh