Skip to content

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 🤷‍♂️

@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