按照官方文档装了一下,步骤还有点多。最后在跑简单的小程序的时候卡住了,遇到了报错:
Traceback (most recent call last):
File "tvmtest.py", line 47, in <module>
n = tvm.var("n")
AttributeError: module 'tvm' has no attribute 'var'
发现新版本的tvm里,var、placeholder等都移到了 tvm.te下,因此改为tvm.te.*即可:
import tvm
n = tvm.var("n")
A = tvm.te.placeholder((n,), name='A')
B = tvm.te.placeholder((n,), name='B')
C = tvm.te.compute(A.shape, lambda i: A[i] + B[i], name="C")
print(type(C))
在网上搜了好久,可能因为国内用tvm的人还是比较少,所以没搜到结论,最后还是去官方文档中直接搜placeholder,看到文档中的示例代码才知道。
所以也分享出来,在后人踩坑时能有个解决方法。
网友评论