You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2022/01/04 17:52:31 UTC
[GitHub] [tvm] tkonolige opened a new pull request #9832: [TOPI] Support grouped conv1d
tkonolige opened a new pull request #9832:
URL: https://github.com/apache/tvm/pull/9832
Generalize the conv2d compute statement to a generic convNd that supports any layout and groups. Replace some existing conv2d and conv1d compute statements with this generic compute. Also add a topi group_conv1d compute that uses the generic convNd compute. Existing schedules for conv1d work with group_conv1d, so they are reused.
I've tested this on an internal model and it provides a good speedup over the previous expand -> conv2d -> squeeze hack.
@mbrookhart
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [tvm] mbrookhart merged pull request #9832: [TOPI] Support grouped conv1d
Posted by GitBox <gi...@apache.org>.
mbrookhart merged pull request #9832:
URL: https://github.com/apache/tvm/pull/9832
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [tvm] mbrookhart commented on pull request #9832: [TOPI] Support grouped conv1d
Posted by GitBox <gi...@apache.org>.
mbrookhart commented on pull request #9832:
URL: https://github.com/apache/tvm/pull/9832#issuecomment-1007818008
Thanks, @tkonolige !
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [tvm] mbrookhart commented on pull request #9832: [TOPI] Support grouped conv1d
Posted by GitBox <gi...@apache.org>.
mbrookhart commented on pull request #9832:
URL: https://github.com/apache/tvm/pull/9832#issuecomment-1007818008
Thanks, @tkonolige !
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [tvm] mbrookhart merged pull request #9832: [TOPI] Support grouped conv1d
Posted by GitBox <gi...@apache.org>.
mbrookhart merged pull request #9832:
URL: https://github.com/apache/tvm/pull/9832
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org