From f514b3581ab5fc7cf7d864d953b4197ffe6cfbf0 Mon Sep 17 00:00:00 2001 From: Jack-Khuu Date: Thu, 23 Jan 2025 10:03:10 -0800 Subject: [PATCH] Update run-readme-pr-linuxaarch64.yml to use correct runner (#1469) * Update run-readme-pr-linuxaarch64.yml to use correct runner * Move to linux.arm64.m7g.4xlarge * Explicitly overriding the docker-image * Bumping Cuda version to 12.6 * Updating GPU Arch type * Testing various linux_job combos: v2 cuda, v2 cpu, v1 cpu * Adding permissions to linux job v2 * Switch everything to CPU linux v2 * Test with devtoolset-11 * Remove devtoolset install * Removing devtoolset from commands --- .../workflows/run-readme-pr-linuxaarch64.yml | 74 ++++++++----------- 1 file changed, 32 insertions(+), 42 deletions(-) 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"