TVM上手, module 'tvm' has no attri

2020-08-17  本文已影响0人  全方位小白

按照官方文档装了一下,步骤还有点多。最后在跑简单的小程序的时候卡住了,遇到了报错:

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,看到文档中的示例代码才知道。

所以也分享出来,在后人踩坑时能有个解决方法。

上一篇下一篇

猜你喜欢

热点阅读