Auto-retry failed CI builds once (facebookresearch#3718)

Summary:
Pull Request resolved: facebookresearch#3718

As title

Differential Revision: D60697701

Author: gtwang01

.github/workflows/build.yml

@@ -262,3 +262,16 @@ jobs:
       - uses: ./.github/actions/build_conda
         with:
           label: main
+  re-run:
+    name: Activate rerun workflow
+    if: fromJSON(github.run_attempt) < 2
+    runs-on: ubuntu-latest
+    env:
+      GITHUB_TOKEN: ${{ secrets.TOKEN }}
+    steps:
+      - name:
+        run: |
+          gh workflow run retry_build.yml `
+            --raw-field run_id=${{ github.run_id }} `
+            --repo ${{ github.repository }}
+          if(!$?) { exit 1 }

.github/workflows/retry_build.yml

@@ -0,0 +1,17 @@
+name: Retry Build
+on:
+  workflow_dispatch:
+    inputs:
+      run_id:
+        required: true
+jobs:
+  rerun-on-failure:
+    runs-on: ubuntu-latest
+    if: ${{ github.event.workflow_run.conclusion == 'success' }}
+    steps:
+      - name: rerun ${{ inputs.run_id }}
+        env:
+          GITHUB_TOKEN: ${{ secrets.TOKEN }}
+        run: |
+          gh run watch ${{ inputs.run_id }} > /dev/null 2>&1
+          gh run rerun ${{ inputs.run_id }} --failed