提交 a9b60fbf 编写于 作者: M Megvii Engine Team

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: 1dc2115948b1d71124e0f7953788d9201af0caef
上级 2eea0009
......@@ -96,8 +96,7 @@ elif [ $SDK_NAME == "cu111" ];then
${CUDNN_LIB_DIR}/libcudnn.so.8"
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
EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON \
-DMGE_CUDA_GENCODE=\"-gencode arch=compute_61,code=sm_61 \
......@@ -278,11 +277,14 @@ fi
set +x
docker_args="-it"
if [ -z "${CI_SERVER_NAME}" ]; then
Target="null"
CI_SERVER_NAME="null"
fi
if [ ${CI_SERVER_NAME} = "GitLab" ];then
docker_args="-i"
fi
if [ ${IN_CI} = "true" ];then
EXTRA_CMAKE_FLAG=" ${EXTRA_CMAKE_FLAG} -DMGE_WITH_TEST=ON"
fi
docker run --rm ${docker_args} $TMPFS_ARGS \
-e UID=${USERID} \
-e PUBLIC_VERSION_POSTFIX=${PUBLIC_VERSION_POSTFIX} \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册