mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-11 14:28:03 +08:00
d9c80ef97d
---- 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, https://github.com/aditew01
25 lines
522 B
Bash
Executable File
25 lines
522 B
Bash
Executable File
#!/bin/bash
|
|
# Script used only in CD pipeline
|
|
|
|
set -ex
|
|
|
|
OPENBLAS_VERSION=${OPENBLAS_VERSION:-"v0.3.30"}
|
|
|
|
# Clone OpenBLAS
|
|
git clone https://github.com/OpenMathLib/OpenBLAS.git -b "${OPENBLAS_VERSION}" --depth 1 --shallow-submodules
|
|
|
|
OPENBLAS_CHECKOUT_DIR="OpenBLAS"
|
|
OPENBLAS_BUILD_FLAGS="
|
|
NUM_THREADS=128
|
|
USE_OPENMP=1
|
|
NO_SHARED=0
|
|
DYNAMIC_ARCH=1
|
|
TARGET=ARMV8
|
|
CFLAGS=-O3
|
|
BUILD_BFLOAT16=1
|
|
"
|
|
|
|
make -j8 ${OPENBLAS_BUILD_FLAGS} -C $OPENBLAS_CHECKOUT_DIR
|
|
sudo make install -C $OPENBLAS_CHECKOUT_DIR
|
|
|
|
rm -rf $OPENBLAS_CHECKOUT_DIR |