diff --git a/CMakeLists.txt b/CMakeLists.txt index 31d07d87cfaeb8e87db85c67215e1946c74720b3..9660f9f7ad15b7edc48b21fe2e6b89cba5f63bda 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 2eb5a66e569670df1164e0d3eeec528490915464..a6b212e9a793336f0d13c6d0c694543de969fd4b 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:\