Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Bug] [Relay] Error when compiling a simple ONNX model #7202

Closed
luyaor opened this issue Jan 4, 2021 · 1 comment
Closed

[Bug] [Relay] Error when compiling a simple ONNX model #7202

luyaor opened this issue Jan 4, 2021 · 1 comment
Assignees

Comments

@luyaor
Copy link
Contributor

luyaor commented Jan 4, 2021

Description

When compiling following model with TVM, it will error.

The model(with ONNX as frontend) with error is as follows, check bug.onnx in bug2.zip.

image

Error Log

The Relay type checker is unable to show the following types match.
In particular dimension 0 conflicts: 3 does not match 1.
The Relay type checker is unable to show the following types match.
In particular `Tensor[(3), float32]` does not match `Tensor[(1), float32]`
note: run with `TVM_BACKTRACE=1` environment variable to display a backtrace.

How to reproduce

Environment

Python3, with tvm, onnx, you could also use Conda as conda env create -f environment.yml

tvm version: c31e338 Wed Dec 9 14:52:58 2020 +0900

  1. Download bug2.zip

  2. Run python check.py.

@jwfromm
Copy link
Contributor

jwfromm commented Jan 5, 2021

Thanks for finding this bug, it's due to a bad assumption made in Prelu conversion about the input layout. The issue is fixed in #7208 and I've confirmed the test script works with that change.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants