提交 8110bb21 编写于 作者: M Megvii Engine Team

fix(cmake/env): fix a bug about env set in cmake build mode

GitOrigin-RevId: 492eaf428a1ab1f8c671730af139fa12cedae3f7
上级 0a665ea4
...@@ -34,6 +34,18 @@ set_property(CACHE MGE_ARCH PROPERTY STRINGS AUTO ...@@ -34,6 +34,18 @@ set_property(CACHE MGE_ARCH PROPERTY STRINGS AUTO
) )
set (MGE_EXPORT_TARGETS MegEngine-targets) set (MGE_EXPORT_TARGETS MegEngine-targets)
if(NOT "$ENV{LD_LIBRARY_PATH}" STREQUAL "")
string(REPLACE ":" ";" ALTER_LD_LIBRARY_PATHS $ENV{LD_LIBRARY_PATH})
else()
set(ALTER_LD_LIBRARY_PATHS "")
endif()
if(NOT "$ENV{LIBRARY_PATH}" STREQUAL "")
string(REPLACE ":" ";" ALTER_LIBRARY_PATHS $ENV{LIBRARY_PATH})
else()
set(ALTER_LIBRARY_PATHS "")
endif()
option(MGE_WITH_JIT "Build MegEngine with JIT." ON) option(MGE_WITH_JIT "Build MegEngine with JIT." ON)
option(MGE_WITH_JIT_MLIR "Build MegEngine with MLIR JIT." OFF) option(MGE_WITH_JIT_MLIR "Build MegEngine with MLIR JIT." OFF)
option(MGE_WITH_HALIDE "Build MegEngine with Halide JIT" OFF) option(MGE_WITH_HALIDE "Build MegEngine with Halide JIT" OFF)
......
if($ENV{LIBRARY_PATH})
string(REPLACE ":" ";" SYSTEM_LIBRARY_PATHS $ENV{LIBRARY_PATH})
endif()
find_library(ACLRT_LIBRARY find_library(ACLRT_LIBRARY
NAMES libascendcl.so NAMES libascendcl.so
PATHS $ENV{LD_LIBRARY_PATH} "$ENV{ACLRT_HOME}/lib64/stub" ${CMAKE_INSTALL_PREFIX} PATHS ${ALTER_LD_LIBRARY_PATHS} "$ENV{ACLRT_HOME}/lib64/stub" ${CMAKE_INSTALL_PREFIX}
HINTS ${SYSTEM_LIBRARY_PATHS} HINTS ${ALTER_LIBRARY_PATHS}
PATH_SUFFIXES stub PATH_SUFFIXES stub
DOC "ACL library." ) DOC "ACL library." )
......
if($ENV{LIBRARY_PATH})
string(REPLACE ":" ";" SYSTEM_LIBRARY_PATHS $ENV{LIBRARY_PATH})
endif()
find_library(CNDEV_LIBRARY find_library(CNDEV_LIBRARY
NAMES libcndev.so NAMES libcndev.so
PATHS $ENV{LD_LIBRARY_PATH} "$ENV{NEUWARE_HOME}/lib64" ${CMAKE_INSTALL_PREFIX} PATHS ${ALTER_LD_LIBRARY_PATHS} "$ENV{NEUWARE_HOME}/lib64" ${CMAKE_INSTALL_PREFIX}
HINTS ${SYSTEM_LIBRARY_PATHS} HINTS ${ALTER_LIBRARY_PATHS}
PATH_SUFFIXES lib lib64 PATH_SUFFIXES lib lib64
DOC "CNDEV library." ) DOC "CNDEV library." )
......
if($ENV{LIBRARY_PATH})
string(REPLACE ":" ";" SYSTEM_LIBRARY_PATHS $ENV{LIBRARY_PATH})
endif()
find_library(CNML_LIBRARY find_library(CNML_LIBRARY
NAMES libcnml.so NAMES libcnml.so
PATHS $ENV{LD_LIBRARY_PATH} "$ENV{NEUWARE_HOME}/lib64" ${CMAKE_INSTALL_PREFIX} PATHS ${ALTER_LD_LIBRARY_PATHS} "$ENV{NEUWARE_HOME}/lib64" ${CMAKE_INSTALL_PREFIX}
HINTS ${SYSTEM_LIBRARY_PATHS} HINTS ${ALTER_LIBRARY_PATHS}
PATH_SUFFIXES lib lib64 PATH_SUFFIXES lib lib64
DOC "CNML library." ) DOC "CNML library." )
......
if($ENV{LIBRARY_PATH})
string(REPLACE ":" ";" SYSTEM_LIBRARY_PATHS $ENV{LIBRARY_PATH})
endif()
find_library(CNRT_LIBRARY find_library(CNRT_LIBRARY
NAMES libcnrt.so NAMES libcnrt.so
PATHS $ENV{LD_LIBRARY_PATH} "$ENV{NEUWARE_HOME}/lib64" ${CMAKE_INSTALL_PREFIX} PATHS ${ALTER_LD_LIBRARY_PATHS} "$ENV{NEUWARE_HOME}/lib64" ${CMAKE_INSTALL_PREFIX}
HINTS ${SYSTEM_LIBRARY_PATHS} HINTS ${ALTER_LIBRARY_PATHS}
PATH_SUFFIXES lib lib64 PATH_SUFFIXES lib lib64
DOC "CNRT library." ) DOC "CNRT library." )
......
...@@ -3,10 +3,6 @@ if(${PkgConfig_FOUND}) ...@@ -3,10 +3,6 @@ if(${PkgConfig_FOUND})
pkg_check_modules(PC_CUDNN QUIET CUDNN) pkg_check_modules(PC_CUDNN QUIET CUDNN)
endif() endif()
if(NOT "$ENV{LIBRARY_PATH}" STREQUAL "")
string(REPLACE ":" ";" SYSTEM_LIBRARY_PATHS $ENV{LIBRARY_PATH})
endif()
if("${CUDNN_ROOT_DIR}" STREQUAL "" AND NOT "$ENV{CUDNN_ROOT_DIR}" STREQUAL "") if("${CUDNN_ROOT_DIR}" STREQUAL "" AND NOT "$ENV{CUDNN_ROOT_DIR}" STREQUAL "")
set(CUDNN_ROOT_DIR $ENV{CUDNN_ROOT_DIR}) set(CUDNN_ROOT_DIR $ENV{CUDNN_ROOT_DIR})
endif() endif()
...@@ -14,15 +10,15 @@ endif() ...@@ -14,15 +10,15 @@ endif()
if(MGE_CUDA_USE_STATIC AND NOT MGE_WITH_CUDNN_SHARED) if(MGE_CUDA_USE_STATIC AND NOT MGE_WITH_CUDNN_SHARED)
find_library(CUDNN_LIBRARY find_library(CUDNN_LIBRARY
NAMES libcudnn_static.a cudnn.lib NAMES libcudnn_static.a cudnn.lib
PATHS $ENV{LD_LIBRARY_PATH} ${CUDNN_ROOT_DIR} ${PC_CUDNN_LIBRARY_DIRS} ${CMAKE_INSTALL_PREFIX} PATHS ${ALTER_LD_LIBRARY_PATHS} ${CUDNN_ROOT_DIR} ${PC_CUDNN_LIBRARY_DIRS} ${CMAKE_INSTALL_PREFIX}
HINTS ${SYSTEM_LIBRARY_PATHS} HINTS ${ALTER_LIBRARY_PATHS}
PATH_SUFFIXES lib lib64 PATH_SUFFIXES lib lib64
DOC "CUDNN library." ) DOC "CUDNN library." )
else() else()
find_library(CUDNN_LIBRARY find_library(CUDNN_LIBRARY
NAMES libcudnn.so libcudnn.dylib cudnn64.dll NAMES libcudnn.so libcudnn.dylib cudnn64.dll
PATHS $ENV{LD_LIBRARY_PATH} ${CUDNN_ROOT_DIR} ${PC_CUDNN_LIBRARY_DIRS} ${CMAKE_INSTALL_PREFIX} PATHS ${ALTER_LD_LIBRARY_PATHS} ${CUDNN_ROOT_DIR} ${PC_CUDNN_LIBRARY_DIRS} ${CMAKE_INSTALL_PREFIX}
HINTS ${SYSTEM_LIBRARY_PATHS} HINTS ${ALTER_LIBRARY_PATHS}
PATH_SUFFIXES lib lib64 PATH_SUFFIXES lib lib64
DOC "CUDNN library." ) DOC "CUDNN library." )
endif() endif()
......
if(NOT "$ENV{LIBRARY_PATH}" STREQUAL "")
string(REPLACE ":" ";" SYSTEM_LIBRARY_PATHS $ENV{LIBRARY_PATH})
endif()
if("${TRT_ROOT_DIR}" STREQUAL "" AND NOT "$ENV{TRT_ROOT_DIR}" STREQUAL "") if("${TRT_ROOT_DIR}" STREQUAL "" AND NOT "$ENV{TRT_ROOT_DIR}" STREQUAL "")
set(TRT_ROOT_DIR $ENV{TRT_ROOT_DIR}) set(TRT_ROOT_DIR $ENV{TRT_ROOT_DIR})
endif() endif()
...@@ -9,15 +5,15 @@ endif() ...@@ -9,15 +5,15 @@ endif()
if(MGE_CUDA_USE_STATIC) if(MGE_CUDA_USE_STATIC)
find_library(TRT_LIBRARY find_library(TRT_LIBRARY
NAMES libnvinfer_static.a nvinfer.lib NAMES libnvinfer_static.a nvinfer.lib
PATHS $ENV{LD_LIBRARY_PATH} ${TRT_ROOT_DIR} ${CMAKE_INSTALL_PREFIX} PATHS ${ALTER_LD_LIBRARY_PATHS} ${TRT_ROOT_DIR} ${CMAKE_INSTALL_PREFIX}
HINTS ${SYSTEM_LIBRARY_PATHS} HINTS ${ALTER_LIBRARY_PATHS}
PATH_SUFFIXES lib lib64 PATH_SUFFIXES lib lib64
DOC "TRT library." ) DOC "TRT library." )
else() else()
find_library(TRT_LIBRARY find_library(TRT_LIBRARY
NAMES libnvinfer.so libnvinfer.dylib nvinfer.dll NAMES libnvinfer.so libnvinfer.dylib nvinfer.dll
PATHS $ENV{LD_LIBRARY_PATH} ${TRT_ROOT_DIR} ${CMAKE_INSTALL_PREFIX} PATHS ${ALTER_LD_LIBRARY_PATHS} ${TRT_ROOT_DIR} ${CMAKE_INSTALL_PREFIX}
HINTS ${SYSTEM_LIBRARY_PATHS} HINTS ${ALTER_LIBRARY_PATHS}
PATH_SUFFIXES lib lib64 PATH_SUFFIXES lib lib64
DOC "TRT library." ) DOC "TRT library." )
endif() endif()
......
...@@ -75,6 +75,10 @@ ...@@ -75,6 +75,10 @@
2: install gcc/g++, which version >= 6, (gcc/g++ >= 7, if need build training mode) 2: install gcc/g++, which version >= 6, (gcc/g++ >= 7, if need build training mode)
3: install build-essential git git-lfs gfortran libgfortran-6-dev autoconf gnupg flex bison gperf curl zlib1g-dev gcc-multilib g++-multilib lib32ncurses5-dev libxml2-utils xsltproc unzip libtool librdmacm-dev rdmacm-utils python3-dev python3-numpy texinfo 3: install build-essential git git-lfs gfortran libgfortran-6-dev autoconf gnupg flex bison gperf curl zlib1g-dev gcc-multilib g++-multilib lib32ncurses5-dev libxml2-utils xsltproc unzip libtool librdmacm-dev rdmacm-utils python3-dev python3-numpy texinfo
4: CUDA env(if enable CUDA), version detail refer to README.md 4: CUDA env(if enable CUDA), version detail refer to README.md
recommend set env about cuda/cudnn/tensorrt as followed:
export CUDA_ROOT_DIR=/path/to/cuda/lib
export CUDNN_ROOT_DIR=/path/to/cudnn/lib
export TRT_ROOT_DIR=/path/to/tensorrt/lib
``` ```
### MacOS host build ### MacOS host build
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册