mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-07 19:59:01 +08:00
109 lines
2.8 KiB
YAML
109 lines
2.8 KiB
YAML
name: Build and Test
|
|
|
|
on:
|
|
pull_request:
|
|
push:
|
|
branches:
|
|
- main
|
|
# For testing CI without starting a pull request:
|
|
- test/*
|
|
|
|
permissions:
|
|
contents: read
|
|
|
|
concurrency:
|
|
group: ${{ github.workflow }}-${{ github.ref }}
|
|
cancel-in-progress: ${{ github.ref != 'refs/heads/main' }}
|
|
|
|
jobs:
|
|
check_lint:
|
|
name: Check Lint
|
|
runs-on: ubuntu-22.04
|
|
steps:
|
|
- uses: actions/checkout@v6
|
|
- uses: pre-commit/action@v3.0.1
|
|
|
|
linux_build_and_test:
|
|
name: Linux (cpu, ${{ matrix.arch }})
|
|
needs: check_lint
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
arch: ['x86_64', 'aarch64']
|
|
runs-on: ${{ matrix.arch == 'x86_64' && 'ubuntu-22.04' || 'ubuntu-22.04-arm' }}
|
|
steps:
|
|
- uses: actions/checkout@v6
|
|
- uses: ./.github/actions/setup-linux
|
|
- uses: ./.github/actions/build-linux
|
|
- uses: ./.github/actions/test-linux
|
|
|
|
cuda_build_and_test:
|
|
name: Linux (${{ matrix.toolkit }}, ${{ matrix.arch }})
|
|
if: github.repository == 'ml-explore/mlx'
|
|
needs: check_lint
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
arch: ['x86_64', 'aarch64']
|
|
toolkit: ['cuda-12.6', 'cuda-12.9']
|
|
runs-on: ${{ matrix.arch == 'x86_64' && 'gpu-t4-4-core' || 'ubuntu-22.04-arm' }}
|
|
steps:
|
|
- uses: actions/checkout@v6
|
|
- uses: ./.github/actions/setup-linux
|
|
with:
|
|
toolkit: ${{ matrix.toolkit }}
|
|
- uses: ./.github/actions/build-linux
|
|
with:
|
|
toolkit: ${{ matrix.toolkit }}
|
|
- uses: ./.github/actions/test-linux
|
|
if: matrix.arch == 'x86_64'
|
|
with:
|
|
has-gpu: true
|
|
|
|
mac_build_and_test:
|
|
name: macOS (${{ matrix.macos-target }})
|
|
if: github.repository == 'ml-explore/mlx'
|
|
strategy:
|
|
matrix:
|
|
macos-target: ["14.0", "15.0"]
|
|
runs-on: [self-hosted, macos]
|
|
env:
|
|
MACOSX_DEPLOYMENT_TARGET: ${{ matrix.macos-target }}
|
|
needs: check_lint
|
|
steps:
|
|
- uses: actions/checkout@v6
|
|
- uses: ./.github/actions/setup-macos
|
|
- uses: ./.github/actions/build-macos
|
|
|
|
build_documentation:
|
|
name: Build Documentation
|
|
if: github.repository == 'ml-explore/mlx'
|
|
runs-on: ubuntu-22.04
|
|
needs: check_lint
|
|
steps:
|
|
- uses: actions/checkout@v6
|
|
- uses: ./.github/actions/build-docs
|
|
|
|
linux_fedora_build_cpp:
|
|
name: Linux Fedora (${{ matrix.arch }})
|
|
needs: check_lint
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
include:
|
|
- host: ubuntu-22.04
|
|
arch: x86_64
|
|
- host: ubuntu-22.04-arm
|
|
arch: aarch64
|
|
|
|
runs-on: ${{ matrix.host }}
|
|
container:
|
|
image: fedora:42
|
|
steps:
|
|
- name: Checkout code
|
|
uses: actions/checkout@v6
|
|
|
|
- name: CPP Build Test - No Release
|
|
run: |
|
|
bash ./.github/scripts/setup+build-cpp-linux-fedora-container.sh
|