Migrate CircleCI to GitHub Actions (#2716)

Co-authored-by: Joseph Heck <j_heck@apple.com>
This commit is contained in:
Mike Drob
2025-10-30 18:26:55 +01:00
committed by GitHub
parent ec72b44417
commit 74c1ed25bb
14 changed files with 830 additions and 15 deletions

View File

@@ -0,0 +1,24 @@
name: 'Build CUDA wheel'
description: 'Build CUDA wheel'
inputs:
nvcc-location:
description: 'Location of nvcc compiler'
required: true
runs:
using: "composite"
steps:
- name: Build package
shell: bash
env:
MLX_BUILD_STAGE: 2
CMAKE_ARGS: -DMLX_BUILD_CUDA=ON -DCMAKE_CUDA_COMPILER=${{ inputs.nvcc-location }}
run: |
pip install auditwheel build patchelf setuptools
python setup.py clean --all
python -m build -w
if [ -f "python/scripts/repair_cuda.sh" ]; then
bash python/scripts/repair_cuda.sh
fi