From 8fed114a516584d8c2d8d2fa3c4540d99ae4e991 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 11 Jul 2022 19:07:58 +0800 Subject: [PATCH] fix(build): disable cupti temporarily GitOrigin-RevId: d53cac6c1ecd68bf2a6a17de20536891b704709e --- CMakeLists.txt | 2 +- scripts/whl/manylinux2014/build_wheel_common.sh | 6 +----- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 31d07d87c..9660f9f7a 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -105,7 +105,7 @@ if(MGE_PROFILE_COMPILE_TIME) endif() # TODO: add windows support -cmake_dependent_option(MGE_WITH_CUPTI "Build with CUPTI" ON +cmake_dependent_option(MGE_WITH_CUPTI "Build with CUPTI" OFF "MGE_WITH_CUDA;MGE_BUILD_IMPERATIVE_RT;NOT MSVC;NOT WIN32" OFF) set(MGB_CUPTI ${MGE_WITH_CUPTI}) diff --git a/scripts/whl/manylinux2014/build_wheel_common.sh b/scripts/whl/manylinux2014/build_wheel_common.sh index 2eb5a66e5..a6b212e9a 100755 --- a/scripts/whl/manylinux2014/build_wheel_common.sh +++ b/scripts/whl/manylinux2014/build_wheel_common.sh @@ -61,8 +61,7 @@ echo "Build with ${SDK_NAME}" if [ $SDK_NAME == "cu101" ];then CUDA_COPY_LIB_LIST="\ - ${CUDA_LIB_DIR}/libnvrtc.so.10.1:\ - ${CUDA_LIB_DIR}/libcupti.so.10.1" + ${CUDA_LIB_DIR}/libnvrtc.so.10.1" EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=OFF -DMGE_WITH_CUBLAS_SHARED=OFF" BUILD_GCC8="ON" REQUIR_CUDA_VERSION="10010" @@ -81,7 +80,6 @@ elif [ $SDK_NAME == "cu102_JetsonNano" ];then ${CUDA_LIB_DIR}/libnvrtc.so.10.2:\ ${CUDA_LIB_DIR}/libcublasLt.so.10:\ ${CUDA_LIB_DIR}/libcublas.so.10:\ - ${CUDA_LIB_DIR}/libcupti.so.10.2:\ ${CUDNN_LIB_DIR}/libcudnn_adv_infer.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_adv_train.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_cnn_infer.so.8:\ @@ -113,7 +111,6 @@ elif [ $SDK_NAME == "cu111" ];then ${CUDA_LIB_DIR}/libnvrtc.so.11.1:\ ${CUDA_LIB_DIR}/libcublasLt.so.11:\ ${CUDA_LIB_DIR}/libcublas.so.11:\ - ${CUDA_LIB_DIR}/libcupti.so.11.1:\ ${CUDNN_LIB_DIR}/libcudnn_adv_infer.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_adv_train.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_cnn_infer.so.8:\ @@ -140,7 +137,6 @@ elif [ $SDK_NAME == "cu112" ];then ${CUDA_LIB_DIR}/libnvrtc.so.11.2:\ ${CUDA_LIB_DIR}/libcublasLt.so.11:\ ${CUDA_LIB_DIR}/libcublas.so.11:\ - ${CUDA_LIB_DIR}/libcupti.so.11.2:\ ${CUDNN_LIB_DIR}/libcudnn_adv_infer.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_adv_train.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_cnn_infer.so.8:\ -- GitLab