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 2ea03aaf5af8e753b7269d0231ed9386dafbb32f..acb4cf7702de6940ebcbc76f30bd716d2b6a9c72 100755 --- a/scripts/whl/manylinux2014/build_wheel_common.sh +++ b/scripts/whl/manylinux2014/build_wheel_common.sh @@ -60,7 +60,8 @@ fi echo "Build with ${SDK_NAME}" if [ $SDK_NAME == "cu101" ];then - CUDA_COPY_LIB_LIST="${CUDA_LIB_DIR}/libnvrtc.so.10.1" + CUDA_COPY_LIB_LIST="\ + ${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"