Browse Source

fix(ci/lite): reopen lite_test build by cmake

as some reason, lite_test need static link lite
when cuda enable on gcc7 and gcc8, if not
cask_trt::AbiInfo::~AbiInfo will double call at
atexit stage, which will lead double free at the
end of test, gcc9 do not have this issue, for
compat all CI env, we use static link!!!

GitOrigin-RevId: 1dc2115948
release-1.5
Megvii Engine Team 4 years ago
parent
commit
a9b60fbfb5
1 changed files with 5 additions and 3 deletions
  1. +5
    -3
      scripts/whl/manylinux2014/build_wheel_common.sh

+ 5
- 3
scripts/whl/manylinux2014/build_wheel_common.sh View File

@@ -96,8 +96,7 @@ elif [ $SDK_NAME == "cu111" ];then
${CUDNN_LIB_DIR}/libcudnn.so.8" ${CUDNN_LIB_DIR}/libcudnn.so.8"


if [ ${IN_CI} = "true" ] && [ ${machine} == "aarch64" ]; then if [ ${IN_CI} = "true" ] && [ ${machine} == "aarch64" ]; then
EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON \
-DMGE_WITH_TEST=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_75,code=sm_75\" "
EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_75,code=sm_75\" "
else else
EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON \ EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON \
-DMGE_CUDA_GENCODE=\"-gencode arch=compute_61,code=sm_61 \ -DMGE_CUDA_GENCODE=\"-gencode arch=compute_61,code=sm_61 \
@@ -278,11 +277,14 @@ fi
set +x set +x
docker_args="-it" docker_args="-it"
if [ -z "${CI_SERVER_NAME}" ]; then if [ -z "${CI_SERVER_NAME}" ]; then
Target="null"
CI_SERVER_NAME="null"
fi fi
if [ ${CI_SERVER_NAME} = "GitLab" ];then if [ ${CI_SERVER_NAME} = "GitLab" ];then
docker_args="-i" docker_args="-i"
fi fi
if [ ${IN_CI} = "true" ];then
EXTRA_CMAKE_FLAG=" ${EXTRA_CMAKE_FLAG} -DMGE_WITH_TEST=ON"
fi
docker run --rm ${docker_args} $TMPFS_ARGS \ docker run --rm ${docker_args} $TMPFS_ARGS \
-e UID=${USERID} \ -e UID=${USERID} \
-e PUBLIC_VERSION_POSTFIX=${PUBLIC_VERSION_POSTFIX} \ -e PUBLIC_VERSION_POSTFIX=${PUBLIC_VERSION_POSTFIX} \


Loading…
Cancel
Save