Skip to content

Conversation

wyc-ruiker
Copy link
Contributor

Pading conv2d NWHC/HWNC ops to legal shapes for using tensorcore on cuda target and add some tests.
could you help review this pr? @jcf94 @Meteorix @jwfromm

Copy link
Contributor

@jcf94 jcf94 left a comment

Choose a reason for hiding this comment

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

Thanks! @wyc-ruiker This looks good to me.

@masahi masahi merged commit 5f94c1e into apache:main Jun 18, 2021
@wyc-ruiker wyc-ruiker deleted the fix-leg branch June 18, 2021 07:12
ylc pushed a commit to ylc/tvm that referenced this pull request Sep 29, 2021
* add conv2d leg

* minor fix

* fix pylint

* fix pylint

Co-authored-by: wangyucheng <wangyucheng@sensetime.com>
zxy844288792 pushed a commit to zxy844288792/tvm that referenced this pull request Mar 4, 2022
* add conv2d leg

* minor fix

* fix pylint

* fix pylint

Co-authored-by: wangyucheng <wangyucheng@sensetime.com>
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