mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Robert Hardwick 582d278983 Build and Install Arm Compute Library in manylinux docker image (#159737 )

----

This PR will be part of a series of PR's that aims to remove `.ci/aarch64_linux` folder entirely, such that Aarch64 manylinux build happens as part of `.ci/manywheel/build.sh`, the same as other platforms.

In this PR:

- We prebuild + install Arm Compute Library in the manylinux docker image ( at /acl ), instead of a build time for every pytorch build.  Also updated jammy install path to be /acl too.
- We can therefore remove build_ArmComputeLibrary functions from the ci build scripts.
- There is also some refactoring of install_openblas.sh and install_acl.sh to align them together ( similar formatting, similar variable names, same place for version number update )
- We had 2 places to define openblas version, this has been reduced to 1 now ( install_openblas.sh ).
- ACL_VERSION and OPENBLAS_VERSION are now able to be overriden at build.sh level for developers, but there is only 1 version of each hardcoded for ci.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159737
Approved by: https://github.com/seemethere
ghstack dependencies: #160078

2025-09-10 15:39:38 +00:00

aarch64_ci_build.sh

Move prioritized text linker optimization code from setup.py to cmake (#160078 )

2025-09-10 09:21:53 +00:00

aarch64_ci_setup.sh

Revert "[build] pin setuptools>=77 to enable PEP 639 (#158104 )"

2025-07-21 22:46:53 +00:00

aarch64_wheel_ci_build.py

Build and Install Arm Compute Library in manylinux docker image (#159737 )

2025-09-10 15:39:38 +00:00

build_aarch64_wheel.py

Build and Install Arm Compute Library in manylinux docker image (#159737 )

2025-09-10 15:39:38 +00:00

embed_library.py

[CD] Move linux-aarch64 build scripts (#139815 )

2024-11-06 04:16:48 +00:00

README.md

[CD] Move linux-aarch64 build scripts (#139815 )

2024-11-06 04:16:48 +00:00

README.md

Aarch64 (ARM/Graviton) Support Scripts

Scripts for building aarch64 PyTorch PIP Wheels. These scripts build the following wheels:

torch
torchvision
torchaudio
torchtext
torchdata

Aarch64_ci_build.sh

This script is design to support CD operations within PyPi manylinux aarch64 container, and be executed in the container. It prepares the container and then executes aarch64_wheel_ci_build.py to build the wheels. The script "assumes" the PyTorch repo is located at: /pytorch and will put the wheels into /artifacts.

Usage

DESIRED_PYTHON=<PythonVersion> aarch64_ci_build.sh

NOTE: CI build is currently EXPERMINTAL

Build_aarch64_wheel.py

This app allows a person to build using AWS EC3 resources and requires AWS-CLI and Boto3 with AWS credentials to support building EC2 instances for the wheel builds. Can be used in a codebuild CD or from a local system.

Usage

build_aarch64_wheel.py --key-name <YourPemKey> --use-docker --python 3.8 --branch <RCtag>