测试文件

测试文件位于github仓库中，测试目录组织说明：

• def_func 目录：

  该目录作用为观察 nnvm 中的 tvm 的函数自定义行为，需要将函数写入到 nnvm/tvm/runtime/c_runtime_api.cc 文件中，然后再重新编译 tvm 才能使新注册的函数成功被 python 调用，具体的操作在 def_func.cc 中注释说明

  • def_func.cc: 一个注册函数 myadd 的例子，这个文件只是示例，具体操作参见这个文件的注释。
  • def_func.py: 当注册好重新编译完成后，调用这个 python 文件即可使用新注册的函数。

• nnvm-llvm 目录：

  该目录作用为观察 nnvm 和 llvm 之间的代码转换关系。

• tvm 目录:

  该目录作为观察 tvm 针对 cpu，gpu 所做的优化情况。

  • optimizer-cpu.py 用于观察 tvm 对矩阵乘法所做优化的情况
  • optimizer-gpu.py 用于观察 tvm 对卷积算法所做优化的情况