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

Add CUDA toolkit bin to $PATH

Browse Source
This commit is contained in:
Cheng
2025-11-22 11:39:48 +09:00
parent d2069c0a3c
commit b7e678bea7
4 changed files with 2 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -18,7 +18,6 @@ runs:
-DMLX_BUILD_CUDA=OFF
CMAKE_ARGS_CUDA: >-
-DCMAKE_COMPILE_WARNING_AS_ERROR=ON
-DCMAKE_CUDA_COMPILER=/usr/local/${{ inputs.toolkit }}/bin/nvcc
-DMLX_BUILD_CUDA=ON
run: |
export CMAKE_ARGS="$CMAKE_ARGS_${{ startsWith(inputs.toolkit, 'cuda') && 'CUDA' || 'CPU' }}"