coffezhou opened a new issue, #17991:
URL: https://github.com/apache/tvm/issues/17991
### Expected behavior
The onnx frontend should import the model correctly.
### Actual behavior
When importing the onnx model, tvm crashes as follows:
```c
Error converting operator BitwiseNot, with inputs: [R.astype(lv,
dtype="int32")]
Traceback (most recent call last):
File "/home/carla/Documents/test/test.py", line 36, in <module>
main()
File "/home/carla/Documents/test/test.py", line 30, in main
tvm_model = from_onnx(onnx_model, keep_params_in_input=True)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File
"/home/carla/Documents/tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py",
line 3696, in from_onnx
return g.from_onnx(graph, opset)
^^^^^^^^^^^^^^^^^^^^^^^^^
File
"/home/carla/Documents/tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py",
line 3327, in from_onnx
self._construct_nodes(graph)
File
"/home/carla/Documents/tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py",
line 3507, in _construct_nodes
raise err
File
"/home/carla/Documents/tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py",
line 3502, in _construct_nodes
op = self._convert_operator(op_name, inputs, attr, self.opset)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File
"/home/carla/Documents/tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py",
line 3602, in _convert_operator
sym = op_function(self.bb, inputs, attrs, [self._nodes, self._params])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File
"/home/carla/Documents/tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py",
line 575, in _impl_v18
return cls.base_impl(bb, inputs, attr, params)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File
"/home/carla/Documents/tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py",
line 531, in base_impl
return super().base_impl(bb, inputs, attr, params)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File
"/home/carla/Documents/tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py",
line 348, in base_impl
return cls.relax_op(inputs[0], inputs[1]) # pylint: disable=not-callable
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: bitwise_not() takes 1 positional argument but 2 were given
[16:11:28] /home/carla/Documents/tvm/src/relax/ir/block_builder.cc:64:
Warning: BlockBuilder destroyed with remaining blocks!
```
### Environment
OS: Ubuntu 20.04
TVM: 0.21.dev0 (3db71bb3a)
### Steps to reproduce
This bug can be reproduced by the following code with the model in the
attachment. As shown in the code, the model can be executed by onnxruntime.
```python
import sys
import numpy as np
import onnx
import onnxruntime
import tvm
from tvm import relax
from tvm.relax.frontend.onnx import from_onnx
import pickle
def main():
onnx_model = onnx.load("a569.onnx")
with open("inputs.pkl", "rb") as fp:
inputs = pickle.load(fp)
try:
ort_session = onnxruntime.InferenceSession(
onnx_model.SerializeToString(),
providers=["CPUExecutionProvider"]
)
ort_output = ort_session.run([], inputs)
except Exception as e:
print(e)
sys.exit(1)
# Convert the onnx model into relax through the onnx importer.
tvm_model = from_onnx(onnx_model, keep_params_in_input=True)
if __name__ == "__main__":
main()
```
[testcase.zip](https://github.com/user-attachments/files/20279963/testcase.zip)
### Triage
* needs-triage
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
