MasterJH5574 opened a new pull request, #17754: URL: https://github.com/apache/tvm/pull/17754
This PR fixes a bug in the general reduction dlight rule, which happens when there is a trailing spatial block, and for the previous reduction blocks, the reduction axes are not on the back. In the case above, the loop orders of the reduction blocks and the trailing spatial block are inconsistent, while the dlight rule before this fix always treat the loop orders as consistent. As a result, though the function after applying the rule is numerically correct, it may require much extra shared memory use (in proportion to the size of spatial loops). And when the spatial dimensions are large, the required share memory size may exceed the device limit. This PR fixes this bug and adds a unit test. -- 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: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
