编译安装源码
合入补丁文件后安装并构建TVM共享库。
- 进入“/path/to/TVM/apache-tvm-src-v0.9.0”目录。
1
cd /path/to/TVM/apache-tvm-src-v0.9.0/
- 创建一个构建目录“build”,将“cmake/config.cmake”复制到该目录。
1 2
mkdir build cp cmake/config.cmake build
- 编辑“build/config.cmake”自定义编译选项。
- 为了便于调试,确保使用通过以下选项启用嵌入式图形执行器和调试功能。
1 2
set(USE_GRAPH_EXECUTOR ON) set(USE_PROFILER ON)
- TVM需要LLVM用于CPU代码生成工具(Codegen),设置USE_LLVM为ON。
1
set(USE_LLVM ON)
- 为了便于调试,确保使用通过以下选项启用嵌入式图形执行器和调试功能。
- 编译CMakeList.txt自定义编译选项,在第605行后增加以下内容。
1
target_link_libraries(tvm PRIVATE /path/to/BiShengCompiler/BiShengCompiler-3.2.0.1-aarch64-linux/lib/libtf_xla_runtime.a)
- 构建TVM及相关库。
1 2 3
cd build cmake -DCMAKE_PREFIX_PATH=/path/to/BiShengCompiler/BiShengCompiler-3.2.0.1-aarch64-linux .. make