p35189024
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add: CMake config

This commit is contained in:
Liyan Zheng 2023-04-09 17:17:10 +08:00
parent 69d894e003
commit 31b1edc600
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cmake/config_debugTVM.cmake Normal file
View File

@ -0,0 +1,13 @@
set(TVM_HOME "/home/zly/Apps/tvm-v0.10.0")
# CMake_BUILD_TYPE is not set
set(USE_CUDA ON)
set(USE_BANG OFF)
set(TVM_INCLUDE_DIR "${TVM_HOME}/include")
set(DMLC_INCLUDE_DIR "${TVM_HOME}/3rdparty/dmlc-core/include")
set(DLPACK_INCLUDE_DIR "${TVM_HOME}/3rdparty/dlpack/include")
set(BUILD_TEST ON)
set(BUILD_TEST_CORE ON)
set(BUILD_TEST_PET OFF)
set(BUILD_TEST_EINNET ON)