ci.yml

name: CI

on:
  push:
    branches:
      - main
  pull_request:

# Ensure only one workflow instance runs at a time. For branches other than the
# default branch, cancel the pending jobs in the group. For the default branch,
# queue them up. This avoids cancelling jobs that are in the middle of deploying
# to production.
concurrency:
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: ${{ github.ref != format('refs/heads/{0}', github.event.repository.default_branch) }}

jobs:
  check:
    name: Check
    runs-on: ubuntu-latest
    env:
      R8_CUDA_PREFIX: ${{ secrets.R8_CUDA_PREFIX }}
      R8_CUDNN_PREFIX: ${{ secrets.R8_CUDNN_PREFIX }}
    steps:
      - uses: actions/checkout@v4
      - uses: astral-sh/setup-uv@v3
      - run: ./script/check

  test:
    name: Test
    runs-on: ubuntu-latest-4-cores
    env:
      R8_CUDA_PREFIX: ${{ secrets.R8_CUDA_PREFIX }}
      R8_CUDNN_PREFIX: ${{ secrets.R8_CUDNN_PREFIX }}
    steps:
      - uses: actions/checkout@v4
      - uses: astral-sh/setup-uv@v3
      - run: ./script/build
      - run: ./script/build-runtime
      - run: ./script/test

  test-mini:
    name: Test Mini
    runs-on: ubuntu-latest-4-cores
    env:
      R8_CUDA_PREFIX: ${{ secrets.R8_CUDA_PREFIX }}
      R8_CUDNN_PREFIX: ${{ secrets.R8_CUDNN_PREFIX }}
    steps:
      - uses: actions/checkout@v4
      - uses: astral-sh/setup-uv@v3
      - run: ./script/build
      - run: ./script/test-mini

  build-release-monobase:
    if: ${{ github.event_name != 'pull_request' }}
    name: Build + release monobase image
    needs:
      - check
      - test
      - test-mini
    permissions:
      contents: read
      id-token: write
    uses: replicate/actions/.github/workflows/buildx.yml@main
    with:
      image: monobase
      fetch-depth: 0
      build-args: BUILDKIT_CONTEXT_KEEP_GIT_DIR=1

  build-release-monobase-runtime:
    if: ${{ github.event_name != 'pull_request' }}
    name: Build + release monobase-runtime image
    needs:
      - check
      - test
      - test-mini
    permissions:
      contents: read
      id-token: write
    uses: replicate/actions/.github/workflows/buildx.yml@main
    with:
      image: monobase-runtime
      file: runtime.Dockerfile
      fetch-depth: 0