diff --git a/.github/workflows/run-readme-pr-linuxaarch64.yml b/.github/workflows/run-readme-pr-linuxaarch64.yml index 1f920a12f..1f22c4f2e 100644 --- a/.github/workflows/run-readme-pr-linuxaarch64.yml +++ b/.github/workflows/run-readme-pr-linuxaarch64.yml @@ -9,22 +9,20 @@ on: jobs: test-readme-cpu: - uses: pytorch/test-infra/.github/workflows/linux_job.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + permissions: + id-token: write + contents: read with: - runner: linux-aarch64 - gpu-arch-type: cuda - gpu-arch-version: "12.1" + runner: linux.arm64.2xlarge + docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main" + gpu-arch-type: cpu-aarch64 timeout: 60 script: | echo "::group::Print machine info" uname -a echo "::endgroup::" - echo "::group::Install newer objcopy that supports --set-section-alignment" - yum install -y devtoolset-10-binutils - export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH - echo "::endgroup::" - TORCHCHAT_DEVICE=cpu .ci/scripts/run-docs readme echo "::group::Completion" @@ -33,41 +31,37 @@ jobs: echo "::endgroup::" test-quantization-cpu: - uses: pytorch/test-infra/.github/workflows/linux_job.yml@main + uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main + permissions: + id-token: write + contents: read with: - runner: linux-aarch64 - gpu-arch-type: cuda - gpu-arch-version: "12.1" + runner: linux.arm64.2xlarge + docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main" + gpu-arch-type: cpu-aarch64 timeout: 60 script: | echo "::group::Print machine info" uname -a echo "::endgroup::" - echo "::group::Install newer objcopy that supports --set-section-alignment" - yum install -y devtoolset-10-binutils - export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH - echo "::endgroup::" - TORCHCHAT_DEVICE=cpu .ci/scripts/run-docs quantization test-gguf-cpu: uses: pytorch/test-infra/.github/workflows/linux_job.yml@main + permissions: + id-token: write + contents: read with: - runner: linux-aarch64 - gpu-arch-type: cuda - gpu-arch-version: "12.1" + runner: linux.arm64.2xlarge + docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main" + gpu-arch-type: cpu-aarch64 timeout: 60 script: | echo "::group::Print machine info" uname -a echo "::endgroup::" - echo "::group::Install newer objcopy that supports --set-section-alignment" - yum install -y devtoolset-10-binutils - export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH - echo "::endgroup::" - TORCHCHAT_DEVICE=cpu .ci/scripts/run-docs gguf echo "::group::Completion" @@ -77,21 +71,19 @@ jobs: test-advanced-cpu: uses: pytorch/test-infra/.github/workflows/linux_job.yml@main + permissions: + id-token: write + contents: read with: - runner: linux-aarch64 - gpu-arch-type: cuda - gpu-arch-version: "12.1" + runner: linux.arm64.2xlarge + docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main" + gpu-arch-type: cpu-aarch64 timeout: 60 script: | echo "::group::Print machine info" uname -a echo "::endgroup::" - echo "::group::Install newer objcopy that supports --set-section-alignment" - yum install -y devtoolset-10-binutils - export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH - echo "::endgroup::" - TORCHCHAT_DEVICE=cpu .ci/scripts/run-docs advanced echo "::group::Completion" @@ -101,21 +93,19 @@ jobs: test-evaluation-cpu: uses: pytorch/test-infra/.github/workflows/linux_job.yml@main + permissions: + id-token: write + contents: read with: - runner: linux-aarch64 - gpu-arch-type: cuda - gpu-arch-version: "12.1" + runner: linux.arm64.2xlarge + docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main" + gpu-arch-type: cpu-aarch64 timeout: 60 script: | echo "::group::Print machine info" uname -a echo "::endgroup::" - echo "::group::Install newer objcopy that supports --set-section-alignment" - yum install -y devtoolset-10-binutils - export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH - echo "::endgroup::" - TORCHCHAT_DEVICE=cpu .ci/scripts/run-docs evaluation echo "::group::Completion"