diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 5372bf9..387edfd 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -13,20 +13,24 @@ jobs: matrix: python-version: - "3.11" - runner: ["linux.12xlarge"] + runner: ["mt-l-x86iavx512-48-384"] gpu-arch-type: ["cpu"] include: - python-version: 3.13 - runner: linux.g5.4xlarge.nvidia.gpu + runner: mt-l-x86aavx2-11-41-a10g gpu-arch-type: cuda - gpu-arch-version: "12.9" + gpu-arch-version: "13.2" fail-fast: false - uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main permissions: id-token: write contents: read with: repository: pytorch/extension-cpp + # linux_job_v3 defaults this to refs/heads/main, which this repo does not + # have -- its default branch is master -- so the checkout dies with + # "fatal: couldn't find remote ref refs/heads/main". + additional-fetch-refs: "" runner: ${{ matrix.runner }} gpu-arch-type: ${{ matrix.gpu-arch-type }} gpu-arch-version: ${{ matrix.gpu-arch-version }}