release.yml

name: Release

on:
  push:
    tags:
      - "v*.*.*"

jobs:
  # Stage 1: Build docc-compiler (no dependencies)
  wheels-compiler:
    name: Compiler (${{ matrix.os }}, ${{ matrix.python }})
    runs-on: ${{ matrix.os }}

    strategy:
      fail-fast: false
      matrix:
        os: [build-amd64-big, build-arm64-big, macos-14]
        python: ["cp311", "cp312", "cp313", "cp314"]
        include:
          - os: build-amd64-big
            cibw_archs: x86_64
          - os: build-arm64-big
            cibw_archs: aarch64
          - os: macos-14
            cibw_archs: arm64

    steps:
      - uses: actions/checkout@v4
        with:
          token: ${{ secrets.BENCHMARK_TESTS_TOKEN }}
          submodules: recursive

      - uses: pypa/cibuildwheel@v3.3.1
        with:
          package-dir: python/
          output-dir: wheelhouse
        env:
          CIBW_ARCHS: ${{ matrix.cibw_archs }}
          CIBW_BUILD: "${{ matrix.python }}-*"

      - uses: actions/upload-artifact@v4
        with:
          name: wheels-docc-compiler-${{ matrix.os }}-${{ matrix.python }}
          path: wheelhouse/*.whl

  # Stage 2: Build docc-ai (depends on docc-compiler)
  wheels-ai:
    name: AI (${{ matrix.os }}, ${{ matrix.python }})
    needs: [wheels-compiler]
    runs-on: ${{ matrix.os }}

    strategy:
      fail-fast: false
      matrix:
        os: [build-amd64-big, build-arm64-big, macos-14]
        python: ["cp311", "cp312"]
        include:
          - os: build-amd64-big
            cibw_archs: x86_64
          - os: build-arm64-big
            cibw_archs: aarch64
          - os: macos-14
            cibw_archs: arm64

    steps:
      - uses: actions/checkout@v4
        with:
          token: ${{ secrets.BENCHMARK_TESTS_TOKEN }}
          submodules: recursive

      # Pin docc-compiler version to match release
      # - name: Pin docc-compiler version
      #   run: |
      #     VERSION=$(cat VERSION)
      #     sed -i.bak "s/\"docc-compiler\"/\"docc-compiler==$VERSION\"/" mlir/pyproject.toml && rm mlir/pyproject.toml.bak

      # Download compiler wheels into the package directory so they're available in container
      # - uses: actions/download-artifact@v4
      #   with:
      #     pattern: wheels-docc-compiler-${{ matrix.os }}-*
      #     path: mlir/compiler-wheels
      #     merge-multiple: true

      - uses: pypa/cibuildwheel@v3.3.1
        with:
          package-dir: mlir/
          output-dir: wheelhouse
        env:
          CIBW_ARCHS: ${{ matrix.cibw_archs }}
          CIBW_BUILD: "${{ matrix.python }}-*"
          # Install docc-compiler before building docc-ai
          # CIBW_BEFORE_BUILD: "pip install --no-index --find-links {project}/compiler-wheels docc-compiler"
          # Make compiler wheels available for dependency resolution during test
          # CIBW_ENVIRONMENT: "PIP_FIND_LINKS={project}/compiler-wheels"

      - uses: actions/upload-artifact@v4
        with:
          name: wheels-docc-ai-${{ matrix.os }}-${{ matrix.python }}
          path: wheelhouse/*.whl

  wheels-publish:
    needs: [wheels-compiler, wheels-ai]
    runs-on: build-amd64-big
    permissions:
      id-token: write

    steps:
      - uses: actions/download-artifact@v4
        with:
          pattern: wheels-*
          path: dist
          merge-multiple: true

      - uses: pypa/gh-action-pypi-publish@v1.10.0

  packages-llvm:
    strategy:
      matrix:
        include:
          - platform: ubuntu-24.04
            package-format: deb
            cpack-generator: DEB
            upload-dist-id: ubuntu
            upload-dist-version: 24.04
            runner: build-amd64-big
            architecture: x64
            image: daisytuner/docc-build-env-llvm21-ubuntu-24.04:latest-amd64
          - platform: ubuntu-26.04
            package-format: deb
            cpack-generator: DEB
            upload-dist-id: ubuntu
            upload-dist-version: 26.04
            runner: build-amd64-big
            architecture: x64
            image: daisytuner/docc-build-env-llvm21-ubuntu-26.04:latest-amd64
          - platform: ubuntu-24.04
            package-format: deb
            cpack-generator: DEB
            upload-dist-id: ubuntu
            upload-dist-version: 24.04
            runner: build-arm64-big
            architecture: arm64
            image: daisytuner/docc-build-env-llvm21-ubuntu-24.04:latest-arm64
          - platform: rhel-10
            package-format: rpm
            cpack-generator: RPM
            upload-dist-id: rhel
            upload-dist-version: 10
            upload-dist-platform-id: platform:el10
            runner: build-amd64-big
            architecture: x64
            image: daisytuner/docc-build-env-llvm21-rhel-10:latest-amd64
          - platform: debian-13
            package-format: deb
            cpack-generator: DEB
            upload-dist-id: debian
            upload-dist-version: 13
            runner: build-amd64-big
            architecture: x64
            image: daisytuner/docc-build-env-llvm21-debian-13:latest-amd64

    runs-on: ${{ matrix.runner }}
    container:
      image: ${{ matrix.image }}

    steps:
      - uses: actions/checkout@v4
        with:
          token: ${{ secrets.BENCHMARK_TESTS_TOKEN }}
          submodules: recursive

      - name: Define Version
        id: define_version
        run: echo "VERSION=${GITHUB_REF#refs/tags/v}" >> $GITHUB_OUTPUT

      - name: Build package
        run: |
          mkdir -p build
          cd build
          cmake -G Ninja \
            -DCMAKE_C_COMPILER=clang-21 \
            -DCMAKE_CXX_COMPILER=clang++-21 \
            -DCMAKE_BUILD_TYPE=Release \
            -DINSTALL_GTEST=OFF \
            -DBUILD_TESTS:BOOL=OFF \
            -DLLVM_BUILD_FRONTEND=ON \
            -DLLVM_BUILD_TESTS=OFF \
            -DBUILD_BENCHMARKS:BOOL=OFF \
            -DBUILD_BENCHMARKS_GOOGLE:BOOL=OFF \
            -DRELEASE_PACKAGE=ON \
            -DPACKAGE_WITH_TOOL_DEPS=ON \
            ..
          ninja -j$(nproc)
          cpack -G ${{ matrix.cpack-generator }}

      - name: Upload docc package as Artifact
        uses: actions/upload-artifact@v4
        with:
          name: docc-${{ matrix.platform }}-${{ matrix.architecture }}
          path: "build/*.${{ matrix.package-format }}"

      - name: Upload docc package to Firebase
        uses: daisytuner/upload-distribution-action@main
        with:
          file: "build/*.${{ matrix.package-format }}"
          version: ${{ steps.define_version.outputs.VERSION }}
          architecture: ${{ matrix.architecture }}
          dist-id: ${{ matrix.upload-dist-id }}
          dist-version: ${{ matrix.upload-dist-version }}
          dist-platform-id: ${{ matrix.upload-dist-platform-id }}
          token: ${{ secrets.DOCC_RELEASE_TOKEN }}
          url: /v1/system/docc-distributions/upload