美文网首页
TVM上手, module 'tvm' has no attri

TVM上手, module 'tvm' has no attri

作者: 全方位小白 | 来源:发表于2020-08-17 22:53 被阅读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,看到文档中的示例代码才知道。

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

    相关文章

      网友评论

          本文标题:TVM上手, module 'tvm' has no attri

          本文链接:https://www.haomeiwen.com/subject/gmaxjktx.html