psrivas2 opened a new issue, #14070: URL: https://github.com/apache/tvm/issues/14070
### This issue is to track progress for Layout Planning in Relax as per the proposal [here](https://github.com/tlc-pack/relax/issues/277) - [x] P1. Add layout_transform operator in Relax (https://github.com/tlc-pack/relax/pull/403) - [ ] P2. Add PrimFunc level analysis pass to suggest layout transform to blocks and buffers based on layout transforms on outputs (https://github.com/apache/tvm/pull/14066) - [ ] P3. Add AlterOpLayout pass that allows users to modify the layout of certain operators. - [ ] P4. Add OptimizeLayout pass that uses P1 & P2 to flow layouts across ops and cancel/fuse adjacent layout transforms. #### Future Possibilities - [ ] Extend P2 to handle implicit padding in output buffer transformations. - [ ] Using P2 to get operator level layout planning pass as discussed in comments in [proposal](https://github.com/tlc-pack/relax/issues/277) - [ ] Extend P2 to handle multiple write buffers in a block. - [ ] Extend P2 to handle operators such as `upsampling` whose buffer access are difficult to analyze. ### Triage Please refer to the list of label tags [here](https://github.com/apache/tvm/wiki/Issue-Triage-Labels) to find the relevant tags and add them below in a bullet format (example below). * branch:unity -- 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]
