[Relay] NameError: name '_OpSetAttr' is not defined


#1

Hi Experts,

I try to run onnx/test_forward.py with the latest master branch but get the following error and I check the environment variable has configured correctly. I would appreciate if you can point me how to solve it, thanks.

Environment variable:

export TVM_HOME=/path/to/tvm
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:$TVM_HOME/nnvm/python:${PYTHONPATH}

Error Message:

python3 /Users/root/tvm/tests/python/frontend/onnx/test_forward.py
Traceback (most recent call last):

  File "/Users/root/tvm/tests/python/frontend/onnx/test_forward.py", line 22, in <module>
    import topi

  File "/Users/root/tvm/topi/python/topi/__init__.py", line 28, in <module>
    from . import arm_cpu

  File "/Users/root/tvm/topi/python/topi/arm_cpu/__init__.py", line 3, in <module>
    from . import conv2d

  File "/Users/root/tvm/topi/python/topi/arm_cpu/conv2d.py", line 25, in <module>
    from tvm import relay

  File "/Users/root/tvm/python/tvm/relay/__init__.py", line 24, in <module>
    from . import expr_functor

  File "/Users/root/tvm/python/tvm/relay/expr_functor.py", line 24, in <module>
    from .op import Op

  File "/Users/root/tvm/python/tvm/relay/op/__init__.py", line 38, in <module>
    from . import _tensor

  File "/Users/root/tvm/python/tvm/relay/op/_tensor.py", line 141, in <module>
    register_shape_func("add", False, broadcast_shape_func)

  File "/Users/root/tvm/python/tvm/relay/op/op.py", line 256, in register_shape_func
    get(op_name).set_attr("TShapeDataDependant", data_dependant, level)

  File "/Users/root/tvm/python/tvm/relay/op/op.py", line 65, in set_attr
    _OpSetAttr(self, attr_name, value, plevel)

NameError: name '_OpSetAttr' is not defined

#2

Hi all,

I solve this issue by rebuild tvm since there is a new PR merged into master

Best,