bump_torch_mlir.yml

# Copyright 2024 The IREE Authors
#
# Licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

# Automatically bumps the torch-mlir submodule to match iree-org/torch-mlir:main.
# Manually triggered for now, but can be scheduled in the future.
# This typically takes ~90s to run.
# Will succeed without creating a PR if there is no newer version for torch-mlir.
name: Bump submodule - torch-mlir

on:
  # TODO: Uncomment the following lines to enable the schedule when this action is stable & useful enough
  # schedule:
  #   - cron: '0 0 * * 0,4'  # Run at 00:00 on Sunday and Thursday
  workflow_dispatch:
    inputs:
      bump_target:
        description: 'Commit hash or branch to bump the submodule to'
        required: false
        default: 'upstream/main'
  pull_request:
    branches:
      - main
    paths:
      - .github/workflows/bump_torch_mlir.yml

jobs:
  bump_submodule:
    runs-on: ubuntu-20.04
    steps:
      - name: Checkout repository
        uses: actions/checkout@ac593985615ec2ede58e132d2e21d2b1cbd6127c  # v3.3.0
        with:
          submodules: false
          ref: main

      - name: Setting git config
        run: |
          git config --local user.email "iree-github-actions-bot@google.com"
          git config --local user.name "Submodule Bump Action"

      - name: Update torch-mlir submodule
        run: |
          git submodule update --init --depth 1 third_party/torch-mlir

      - name: Bump torch-mlir submodule
        id: bump_submodule
        run: |
          cd third_party/torch-mlir
          git checkout main
          if [ "workflow_dispatch" == "workflow_dispatch" ]; then
            bump_target=${{ github.event.inputs.bump_target }}
          else
            bump_target=upstream/main
          fi
          git pull ${bump_target//\// }
          TORCH_MLIR_COMMIT=$(git rev-parse HEAD)
          cd ../..
          git add third_party/torch-mlir
          git commit -m "Bump torch-mlir submodule to $TORCH_MLIR_COMMIT"
          echo "TORCH_MLIR_COMMIT=$TORCH_MLIR_COMMIT" >> $GITHUB_OUTPUT

      - name: Create Pull Request
        uses: peter-evans/create-pull-request@v6.0.5
        with:
          token: ${{ secrets.GITHUB_TOKEN }}
          base: main
          branch: integrates/torch-mlir
          delete-branch: true # automatically supercedes the previous bump
          title: 'Bump torch-mlir submodule to ${{ steps.bump_submodule.outputs.TORCH_MLIR_COMMIT }}'
          body: |
            This PR updates the torch-mlir submodule to https://github.com/llvm/torch-mlir/commit/${{ steps.bump_submodule.outputs.TORCH_MLIR_COMMIT }}.

            Auto-generated by GitHub Actions using `.github/workflows/bump_torch_mlir.yml`.
          commit-message: 'Bump torch-mlir submodule to ${{ steps.bump_submodule.outputs.TORCH_MLIR_COMMIT }}'