commit dfc5bb009996814f0439c1978d53a5503f7cf9d9
author Thanh Ha <thanh.ha@linuxfoundation.org> Thu Aug 08 16:34:43 2024 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Thu Aug 08 16:34:46 2024 +0000
tree b479444857e036bda9bb292dab150b5f13a88bd1
parent 4a4dc9d6d9f09752b5ed6e492ade10fd82aa050b

Login to Meta's ECR when using non-meta runner (#132870)

The project depends on fetching container images from Meta's ECR repo so when run on non-meta runners we need to ensure that we also login to Meta's ECR too.

Closes pytorch/ci-infra#252.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/132870
Approved by: https://github.com/ZainRizvi

.github/actions/setup-linux/action.yml

diff --git a/.github/actions/setup-linux/action.yml b/.github/actions/setup-linux/action.yml
index 193dc7d..d4ea665 100644
--- a/.github/actions/setup-linux/action.yml
+++ b/.github/actions/setup-linux/action.yml
@@ -59,6 +59,13 @@
           aws ecr get-login-password --region "$AWS_DEFAULT_REGION" | docker login --username AWS \
               --password-stdin "$AWS_ACCOUNT_ID.dkr.ecr.$AWS_DEFAULT_REGION.amazonaws.com"
 
+          # For LF Runners we need to make sure we also login to Meta's ECR docker registry too.
+          META_AWS_ACCOUNT_ID=308535385114
+          if [ "$AWS_ACCOUNT_ID" != "$META_AWS_ACCOUNT_ID" ] ; then
+              aws ecr get-login-password --region "$AWS_DEFAULT_REGION" | docker login --username AWS \
+                  --password-stdin "$META_AWS_ACCOUNT_ID.dkr.ecr.$AWS_DEFAULT_REGION.amazonaws.com"
+          fi
+
     - name: Preserve github env variables for use in docker
       shell: bash
       run: |