build-triton.yml

name: Build Triton wheel

jobs:
  build:
    runs-on: ubuntu-latest

    steps:
      - name: Checkout
        uses: actions/checkout@v4
        with:
          ref: ${{ github.event.inputs.ref }}
          repository: ${{ github.event.inputs.repository }}

      - name: Apply patches
        run: |
          set -e

          # List of patches
          patches=(
            "https://raw.githubusercontent.com/${{ github.repository }}/${{ github.sha }}/patches/triton/0000-add-support-for-conversion-fp16-to-fp32.patch"
            "https://raw.githubusercontent.com/${{ github.repository }}/${{ github.sha }}/patches/triton/0000-fix-max-ptx-version.patch"
          )

          # Apply patches
          for patch in $patches; do
            curl "$patch" | patch -p1
          done

      - name: Build wheels
        uses: pypa/cibuildwheel@v2.18.0
        env:
          CIBW_BUILD: cp311-manylinux_x86_64
          CIBW_ENVIRONMENT: "TRITON_BUILD_WITH_CLANG_LLD=1"

      - name: Create release
        uses: softprops/action-gh-release@v2
        with:
          files: wheelhouse/*.whl
          tag_name: ${{ github.event.inputs.tag_name }}

on:
  workflow_dispatch:
    inputs:
      repository:
        default: triton-lang/triton
        description: Source repository
        required: true
        type: string

      ref:
        default: release/2.3.x
        description: Source ref
        required: true
        type: string

      tag_name:
        description: Target tag
        required: true
        type: string

permissions:
  contents: write