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

[Relay][Frontend][Onnx] Enable group_conv1d import through conv2d conversion. #8321

Merged
merged 2 commits into from
Jun 24, 2021

Conversation

jwfromm
Copy link
Contributor

@jwfromm jwfromm commented Jun 24, 2021

TVM doesn't support group_conv1d currently and adding it formally is a non-trivial amount of work. Meanwhile, we're encountering an increasing number of models that use it. As an interim solution, we can import group_conv1d using group_conv2d by simply adding a new spatial axis with size 1. This PR adds that converter into the onnx frontend along with corresponding tests.

@jwfromm jwfromm requested a review from mbrookhart June 24, 2021 00:55
@jwfromm
Copy link
Contributor Author

jwfromm commented Jun 24, 2021

@mbrookhart @masahi can you guys take a look at this PR?

Copy link
Contributor

@mbrookhart mbrookhart left a comment

Choose a reason for hiding this comment

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

Nit, but needs to be changed. Otherwise looks good. I appreciate the TODO.

In general, and for what it's worth, I find the -k option in pytest easier than commenting out the test list 🤷‍♂️

tests/python/frontend/onnx/test_forward.py Outdated Show resolved Hide resolved
@mbrookhart mbrookhart merged commit 6b7b966 into apache:main Jun 24, 2021
ylc pushed a commit to ylc/tvm that referenced this pull request Sep 29, 2021
…version. (apache#8321)

* Enable group conv1d import through conv2d hack.

* remove silly commented out lines.
zxy844288792 pushed a commit to zxy844288792/tvm that referenced this pull request Mar 4, 2022
…version. (apache#8321)

* Enable group conv1d import through conv2d hack.

* remove silly commented out lines.
@jwfromm jwfromm deleted the group_conv1d_onnx branch April 12, 2023 15:55
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.

2 participants