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: [email protected]
