Browse Source

fix(build): disable cupti temporarily

GitOrigin-RevId: d53cac6c1e
HuaHua404-patch-4
Megvii Engine Team 2 years ago
parent
commit
8fed114a51
2 changed files with 2 additions and 6 deletions
  1. +1
    -1
      CMakeLists.txt
  2. +1
    -5
      scripts/whl/manylinux2014/build_wheel_common.sh

+ 1
- 1
CMakeLists.txt View File

@@ -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})


+ 1
- 5
scripts/whl/manylinux2014/build_wheel_common.sh View File

@@ -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:\


Loading…
Cancel
Save