zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-12 07:18:52 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Disable pythong stubs for arm64 CUDA build

Browse Source
This commit is contained in:
Cheng
2025-11-17 16:24:22 +09:00
parent 978e9c589f
commit 19205e68e2
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -21,7 +21,7 @@ runs:
-DMLX_BUILD_CUDA=ON -DMLX_BUILD_CUDA=ON
run: | run: |
CMAKE_ARGS="$CMAKE_ARGS_${{ startsWith(inputs.toolkit, 'cuda') && 'CUDA' || 'CPU' }}" CMAKE_ARGS="$CMAKE_ARGS_${{ startsWith(inputs.toolkit, 'cuda') && 'CUDA' || 'CPU' }}"
# FIXME: Unable to build tests without libnccl. # FIXME: Unable to build tests without libnccl on arm64.
if ${{ startsWith(inputs.toolkit, 'cuda') && runner.arch == 'arm64' }} ; then if ${{ startsWith(inputs.toolkit, 'cuda') && runner.arch == 'arm64' }} ; then
CMAKE_ARGS="$CMAKE_ARGS -DMLX_BUILD_TESTS=OFF" CMAKE_ARGS="$CMAKE_ARGS -DMLX_BUILD_TESTS=OFF"
fi fi
@@ -34,6 +34,8 @@ runs:
run: pip install --no-build-isolation -e ".[dev]" -v run: pip install --no-build-isolation -e ".[dev]" -v
- name: Generate package stubs - name: Generate package stubs
# FIXME: Unable to generate stubs without libnccl on arm64.
if: ${{ !(startsWith(inputs.toolkit, 'cuda') && runner.arch == 'arm64') }}
shell: sh shell: sh
run: | run: |
pip install typing_extensions pip install typing_extensions