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

Build and test with multiple CUDA versions (#2780)

Browse Source
This commit is contained in:
Cheng
2025-11-17 09:19:02 +09:00
committed by GitHub
parent b7214ff01e
commit 472c43a0c8
6 changed files with 52 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.github/actions/build-cuda-release/action.yml vendored
View File

@@ -2,8 +2,8 @@ name: 'Build CUDA wheel'
description: 'Build CUDA wheel'
inputs:
nvcc-location:
description: 'Location of nvcc compiler'
toolkit:
description: 'The CUDA toolkit'
required: true
runs:
@@ -12,7 +12,7 @@ runs:
- name: Build package
shell: bash
env:
CMAKE_ARGS: -DMLX_BUILD_CUDA=ON -DCMAKE_CUDA_COMPILER=${{ inputs.nvcc-location }}
CMAKE_ARGS: -DMLX_BUILD_CUDA=ON -DCMAKE_CUDA_COMPILER=/usr/local/${{ inputs.toolkit }}/bin/nvcc
run: |
pip install auditwheel build patchelf setuptools
python setup.py clean --all