4141 reenabled-issues : ${{ steps.filter.outputs.reenabled-issues }}
4242 steps :
4343 - name : Checkout PyTorch
44- uses : pytorch/pytorch/.github/actions/checkout-pytorch@main
44+ uses : pytorch/pytorch/.github/actions/checkout-pytorch@release/2.1
4545 with :
4646 fetch-depth : 1
4747 submodules : false
@@ -63,30 +63,30 @@ jobs:
6363 runs-on : ${{ matrix.runner }}
6464 steps :
6565 - name : Setup SSH (Click me for login details)
66- uses : pytorch/test-infra/.github/actions/setup-ssh@main
66+ uses : pytorch/test-infra/.github/actions/setup-ssh@release/2.1
6767 with :
6868 github-secret : ${{ secrets.GITHUB_TOKEN }}
6969
7070 # [see note: pytorch repo ref]
7171 - name : Checkout PyTorch
72- uses : pytorch/pytorch/.github/actions/checkout-pytorch@main
72+ uses : pytorch/pytorch/.github/actions/checkout-pytorch@release/2.1
7373
7474 - name : Setup Linux
7575 uses : ./.github/actions/setup-linux
7676
7777 - name : Calculate docker image
7878 id : calculate-docker-image
79- uses : pytorch/test-infra/.github/actions/calculate-docker-image@main
79+ uses : pytorch/test-infra/.github/actions/calculate-docker-image@release/2.1
8080 with :
8181 docker-image-name : ${{ inputs.docker-image-name }}
8282
8383 - name : Pull docker image
84- uses : pytorch/test-infra/.github/actions/pull-docker-image@main
84+ uses : pytorch/test-infra/.github/actions/pull-docker-image@release/2.1
8585 with :
8686 docker-image : ${{ steps.calculate-docker-image.outputs.docker-image }}
8787
8888 - name : Install nvidia driver, nvidia-docker runtime, set GPU_FLAG
89- uses : pytorch/test-infra/.github/actions/setup-nvidia@main
89+ uses : pytorch/test-infra/.github/actions/setup-nvidia@release/2.1
9090 if : ${{ inputs.cuda-version != 'cpu' }}
9191
9292 - name : Output disk space left
@@ -197,5 +197,5 @@ jobs:
197197 file-suffix : bazel-${{ github.job }}_${{ steps.get-job-id.outputs.job-id }}
198198
199199 - name : Teardown Linux
200- uses : pytorch/test-infra/.github/actions/teardown-linux@main
200+ uses : pytorch/test-infra/.github/actions/teardown-linux@release/2.1
201201 if : always()
0 commit comments