#!/usr/bin/env bash set -e BUILD_TYPE=Release MGE_WITH_CUDA=OFF MGE_INFERENCE_ONLY=ON MGE_WINDOWS_BUILD_ARCH=x64 MGE_WINDOWS_BUILD_MARCH=m64 MGE_ARCH=x86_64 REMOVE_OLD_BUILD=false NINJA_VERBOSE=OFF BUILD_DEVELOP=ON NINJA_DRY_RUN=OFF SPECIFIED_TARGET="install/strip" READLINK=readlink OS=$(uname -s) if [[ $OS =~ "NT" ]]; then echo "Windows no need strip, caused by pdb file always split with exe" SPECIFIED_TARGET="install" fi if [ $OS = "Darwin" ];then READLINK=greadlink fi SRC_DIR=$($READLINK -f "`dirname $0`/../../") source $SRC_DIR/scripts/cmake-build/utils/utils.sh config_ninja_default_max_jobs echo "EXTRA_CMAKE_ARGS: ${EXTRA_CMAKE_ARGS}" function usage() { echo "$0 args1 args2 .." echo "available args detail:" echo "-d : Build with Debug mode, default Release mode" echo "-c : Build with CUDA, default without CUDA" echo "-t : Build with training mode, default inference only" echo "-m : Build with m32 mode(only for windows build), default m64" echo "-r : remove old build dir before make, default off" echo "-v : ninja with verbose and explain, default off" echo "-s : Do not build develop even build with training mode, default on when build with training, always for wheel" echo "-n : ninja with -n dry run (don't run commands but act like they succeeded)" echo "-j : run N jobs in parallel for ninja, defaut is cpu_number + 2" echo "-e : build a specified target (always for debug, NOTICE: do not do strip/install target when use -e)" echo "-l : list CMakeLists.txt all options, can be use to config EXTRA_CMAKE_ARGS" echo "-h : show usage" echo "append other cmake config by config EXTRA_CMAKE_ARGS, for example, enable MGE_WITH_TEST and build with Debug mode:" echo "EXTRA_CMAKE_ARGS=\"-DMGE_WITH_TEST=ON\" $0 -d" exit -1 } while getopts "lnsrhdctmve:j:" arg do case $arg in j) NINJA_MAX_JOBS=$OPTARG echo "config NINJA_MAX_JOBS to ${NINJA_MAX_JOBS}" ;; l) echo "list CMakeLists.txt all options, can be used to config EXTRA_CMAKE_ARGS" show_cmakelist_options exit 0 ;; d) echo "Build with Debug mode" BUILD_TYPE=Debug ;; c) echo "Build with CUDA" MGE_WITH_CUDA=ON ;; t) echo "Build with training mode" MGE_INFERENCE_ONLY=OFF ;; h) echo "show usage" usage ;; r) echo "config REMOVE_OLD_BUILD=true" REMOVE_OLD_BUILD=true ;; s) echo "config BUILD_DEVELOP=OFF" BUILD_DEVELOP=OFF ;; v) echo "config NINJA_VERBOSE=ON" NINJA_VERBOSE=ON ;; n) echo "config NINJA_DRY_RUN=ON" NINJA_DRY_RUN=ON ;; m) echo "build for m32(only valid use for windows)" MGE_WINDOWS_BUILD_ARCH=x86 MGE_WINDOWS_BUILD_MARCH=m32 MGE_ARCH=i386 ;; e) SPECIFIED_TARGET=$OPTARG ;; ?) echo "unkonw argument" usage ;; esac done echo "------------------------------------" echo "build config summary:" echo "BUILD_TYPE: $BUILD_TYPE" echo "MGE_WITH_CUDA: $MGE_WITH_CUDA" echo "MGE_INFERENCE_ONLY: $MGE_INFERENCE_ONLY" echo "SPECIFIED_TARGET: ${SPECIFIED_TARGET}" echo "NINJA_MAX_JOBS: ${NINJA_MAX_JOBS}" echo "------------------------------------" if [ $OS = "Darwin" ];then if [ $MGE_WITH_CUDA = "ON" ];then echo "MACOS DO NOT SUPPORT TensorRT, ABORT NOW!!" exit -1 fi elif [[ $OS =~ "NT" ]]; then echo "BUILD in NT ..." fi if [ ${MGE_INFERENCE_ONLY} = "ON" ]; then echo "config BUILD_DEVELOP=OFF when MGE_INFERENCE_ONLY=ON" BUILD_DEVELOP=OFF fi function cmake_build() { BUILD_DIR=$SRC_DIR/build_dir/host/MGE_WITH_CUDA_$1/MGE_INFERENCE_ONLY_$2/$3/build INSTALL_DIR=$BUILD_DIR/../install MGE_WITH_CUDA=$1 MGE_INFERENCE_ONLY=$2 BUILD_TYPE=$3 echo "build dir: $BUILD_DIR" echo "install dir: $INSTALL_DIR" echo "build type: $BUILD_TYPE" echo "MGE_WITH_CUDA: $MGE_WITH_CUDA" echo "MGE_INFERENCE_ONLY: $MGE_INFERENCE_ONLY" try_remove_old_build $REMOVE_OLD_BUILD $BUILD_DIR $INSTALL_DIR echo "create build dir" mkdir -p $BUILD_DIR mkdir -p $INSTALL_DIR cd_real_build_dir $BUILD_DIR # fork a new bash to handle EXTRA_CMAKE_ARGS env with space bash -c "cmake -G Ninja \ -DCMAKE_BUILD_TYPE=$BUILD_TYPE \ -DMGE_INFERENCE_ONLY=$MGE_INFERENCE_ONLY \ -DMGE_WITH_CUDA=$MGE_WITH_CUDA \ -DCMAKE_INSTALL_PREFIX=$INSTALL_DIR \ ${EXTRA_CMAKE_ARGS} \ ${SRC_DIR} " config_ninja_target_cmd ${NINJA_VERBOSE} ${BUILD_DEVELOP} "${SPECIFIED_TARGET}" ${NINJA_DRY_RUN} ${NINJA_MAX_JOBS} bash -c "${NINJA_CMD}" } function windows_env_err() { echo "check windows env failed!!" echo "please install env refs for: scripts/cmake-build/BUILD_README.md" exit -1 } function prepare_env_for_windows_build() { echo "check Visual Studio install path env..." if [[ -z $VS_PATH ]];then echo "can not find visual_studio_path env, pls export you Visual Studio install dir to VS_PATH" echo "examle for export Visual Studio 2019 Enterprise default install dir" echo "export VS_PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2019/Enterprise" exit -1 fi echo $VS_PATH # only use cmake/Ninja install from Visual Studio, if not, may build failed # some user env may install cmake/clang-cl/Ninja at windows-git-bash env, so we put Visual Studio # path at the head of PATH, and check the valid echo "check cmake install..." export PATH=$VS_PATH/Common7/IDE/CommonExtensions/Microsoft/CMake/CMake/bin/:$PATH which cmake cmake_loc=`which cmake` if [[ $cmake_loc =~ "Visual" ]]; then echo "cmake valid ..." else echo "cmake Invalid: ..." windows_env_err fi echo "check clang-cl install..." # llvm install by Visual Studio have some issue, eg, link crash on large project, so we # use official LLVM download from https://releases.llvm.org/download.html if [[ -z ${LLVM_PATH} ]];then echo "can not find LLVM_PATH env, pls export you LLVM install dir to LLVM_PATH" echo "examle for export LLVM_12_0_1" echo "export LLVM_PATH=/c/Program\ Files/LLVM_12_0_1" exit -1 fi echo ${LLVM_PATH} export PATH=${LLVM_PATH}/bin/:$PATH clang_loc=`which clang-cl` if [[ $clang_loc =~ "Visual" ]]; then echo "clang-cl Invalid: we do not support use LLVM installed by Visual Studio" windows_env_err else echo "clang-cl valid ..." fi echo "check Ninja install..." export PATH=$VS_PATH/Common7/IDE/CommonExtensions/Microsoft/CMake/Ninja/:$PATH which Ninja ninja_loc=`which Ninja` if [[ $ninja_loc =~ "Visual" ]]; then echo "Ninja valid ..." else echo "Ninja Invalid: ..." windows_env_err fi echo "put vcvarsall.bat path to PATH env.." export PATH=$VS_PATH/VC/Auxiliary/Build:$PATH echo "config cuda/cudnn/TensorRT env..." export NIVIDA_INSTALL_PRE=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit export CUDA_V=v10.1 export CUDNN_V=cudnn-10.1-windows10-x64-v7.6.5.32 export TRT_V=TensorRT-6.0.1.5 export CUDA_PATH=$NIVIDA_INSTALL_PRE/CUDA/${CUDA_V} export PATH=$PATH:$CUDA_PATH/bin export CUDA_BIN_PATH=$CUDA_PATH export PC_CUDNN_INCLUDE_DIRS=$NIVIDA_INSTALL_PRE/${CUDNN_V}/cuda/include export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$NIVIDA_INSTALL_PRE/${TRT_V}/lib:$NIVIDA_INSTALL_PRE/CUDA/${CUDA_V}/lib/x64:$NIVIDA_INSTALL_PRE/${CUDNN_V}/cuda/lib/x64 export CPATH=$CPATH:$NIVIDA_INSTALL_PRE/${TRT_V}/include:$NIVIDA_INSTALL_PRE/CUDA/${CUDA_V}/include:$NIVIDA_INSTALL_PRE/CUDA/${CUDA_V}/include/nvtx3:$PC_CUDNN_INCLUDE_DIRS export LIBRARY_PATH=$LIBRARY_PATH:$LD_LIBRARY_PATH export INCLUDE=$INCLUDE:$CPATH export CUDNN_ROOT_DIR=${NIVIDA_INSTALL_PRE}/${CUDNN_V}/cuda export TRT_ROOT_DIR=${NIVIDA_INSTALL_PRE}/${TRT_V} # python version will be config by whl build script or ci script, we need # a DFT version for build success when we just call host_build.sh if [[ -z ${ALREADY_CONFIG_PYTHON_VER} ]] then echo "config a default python3" DFT_PYTHON_BIN=/c/Users/${USER}/mge_whl_python_env/3.8.3 if [ ! -f "${DFT_PYTHON_BIN}/python3.exe" ]; then echo "ERR: can not find ${DFT_PYTHON_BIN}/python3.exe , Invalid env" windows_env_err else echo "put python3 to env..." export PATH=${DFT_PYTHON_BIN}:$PATH which python3 fi fi echo "export swig pwd to PATH" export PATH=/c/Users/${USER}/swigwin-4.0.2::$PATH } function cmake_build_windows() { # windows do not support long path, so we cache the BUILD_DIR ASAP prepare_env_for_windows_build BUILD_DIR=$SRC_DIR/build_dir/host/build # ninja have compat issue with bash env, which fork from windows-git # which will map C: to /c/c/ dir, which will lead to install file to /c/c/.. # as a solution, we map INSTALL_DIR to INSTALL_DIR_WIN (/c --> C:) INSTALL_DIR=${BUILD_DIR}/../install INSTALL_DIR_PREFIX=${INSTALL_DIR:0:2} if [ ${INSTALL_DIR_PREFIX} = "/c" ];then echo "INSTALL_DIR_PREFIX is ${INSTALL_DIR_PREFIX}, map to C:" INSTALL_DIR_WIN="C:${INSTALL_DIR:2}" else INSTALL_DIR_WIN=${INSTALL_DIR} fi MGE_WITH_CUDA=$1 MGE_INFERENCE_ONLY=$2 BUILD_TYPE=$3 echo "build dir: $BUILD_DIR" echo "install dir: $INSTALL_DIR" echo "install dir for ninja: $INSTALL_DIR_WIN" echo "build type: $BUILD_TYPE" echo "MGE_WITH_CUDA: $MGE_WITH_CUDA" echo "MGE_INFERENCE_ONLY: $MGE_INFERENCE_ONLY" try_remove_old_build $REMOVE_OLD_BUILD $BUILD_DIR $INSTALL_DIR echo "create build dir" mkdir -p $BUILD_DIR cd_real_build_dir $BUILD_DIR echo "now try build windows native with cmake/clang-ci/Ninja/Visual Studio ....." export CFLAGS=-$MGE_WINDOWS_BUILD_MARCH export CXXFLAGS=-$MGE_WINDOWS_BUILD_MARCH cmd.exe /c " \ vcvarsall.bat $MGE_WINDOWS_BUILD_ARCH && cmake -G "Ninja" \ -DMGE_ARCH=$MGE_ARCH \ -DMGE_INFERENCE_ONLY=$MGE_INFERENCE_ONLY \ -DMGE_WITH_CUDA=$MGE_WITH_CUDA \ -DCMAKE_BUILD_TYPE=$BUILD_TYPE \ -DCMAKE_INSTALL_PREFIX:PATH=$INSTALL_DIR_WIN \ -DCMAKE_C_COMPILER=clang-cl.exe \ -DCMAKE_CXX_COMPILER=clang-cl.exe \ -DCMAKE_MAKE_PROGRAM=ninja.exe \ ${EXTRA_CMAKE_ARGS} ../../.. " config_ninja_target_cmd ${NINJA_VERBOSE} ${BUILD_DEVELOP} "${SPECIFIED_TARGET}" ${NINJA_DRY_RUN} ${NINJA_MAX_JOBS} cmd.exe /c " vcvarsall.bat $MGE_WINDOWS_BUILD_ARCH && ${NINJA_CMD} " } if [[ $OS =~ "NT" ]]; then if [ ${MGE_ARCH} = "i386" ] && [ ${MGE_INFERENCE_ONLY} = "OFF" ]; then echo "ERR: training mode(-t) only support 64 bit mode" echo "pls remove -t or remove -m" exit -1 fi cmake_build_windows $MGE_WITH_CUDA $MGE_INFERENCE_ONLY $BUILD_TYPE else cmake_build $MGE_WITH_CUDA $MGE_INFERENCE_ONLY $BUILD_TYPE fi