Update test-rl-gpu.yml

pytorch · Jan 21, 2025 · dd12713 · dd12713
1 parent 6cc5645
commit dd12713
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/test-rl-gpu.yml b/.github/workflows/test-rl-gpu.yml
@@ -26,6 +26,9 @@ jobs:
         cuda_arch_version: ["12.1"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
+    permissions:
+      id-token: write
+      contents: read
     with:
       runner: linux.g5.4xlarge.nvidia.gpu
       repository: pytorch/tensordict