[GitHub] [tvm] jwfromm opened a new pull request #7957: [Relay][Autoscheduler] Fix autoscheduler matmul without units.

[GitBox <gi...@apache.org>]

jwfromm opened a new pull request #7957:
URL: https://github.com/apache/tvm/pull/7957


   When autoscheduling a dense (or matmul) operator, sometimes the layout will be rewritten to have more than two axes. The shape function in relay only deals with this properly when the `units` attribute is provided. However, some of our frontends do not specify units when constructing a graph, resulting in compilation errors after autoscheduling. This PR adds special handling for rewritten dense ops to fix the problem.


-- 
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

[GitHub] [tvm] jwfromm commented on pull request #7957: [Relay][Autoscheduler] Fix autoscheduler matmul without units.

[GitBox <gi...@apache.org>]

jwfromm commented on pull request #7957:
URL: https://github.com/apache/tvm/pull/7957#issuecomment-830451135


   @mbrookhart @merrymercy can you guys take a look at this fix?


-- 
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

[GitHub] [tvm] jwfromm merged pull request #7957: [Relay][Autoscheduler] Fix autoscheduler matmul without units.

[GitBox <gi...@apache.org>]

jwfromm merged pull request #7957:
URL: https://github.com/apache/tvm/pull/7957


   


-- 
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