TVM上手, module ‘tvm‘ has no attribute ‘var‘

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

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))