From d0bf444c705b73667b4d9508734cc2499f54bacd Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 Jan 2025 08:43:12 -0800 Subject: [PATCH] Fix linux_job_v2 after https://github.com/pytorch/test-infra/pull/6104 (#2692) Summary: Same as https://github.com/pytorch/executorch/pull/7731 Pull Request resolved: https://github.com/pytorch/torchrec/pull/2692 Reviewed By: PaulZhang12 Differential Revision: D68442367 Pulled By: atalman fbshipit-source-id: 6693cdab50a0748997b955958c65b876dc6f2dec --- .github/workflows/unittest_ci.yml | 3 +++ .github/workflows/unittest_ci_cpu.yml | 3 +++ 2 files changed, 6 insertions(+) diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml index 1f0b18dcd..89aa2768f 100644 --- 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 index 4c92e6027..a0b0ef9eb 100644 --- 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