diff --git a/.github/actions/setup-linux/action.yml b/.github/actions/setup-linux/action.yml index e7ac7ccde..721a097a3 100644 --- a/.github/actions/setup-linux/action.yml +++ b/.github/actions/setup-linux/action.yml @@ -58,7 +58,9 @@ runs: "cuda-13.0": "libcudnn9-dev-cuda-13 cuda-toolkit-13-0" } run: | - export ARCH=${{ runner.arch == 'arm64' && 'sbsa' || 'x86_64' }} + # The CUDA binaries are hosted in the "sbsa" repo, the "arm64" repo is + # Jetson specific. SBSA means Arm Server Base System Architecture. + ARCH=${{ runner.arch == 'arm64' && 'sbsa' || 'x86_64' }} wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/$ARCH/cuda-keyring_1.1-1_all.deb sudo dpkg -i cuda-keyring_1.1-1_all.deb sudo apt-get update diff --git a/.github/workflows/pull_request.yml b/.github/workflows/build_and_test.yml similarity index 94% rename from .github/workflows/pull_request.yml rename to .github/workflows/build_and_test.yml index 3aea42610..f5d566213 100644 --- a/.github/workflows/pull_request.yml +++ b/.github/workflows/build_and_test.yml @@ -17,6 +17,7 @@ concurrency: jobs: prepare: + name: Lint and Prepare runs-on: ubuntu-22.04 outputs: matrix: ${{ steps.generator.outputs.matrix }} @@ -55,6 +56,7 @@ jobs: core.setOutput('runner', JSON.stringify(runner)) linux_build_and_test: + name: Linux (${{ matrix.toolkit }}, ${{ matrix.arch }}) needs: prepare strategy: fail-fast: false @@ -74,6 +76,7 @@ jobs: cpu-only: ${{ matrix.toolkit == 'cpu' }} mac_build_and_test: + name: macOS (${{ matrix.macos-target }}) if: github.repository == 'ml-explore/mlx' strategy: matrix: @@ -88,6 +91,7 @@ jobs: - uses: ./.github/actions/build-macos build_documentation: + name: Build Documentation if: github.repository == 'ml-explore/mlx' runs-on: ubuntu-22.04 needs: prepare @@ -96,7 +100,7 @@ jobs: - uses: ./.github/actions/build-docs linux_fedora_build_cpp: - name: Linux Fedora CPP Build (${{ matrix.arch }}) + name: Linux Fedora (${{ matrix.arch }}) needs: prepare strategy: fail-fast: false