Fix linux_job_v2 after pytorch/test-infra#6104 (#2692)

Summary: Same as pytorch/executorch#7731 Pull Request resolved: #2692 Reviewed By: PaulZhang12 Differential Revision: D68442367 Pulled By: atalman fbshipit-source-id: 6693cdab50a0748997b955958c65b876dc6f2dec
pytorch · Jan 21, 2025 · d0bf444 · d0bf444
1 parent 9dfdfb8
commit d0bf444
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml
@@ -64,6 +64,9 @@ jobs:
             python-tag: "py312"
             cuda-tag: "cu124"
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
+    permissions:
+      id-token: write
+      contents: read
     with:
       runner: ${{ matrix.os }}
       timeout: 30

diff --git a/.github/workflows/unittest_ci_cpu.yml b/.github/workflows/unittest_ci_cpu.yml
@@ -36,6 +36,9 @@ jobs:
               python-version: '3.12'
               python-tag: "py312"
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
+    permissions:
+      id-token: write
+      contents: read
     with:
       runner: ${{ matrix.os }}
       timeout: 15