jcf94 opened a new pull request #6750: URL: https://github.com/apache/incubator-tvm/pull/6750
In Ansor, we have an optimization called "Layout rewrite", which modifies the input weight of a specific op according to its schedule to get better performance, a previous PR about this is #6297 . This PR brings another option for this feature, besides directly modify the input placeholder, we now support to insert a transpose stage between the placeholder and compute op. Others: - Add a CopyOnWrite methods for transform_steps - Some small fix cc @merrymercy @comaniac @minminsun ---------------------------------------------------------------- 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]
