diff --git a/CMakeLists.txt b/CMakeLists.txt index e1e457d31b52..c9daafa147e4 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -38,6 +38,12 @@ tvm_option(USE_RTTI "Build with RTTI" ON) tvm_option(USE_MSVC_MT "Build with MT" OFF) tvm_option(INSTALL_DEV "Install compiler infrastructure" OFF) +# 3rdparty libraries +tvm_option(DLPACK_PATH "Path to DLPACK" "3rdparty/dlpack/include") +tvm_option(DMLC_PATH "Path to DMLC" "3rdparty/dmlc-core/include") +tvm_option(RANG_PATH "Path to RANG" "3rdparty/rang/include") +tvm_option(COMPILER_RT_PATH "Path to COMPILER-RT" "3rdparty/compiler-rt") + # Contrib library options tvm_option(USE_BLAS "The blas library to be linked" none) tvm_option(USE_MKL_PATH "MKL root path when use MKL blas" none) @@ -51,11 +57,12 @@ tvm_option(USE_RANDOM "Build with random support" OFF) tvm_option(USE_ANTLR "Build with ANTLR for Relay parsing" OFF) # include directories +include_directories(${CMAKE_INCLUDE_PATH}) include_directories("include") -include_directories("3rdparty/dlpack/include") -include_directories("3rdparty/dmlc-core/include") -include_directories("3rdparty/rang/include") -include_directories("3rdparty/compiler-rt") +include_directories(${DLPACK_PATH}) +include_directories(${DMLC_PATH}) +include_directories(${RANG_PATH}) +include_directories(${COMPILER_RT_PATH}) # initial variables set(TVM_LINKER_LIBS "")