-
Notifications
You must be signed in to change notification settings - Fork 518
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
[MLIR][TORCH] Add support for 1-d group convolution #3904
Conversation
fd98ffe
to
fd771f0
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
If you used most of the logic from squeeze.dim
and unsqueeze
lowerings to linalg for these helper functions, should we use these util functions there as well?
This commit adds the support for 1-d group convolution by transforming it into a 2-d group convolution which is already supported. This commit also refactors the unsqueeze and squeeze tensor utility. Signed-off-by: Vivek Khandelwal <[email protected]>
99943ef
to
0eeecdc
Compare
I have already made those changes and replaced the code. Could you please re-check? Or, let me know if I am missing something. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
A few nits, then LGTM.
This reverts commit 8e0eafd.
This commit adds the support for 1-d group convolution by transforming it into a 2-d group convolution which is already supported.
This commit also refactors the unsqueeze and squeeze tensor utility.