From 381236a186ee7f918a4102553cfd42b272b8ceb7 Mon Sep 17 00:00:00 2001
From: Xida Ren <cedar.ren@gmail.com>
Date: Wed, 12 Jun 2024 15:48:40 +0000
Subject: [PATCH] try again

---
 .github/workflows/bump_torch_mlir.yml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/bump_torch_mlir.yml b/.github/workflows/bump_torch_mlir.yml
index 85fb1d42e30e1..be52acba0ecd8 100644
--- a/.github/workflows/bump_torch_mlir.yml
+++ b/.github/workflows/bump_torch_mlir.yml
@@ -48,9 +48,10 @@ jobs:
         id: bump_submodule
         run: |
           cd third_party/torch-mlir
-          git fetch origin
+          bump_target=${{ github.event.inputs.bump_target }}
+          git fetch ${bump_target//\// }
           if [ "workflow_dispatch" == "workflow_dispatch" ]; then
-            git checkout upstream/main || { echo "Checkout failed. Available remotes:"; git remote -v; exit 1; }
+            git checkout ${{ github.event.inputs.bump_target }} || { echo "Checkout failed. Available remotes:"; git remote -v; exit 1; }
           else
             git checkout origin/main
           fi