Browse Source

fix(build): disable cupti temporarily

GitOrigin-RevId: d53cac6c1e
release-1.10
Megvii Engine Team chenjinsong 2 years ago
parent
commit
b6f247d3df
2 changed files with 3 additions and 2 deletions
  1. +1
    -1
      CMakeLists.txt
  2. +2
    -1
      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})


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

@@ -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"


Loading…
Cancel
Save