MasterJH5574 opened a new pull request, #15383: URL: https://github.com/apache/tvm/pull/15383
Prior to this PR, the reduction dlight rule sometimes may over-normalizes the last block and removes some spatial loops, making the last block have fewer block iters than previous reduction blocks. In this case, the dlight rule is not able to apply due to assertion failure while it is supposed to apply. This PR thus adds unit block iter to the last block when it has fewer iters than other reduction blocks in the front, so that the numbers of block iters can align. One softmax unit test is added to ensure that dlight reduction rule can work. -- 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]
