diff --git a/.github/workflows/linux_cpu_kineto.yml b/.github/workflows/linux_cpu_kineto.yml index a60abc179..820c490c3 100644 --- a/.github/workflows/linux_cpu_kineto.yml +++ b/.github/workflows/linux_cpu_kineto.yml @@ -15,10 +15,10 @@ permissions: jobs: pr-test: - uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main with: job-name: linux-cpu-kineto-build-and-test - runner: linux.12xlarge + runner: mt-l-x86iavx512-48-384 timeout: 180 # Checkout the Kineto repo at the PR ref with submodules repository: ${{ github.repository }} diff --git a/.github/workflows/linux_cpu_pytorch.yml b/.github/workflows/linux_cpu_pytorch.yml index 29fde6eb6..b24ecda4f 100644 --- a/.github/workflows/linux_cpu_pytorch.yml +++ b/.github/workflows/linux_cpu_pytorch.yml @@ -15,10 +15,10 @@ permissions: jobs: pr-test: - uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main with: job-name: linux-cpu-pytorch-build-and-test - runner: linux.12xlarge + runner: mt-l-x86iavx512-48-384 timeout: 180 # Checkout the Kineto repo at the PR ref with submodules repository: ${{ github.repository }} diff --git a/.github/workflows/linux_cuda_kineto.yml b/.github/workflows/linux_cuda_kineto.yml index e864d416a..d5530495c 100644 --- a/.github/workflows/linux_cuda_kineto.yml +++ b/.github/workflows/linux_cuda_kineto.yml @@ -25,11 +25,11 @@ jobs: # that satisfies the CUDA 13.2 container requirement. - cuda_version: "13.0" pm_sampling: "ON" - uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main with: job-name: linux-cuda-${{ matrix.cuda_version }}-kineto-build-and-test # AWS A10G GPU instance - runner: linux.g5.4xlarge.nvidia.gpu + runner: mt-l-x86aavx2-11-41-a10g timeout: 180 # Checkout the Kineto repo at the PR ref with submodules repository: ${{ github.repository }} diff --git a/.github/workflows/linux_cuda_pytorch.yml b/.github/workflows/linux_cuda_pytorch.yml index 4187a0248..3cd184ae9 100644 --- a/.github/workflows/linux_cuda_pytorch.yml +++ b/.github/workflows/linux_cuda_pytorch.yml @@ -15,11 +15,11 @@ permissions: jobs: pr-test: - uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main with: job-name: linux-cuda-pytorch-build-and-test # AWS A10G GPU instance - runner: linux.g5.4xlarge.nvidia.gpu + runner: mt-l-x86aavx2-11-41-a10g timeout: 180 # Checkout the Kineto repo at the PR ref with submodules repository: ${{ github.repository }} diff --git a/.github/workflows/linux_rocm_kineto.yml b/.github/workflows/linux_rocm_kineto.yml index 99832c047..fa6d0995c 100644 --- a/.github/workflows/linux_rocm_kineto.yml +++ b/.github/workflows/linux_rocm_kineto.yml @@ -24,10 +24,10 @@ jobs: strategy: matrix: rocm: ["10.0"] - uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main with: job-name: linux-rocm-${{ matrix.rocm }}-kineto-build - runner: linux.4xlarge + runner: mt-l-x86iavx512-16-128 timeout: 180 # Checkout the Kineto repo at the PR ref with submodules repository: ${{ github.repository }} diff --git a/.github/workflows/linux_rocm_pytorch.yml b/.github/workflows/linux_rocm_pytorch.yml index 66ace7f82..012192cd0 100644 --- a/.github/workflows/linux_rocm_pytorch.yml +++ b/.github/workflows/linux_rocm_pytorch.yml @@ -24,10 +24,10 @@ jobs: strategy: matrix: rocm: ["10.0"] - uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main with: job-name: linux-rocm-${{ matrix.rocm }}-pytorch-build - runner: linux.12xlarge + runner: mt-l-x86iavx512-48-384 timeout: 180 repository: ${{ github.repository }} ref: ${{ github.ref }}