如何编译TVM工程

如何编译TVM工程

TVM官网:[link]https://tvm.ai/
TVM官网编译帮助:[link]https://docs.tvm.ai/install/from_source.html#build-the-shared-library

下载源代码

// download tvm source code
git clone --recursive https://github.com/dmlc/tvm 		//  注意增加recursive参数
// download google gtest
cd tvm
TVM_ROOT=`pwd`
CACHE_PREFIX=. make -f 3rdparty/dmlc-core/scripts/packages.mk gtest

编译TVM工程和测试用例

// 修改LLVM配置信息,修改cmake/config.cmake文件
#set(USE_LLVM OFF)
set(USE_LLVM /path/to/llvm/llvm/build/bin/llvm-config)	// path是LLVM的路径
--------------------------------------------------------------------------------------------
mkdir build; cd build
cp cmake/config.cmake .
GTEST_LIB=$TVM_ROOT/lib cmake ..
make -j 40 						// build tvm
make cpptest -j 40     			// build cpptest

生成的TVM库文件和cpptest相关的可执行程序在tvm/build目录

猜你喜欢

转载自blog.csdn.net/weixin_31614117/article/details/87893119