From 77caa0aa53a8368a455f9f074bf3fb43a46bfd6b Mon Sep 17 00:00:00 2001
From: Jithun Nair <37884920+jithunnair-amd@users.noreply.github.com>
Date: Mon, 6 Oct 2025 14:53:47 -0500
Subject: [PATCH 1/2] Update linux_job_v2.yml

---
 .github/workflows/linux_job_v2.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/linux_job_v2.yml b/.github/workflows/linux_job_v2.yml
index 595af0cffa..62719524a4 100644
--- a/.github/workflows/linux_job_v2.yml
+++ b/.github/workflows/linux_job_v2.yml
@@ -165,7 +165,7 @@ jobs:
         if: ${{ inputs.gpu-arch-type != 'rocm' }}
 
       - name: Setup ROCM
-        uses: pytorch/pytorch/.github/actions/setup-rocm@main
+        uses: pytorch/pytorch/.github/actions/setup-rocm@move_aws_steps_inside_setup_rocm
         if: ${{ inputs.gpu-arch-type == 'rocm' }}
 
       - name: Setup SSH

From 6e9b6786b89d3221496639dd7b190bb0a56877b9 Mon Sep 17 00:00:00 2001
From: Jithun Nair <37884920+jithunnair-amd@users.noreply.github.com>
Date: Tue, 7 Oct 2025 14:38:16 -0500
Subject: [PATCH 2/2] Add id-token:write permissions to linux_job_v2.yml

---
 .github/workflows/linux_job_v2.yml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.github/workflows/linux_job_v2.yml b/.github/workflows/linux_job_v2.yml
index 62719524a4..0b90193a9d 100644
--- a/.github/workflows/linux_job_v2.yml
+++ b/.github/workflows/linux_job_v2.yml
@@ -109,6 +109,11 @@ on:
         default: false
         type: boolean
 
+# Needed for ROCm-conditional steps that need to use OIDC
+permissions:
+  id-token: write
+  contents: read
+
 jobs:
   job:
     strategy: