tvm

基本按照官方的说明来安装就可以了。

1. tvm后端编译

  • 拷贝make/config.mk文件到根目录
  • 下载预编译好的llvm，解压到某个文件夹
  • 修改LLVM_CONFIG=/path/to/your/llvm/bin/llvm-config，
  • 然后就可以用make进行编译了
    make -j4
  • 编译完成之后在lib文件下有libtvm.so,libtvm_runtime.so,libtvmtopi.so等动态库

2. tvm前端安装

在~/.bashrc下添加如下一句
export PYTHONPATH=/path/to/tvm/python:/path/to/tvm/topi/python:${PYTHONPATH}

3. 测试

python -c “import tvm;print(tvm.version)”

0.2.0

4. cmake编译后端

需要对CMakeLists.txt做2处修改

  1. 修改llvm为on：tvm_option(USE_LLVM “Build with LLVM” ON)
  2. 在其中添加一句set(youpath/llvm-5.0/lib/cmake/llvm)

然后就可以编译了

mkdir build && cd build
cmake ..
make -j4