mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
46 lines
1.2 KiB
YAML
46 lines
1.2 KiB
YAML
name: 'Build and Test with CUDA'
|
|
description: 'Build and test MLX with CUDA'
|
|
|
|
inputs:
|
|
nvcc-location:
|
|
description: 'Location of nvcc compiler'
|
|
required: true
|
|
default: '/usr/local/cuda-12.9/bin/nvcc'
|
|
|
|
runs:
|
|
using: "composite"
|
|
steps:
|
|
- name: Install Python package
|
|
shell: bash
|
|
env:
|
|
DEBUG: 1
|
|
CMAKE_ARGS: -DMLX_BUILD_CUDA=ON -DCMAKE_COMPILE_WARNING_AS_ERROR=ON -DCMAKE_CUDA_COMPILER=${{ inputs.nvcc-location }}
|
|
run: pip install -e ".[dev]" -v
|
|
|
|
- name: Run Python tests - CPU
|
|
shell: bash
|
|
env:
|
|
LOW_MEMORY: 1
|
|
DEVICE: cpu
|
|
run: python -m unittest discover python/tests -v
|
|
|
|
- name: Run Python tests - GPU
|
|
shell: bash
|
|
env:
|
|
LOW_MEMORY: 1
|
|
DEVICE: gpu
|
|
run: python -m tests discover python/tests -v
|
|
|
|
- name: Build CPP only
|
|
shell: bash
|
|
run: |
|
|
cmake . -B build \
|
|
-DMLX_BUILD_CUDA=ON \
|
|
-DCMAKE_CUDA_COMPILER=${{ inputs.nvcc-location }} \
|
|
-DCMAKE_BUILD_TYPE=DEBUG
|
|
cmake --build build -j $(nproc)
|
|
|
|
- name: Run CPP tests
|
|
shell: bash
|
|
run: ./build/tests/tests -sfe="*fft_tests.cpp,*linalg_tests.cpp"
|