![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tvm
kh815
这个作者很懒,什么都没留下…
展开
-
tvm.build过程
代码示例import tvmfrom tvm import ten = 2A = te.placeholder((n,), name='A')B = te.placeholder((n,), name='B')C = te.compute(A.shape, lambda *i: A(*i) + B(*i), name='C')s = tvm.te.create_schedule(C.op)m = tvm.lower(s, [A, B, C], name="test_add")rt_mod原创 2021-04-09 22:35:10 · 548 阅读 · 0 评论 -
TVM中python访问c++函数机制
/data/TVM/tvm/python/tvm/_ffi/registry.py/data/TVM/tvm/python/tvm/_ffi/registry.py/data/TVM/tvm/python/tvm/_ffi/_ctypes/packed_func.py/data/TVM/tvm/python/tvm/_ffi/base.py查找libtvm.s0库的路径/data/TVM/tvm/python/tvm/_ffi/libinfo.py获取库的路径/data/TVM/tv原创 2021-04-09 16:22:12 · 246 阅读 · 0 评论 -
TVM源码阅读记录
参考:https://www.zhihu.com/column/c_1254058636869603328https://www.zhihu.com/column/c_1093448543469166592#define TVM_REGISTER_GLOBAL(OpName) \ TVM_STR_CONCAT(TVM_FUNC_REG_VAR_DEF, __COUNTER__) = ::tvm::runtime::Registry::Register(OpName) TVM_REGISTER原创 2021-04-09 10:31:11 · 385 阅读 · 0 评论 -
tvm安装
参考:http://tvm.apache.org/docs/install/from_source.html1.下载代码git clone --recursive https://github.com/apache/tvm tvm2.编译共享库依赖包安装:sudo apt-get updatesudo apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essent原创 2021-04-06 16:47:21 · 323 阅读 · 0 评论