From b5d5193968bec87a5ca75739ceccf0e57fa9ae1a Mon Sep 17 00:00:00 2001 From: Jane Xu Date: Wed, 22 Jan 2025 08:48:11 -0800 Subject: [PATCH 1/2] Earlier update was not sufficient --- .github/workflows/tests.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index a024caa..3c49b56 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -22,6 +22,9 @@ jobs: gpu-arch-version: "11.8" fail-fast: false uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + permissions: + id-token: write + contents: read with: repository: pytorch/extension-cpp runner: ${{ matrix.runner }} From cb143fc75d06e15b5f899286f3d7612cf7a2e940 Mon Sep 17 00:00:00 2001 From: Jane Xu Date: Wed, 22 Jan 2025 08:58:00 -0800 Subject: [PATCH 2/2] Use newest pyve and cuda ver we support --- .github/workflows/tests.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 3c49b56..1565937 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -16,10 +16,10 @@ jobs: runner: ["linux.12xlarge"] gpu-arch-type: ["cpu"] include: - - python-version: 3.8 + - python-version: 3.13 runner: linux.g5.4xlarge.nvidia.gpu gpu-arch-type: cuda - gpu-arch-version: "11.8" + gpu-arch-version: "12.4" fail-fast: false uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main permissions: