mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-14 22:25:03 +08:00
d2be06f673
---- - Updates Arm Compute Library (ACL) to v52.6.0 - v52.6.0 contains https://github.com/ARM-software/ComputeLibrary/pull/1201 which fixes crashes with tensors of sizes > 2^31-1 fixes: #165654 Pull Request resolved: https://github.com/pytorch/pytorch/pull/165904 Approved by: https://github.com/malfet
27 lines
681 B
Bash
Executable File
27 lines
681 B
Bash
Executable File
#!/bin/bash
|
|
# Script used only in CD pipeline
|
|
|
|
set -eux
|
|
|
|
ACL_VERSION=${ACL_VERSION:-"v52.6.0"}
|
|
ACL_INSTALL_DIR="/acl"
|
|
|
|
# Clone ACL
|
|
git clone https://github.com/ARM-software/ComputeLibrary.git -b "${ACL_VERSION}" --depth 1 --shallow-submodules
|
|
|
|
ACL_CHECKOUT_DIR="ComputeLibrary"
|
|
# Build with scons
|
|
pushd $ACL_CHECKOUT_DIR
|
|
scons -j8 Werror=0 debug=0 neon=1 opencl=0 embed_kernels=0 \
|
|
os=linux arch=armv8a build=native multi_isa=1 \
|
|
fixed_format_kernels=1 openmp=1 cppthreads=0
|
|
popd
|
|
|
|
# Install ACL
|
|
sudo mkdir -p ${ACL_INSTALL_DIR}
|
|
for d in arm_compute include utils support src build
|
|
do
|
|
sudo cp -r ${ACL_CHECKOUT_DIR}/${d} ${ACL_INSTALL_DIR}/${d}
|
|
done
|
|
|
|
rm -rf $ACL_CHECKOUT_DIR |