mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
ef281f850a
### Description 1. Add XNNPack build on Linux ARM64 2. Build only one python wheel for PR request. [AB#49763](https://aiinfra.visualstudio.com/6a833879-cd9b-44a4-a9de-adc2d818f13c/_workitems/edit/49763) ### Motivation and Context Why I add xnnpack build on Linux ARM64 rather than Windows ARM64. Becuase KleidiAI doesn't support Windows ``` IF(XNNPACK_TARGET_PROCESSOR STREQUAL "arm64" AND XNNPACK_ENABLE_ARM_I8MM AND NOT CMAKE_C_COMPILER_ID STREQUAL "MSVC") IF (XNNPACK_ENABLE_KLEIDIAI) MESSAGE(STATUS "Enabling KleidiAI for Arm64") ENDIF() ELSE() SET(XNNPACK_ENABLE_KLEIDIAI OFF) ENDIF() ``` ---------
43 lines
1.6 KiB
Bash
Executable file
43 lines
1.6 KiB
Bash
Executable file
#!/bin/bash
|
|
set -e -x
|
|
BUILD_CONFIG="Release"
|
|
|
|
while getopts "i:d:x:c:p:" parameter_Option
|
|
do case "${parameter_Option}"
|
|
in
|
|
i) DOCKER_IMAGE=${OPTARG};;
|
|
d) DEVICE=${OPTARG};;
|
|
x) BUILD_EXTR_PAR=${OPTARG};;
|
|
c) BUILD_CONFIG=${OPTARG};;
|
|
p) PYTHON_EXES=${OPTARG};;
|
|
*) echo "Usage: $0 -i <docker_image> -d <GPU|CPU> [-x <extra_build_arg>] [-c <build_config>] [-p <python_exe_path>]"
|
|
exit 1;;
|
|
esac
|
|
done
|
|
|
|
mkdir -p "${HOME}/.onnx"
|
|
DOCKER_SCRIPT_OPTIONS="-d ${DEVICE} -c ${BUILD_CONFIG}"
|
|
|
|
if [ "${BUILD_EXTR_PAR}" != "" ] ; then
|
|
DOCKER_SCRIPT_OPTIONS+=" -x ${BUILD_EXTR_PAR}"
|
|
fi
|
|
|
|
docker run --rm \
|
|
--volume /data/onnx:/data/onnx:ro \
|
|
--volume "${BUILD_SOURCESDIRECTORY}:/onnxruntime_src" \
|
|
--volume "${BUILD_BINARIESDIRECTORY}:/build" \
|
|
--volume /data/models:/build/models:ro \
|
|
--volume "${HOME}/.onnx:/home/onnxruntimedev/.onnx" \
|
|
-w /onnxruntime_src \
|
|
-e NIGHTLY_BUILD \
|
|
-e BUILD_BUILDNUMBER \
|
|
-e ORT_DISABLE_PYTHON_PACKAGE_LOCAL_VERSION \
|
|
-e DEFAULT_TRAINING_PACKAGE_DEVICE \
|
|
$ADDITIONAL_DOCKER_PARAMETER \
|
|
$DOCKER_IMAGE tools/ci_build/github/linux/build_linux_python_package.sh -c $BUILD_CONFIG -p $PYTHON_EXES $DOCKER_SCRIPT_OPTIONS
|
|
|
|
sudo rm -rf "${BUILD_BINARIESDIRECTORY}/${BUILD_CONFIG}/onnxruntime" "${BUILD_BINARIESDIRECTORY}/${BUILD_CONFIG}/pybind11" \
|
|
"${BUILD_BINARIESDIRECTORY}/${BUILD_CONFIG}/models" "${BUILD_BINARIESDIRECTORY}/${BUILD_CONFIG}/_deps" \
|
|
"${BUILD_BINARIESDIRECTORY}/${BUILD_CONFIG}/CMakeFiles"
|
|
cd "${BUILD_BINARIESDIRECTORY}/${BUILD_CONFIG}"
|
|
find -executable -type f > "${BUILD_BINARIESDIRECTORY}/${BUILD_CONFIG}/perms.txt"
|