Open makaveli10 opened 1 year ago
I can try to fix this if you could provide a small test case to reproduce this issue locally. Perhaps a small onnx model that contains just the subgraph including the gather op?
P.S. I've found the onnx-modifier tool to be very useful to create small subgraphs for test cases in onnx. Or if you could share the whole model, that should also be fine.
Environment
Ubuntu 20.04, onnx opset version 11, tvm version 0.13.dev0
Steps to reproduce
tvmc tune --target "cuda" --output autotune.json encoder_model.onnx
For the gather op I see a segmentation fault in
relay.frontend.from_onnx
on line. Which callsfold_constant()
on this nodeless(0i64 /* span=/layers.0/self_attn/Gather./layers.0/self_attn/Constant_output_0:0:0 */, 0i64)
:there are two inputs to Gather:
cc @Hzfengsy @junrushao @shingjan