Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Create run-readme-pr-linuxaarch64 #1350

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
124 changes: 124 additions & 0 deletions .github/workflows/run-readme-pr-linuxaarch64.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,124 @@
name: Run the README instructions - with stories - on Linux aarch64

on:
pull_request:
push:
branches:
- main
workflow_dispatch:

jobs:
test-readme-cpu:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux-aarch64
gpu-arch-type: cuda
gpu-arch-version: "12.1"
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"
echo "tests complete"
echo "*******************************************"
echo "::endgroup::"

test-quantization-cpu:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux-aarch64
gpu-arch-type: cuda
gpu-arch-version: "12.1"
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
with:
runner: linux-aarch64
gpu-arch-type: cuda
gpu-arch-version: "12.1"
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"
echo "tests complete"
echo "*******************************************"
echo "::endgroup::"

test-advanced-cpu:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux-aarch64
gpu-arch-type: cuda
gpu-arch-version: "12.1"
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"
echo "tests complete"
echo "*******************************************"
echo "::endgroup::"

test-evaluation-cpu:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux-aarch64
gpu-arch-type: cuda
gpu-arch-version: "12.1"
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"
echo "tests complete"
echo "*******************************************"
echo "::endgroup::"
4 changes: 2 additions & 2 deletions docs/ADVANCED-USERS.md
Original file line number Diff line number Diff line change
Expand Up @@ -479,7 +479,7 @@ in a Python-free environment with AOT Inductor and ExecuTorch.
| Hardware | OS | Eager | Eager + Compile | AOT Compile | ET Runtime |
|-----|------|-----|-----|-----|-----|
| x86 | Linux | ✅ | ✅ | ✅ | ✅ |
| aarch64 | Linux | n/t | n/t | n/t | n/t |
| aarch64 | Linux | | | | n/t |
| aarch64 | macOS | ✅ | ✅ | ✅ | ✅ |
| AMD GPU | Linux | ✅ | ✅ | ✅ | ❌|
| Nvidia GPU | Linux | ✅ | ✅ | ✅ | ❌|
Expand All @@ -490,7 +490,7 @@ in a Python-free environment with AOT Inductor and ExecuTorch.
| Mobile GPU (Vulkan) | Android | ❌|❌|❌| ✅ |
| CoreML | iOS | ❌|❌|❌| ✅ |
| Hexagon DSP | Android | ❌|❌|❌| ✅ |
| Raspberry Pi 4/5 | Raspbian | n/t | n/t | n/t | ✅ |
| Raspberry Pi 4/5 | Raspbian | | | | ✅ |
| Raspberry Pi 4/5 | Android | ❌ | ❌ | ❌ | n/t |
| ARM 32b (up to v7) | any | ❌|❌|❌|❌|

Expand Down
Loading