mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
c9f4dc851f
Some checks failed
Build and Test / Check Lint (push) Has been cancelled
Build and Test / Linux (cpu, aarch64) (push) Has been cancelled
Build and Test / Linux (cpu, x86_64) (push) Has been cancelled
Build and Test / Linux (cuda-12.6, aarch64) (push) Has been cancelled
Build and Test / Linux (cuda-12.9, aarch64) (push) Has been cancelled
Build and Test / Linux (cuda-12.6, x86_64) (push) Has been cancelled
Build and Test / Linux (cuda-12.9, x86_64) (push) Has been cancelled
Build and Test / macOS (14.0) (push) Has been cancelled
Build and Test / macOS (15.0) (push) Has been cancelled
Build and Test / Build Documentation (push) Has been cancelled
Build and Test / Linux Fedora (aarch64) (push) Has been cancelled
Build and Test / Linux Fedora (x86_64) (push) Has been cancelled
16 lines
383 B
YAML
16 lines
383 B
YAML
name: 'Build CUDA wheel'
|
|
description: 'Build CUDA wheel'
|
|
|
|
runs:
|
|
using: "composite"
|
|
steps:
|
|
- name: Build package
|
|
shell: bash
|
|
env:
|
|
CMAKE_ARGS: -DMLX_BUILD_CUDA=ON
|
|
run: |
|
|
pip install auditwheel build patchelf setuptools
|
|
python setup.py clean --all
|
|
MLX_BUILD_STAGE=2 python -m build -w
|
|
bash python/scripts/repair_cuda.sh
|