tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-tvm] jwfromm opened a new pull request #5284: [Topi] Tensorcore support for Conv3D
Date Wed, 08 Apr 2020 23:06:42 GMT
jwfromm opened a new pull request #5284: [Topi] Tensorcore support for Conv3D
URL: https://github.com/apache/incubator-tvm/pull/5284
 
 
   This PR is a pretty direct port of the conv2d tensorcore schedules introduced in [PR #5099](https://github.com/apache/incubator-tvm/pull/5099)
to conv3d. In my early testing I've found this new schedule to be up to 10X faster than the
default conv3d schedule. I also snuck one little adjustment into the conv3d winograd schedule
that helps for smaller workloads.
   

----------------------------------------------------------------
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.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services

Mime
View raw message