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

[ONNX] Fix test to disable default ONNX frontend constant folding #12532

Merged
merged 1 commit into from
Aug 22, 2022

Conversation

leandron
Copy link
Contributor

In TVM ONNX frontend, constants are folded by default, which makes test_load_model__onnx to fail because it is looking for "params" that were already converted into constants.

This patch fixes the test to disable constant folding so that we can assert that "params" in the model are present as expected.

cc @ashutosh-arm @driazati @areusch @Mousius @NicolaLancellotti

In TVM ONNX frontend, constants are folded by default, which makes
test_load_model__onnx to fail because it is looking for "params"
that were already converted into constants.

This patch fixes the test to disable constant folding so that we
can assert that "params" in the model are present as expected.

Change-Id: Ifa5aefa3bbafcd85e383e1292f57b479bdf28c8b
Copy link
Contributor

@NicolaLancellotti NicolaLancellotti left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

Copy link
Contributor

@lhutton1 lhutton1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @leandron, LGTM!

@lhutton1 lhutton1 merged commit 48a8cbd into apache:main Aug 22, 2022
xinetzone pushed a commit to daobook/tvm that referenced this pull request Nov 25, 2022
…ache#12532)

In TVM ONNX frontend, constants are folded by default, which makes `test_load_model__onnx` to fail because it is looking for "params" that were already converted into constants.

This patch fixes the test to disable constant folding so that we can assert that "params" in the model are present as expected.
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

Successfully merging this pull request may close these issues.

3 participants