forked from jiuyuan/InfiniTensor
14 lines
378 B
CMake
14 lines
378 B
CMake
|
set(TVM_HOME "/home/zly/Apps/tvm-v0.10.0")
|
||
|
|
set(TVM_INCLUDE_DIR "${TVM_HOME}/include")
|
||
|
|
set(TVM_LIB_DIR "${TVM_HOME}/build")
|
||
|
|
set(DMLC_INCLUDE_DIR "${TVM_HOME}/3rdparty/dmlc-core/include")
|
||
|
|
set(DLPACK_INCLUDE_DIR "${TVM_HOME}/3rdparty/dlpack/include")
|
||
|
|
|
||
|
|
set(USE_CUDA ON)
|
||
|
|
set(USE_BANG OFF)
|
||
|
|
|
||
|
|
set(BUILD_TEST ON)
|
||
|
|
set(BUILD_TEST_CORE ON)
|
||
|
|
set(BUILD_TEST_PET OFF)
|
||
|
|
set(BUILD_TEST_EINNET ON)
|