From a9b60fbfb580ae1d410cc5205596872d86414617 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 9 Jun 2021 14:17:38 +0800
Subject: [PATCH] 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
---
 scripts/whl/manylinux2014/build_wheel_common.sh | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/scripts/whl/manylinux2014/build_wheel_common.sh b/scripts/whl/manylinux2014/build_wheel_common.sh
index 1c1a2ede..ce719ec4 100755
--- a/scripts/whl/manylinux2014/build_wheel_common.sh
+++ b/scripts/whl/manylinux2014/build_wheel_common.sh
@@ -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} \
-- 
GitLab