Create _setup.yml to wrap common setup steps.

pytorch · Jun 13, 2024 · d180514 · d180514
1 parent 1cad403
commit d180514
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 21 deletions.
diff --git a/.github/workflows/_build_torch_xla.yml b/.github/workflows/_build_torch_xla.yml
@@ -27,31 +27,15 @@ jobs:
       image: ${{ inputs.dev-image }}
     env:
       GCLOUD_SERVICE_KEY: ${{ secrets.gcloud-service-key }}
-      GOOGLE_APPLICATION_CREDENTIALS: /tmp/default_credentials.json
-      BAZEL_JOBS: 16
       BAZEL_REMOTE_CACHE: ${{ github.event_name == 'push' || github.event.pull_request.head.repo.full_name == github.repository }}
+      BAZEL_JOBS: 16
       BUILD_CPP_TESTS: 1
     steps:
-      # See https://github.com/actions/checkout/issues/1014#issuecomment-1906802802
-      - name: Clean up workspace
-        run: |
-          ls -la
-          rm -rvf ${GITHUB_WORKSPACE}/*
-      - name: Setup gcloud
-        shell: bash
-        run: |
-          echo "${GCLOUD_SERVICE_KEY}" > $GOOGLE_APPLICATION_CREDENTIALS
-      - name: Checkout PyTorch Repo
-        uses: actions/checkout@v4
-        with:
-          repository: pytorch/pytorch
-          path: pytorch
-          ref: ${{ inputs.torch-commit }}
-          submodules: recursive
-      - name: Checkout PyTorch/XLA Repo
-        uses: actions/checkout@v4
+      - name: Setup
+        uses: ./.github/workflows/_setup.yml
         with:
-          path: pytorch/xla
+          torch-commit: ${{ inputs.torch-commit }}
+          setup-remote-cache: ${{ env.BAZEL_REMOTE_CACHE }}
       - name: Build
         shell: bash
         run: |

diff --git a/.github/workflows/_setup.yml b/.github/workflows/_setup.yml
@@ -0,0 +1,34 @@
+name: Set up PyTorch/XLA
+inputs:
+  torch-commit:
+    required: true
+    type: string
+  remote-cache:
+    required: true
+    type: boolean
+    description: Whether to set up Google Cloud credentials for Bazel cache.
+runs:
+  using: "composite"
+  steps:
+    # See https://github.com/actions/checkout/issues/1014#issuecomment-1906802802
+    - name: Clean up workspace
+      run: |
+        ls -la
+        rm -rvf ${GITHUB_WORKSPACE}/*
+    - name: Setup gcloud
+      shell: bash
+      run: |
+        echo "${GCLOUD_SERVICE_KEY}" > /tmp/default_credentials.json
+        echo "GOOGLE_APPLICATION_CREDENTIALS=/tmp/default_credentials.json" >> $GITHUB_ENV
+      if: ${{ inputs.remote-cache }}
+    - name: Checkout PyTorch Repo
+      uses: actions/checkout@v4
+      with:
+        repository: pytorch/pytorch
+        path: pytorch
+        ref: ${{ inputs.torch-commit }}
+        submodules: recursive
+    - name: Checkout PyTorch/XLA Repo
+      uses: actions/checkout@v4
+      with:
+        path: pytorch/xla