tqchen opened a new pull request #5372: [TIR][REFACTOR] Remove te::Tensor dependencies from TIR passes. URL: https://github.com/apache/incubator-tvm/pull/5372 This is needed to migrate passes before StorageFlatten to the PassManager. te::Tensor is an useful object for tensor expression. However, making it part of the TIR makes the abstraction less clean. This PR is a first step to remove this dependency. In particular, we will use Buffer in all the places where the te::Tensor is being used. We can not use IRModule of PrimFunc to represent TIR at any point of the optimizations. - The TIR passes will no longer need to consider Provide/HalideCall and Realize - Instead, we will use the BufferLoad/Store and BufferRealize. Buffer will serve as the only abstraction for the TIR data models to represent the intermediate storage among computations. We still keep Realize/HalideCall and Provide as TIR nodes for now to make the change minimum. Right after ScheduleOps, we call SchedulePostProcToPrimFunc to canonicalize the temporary IR generated by TE(which contains these nodes) to the TIR. The TIR optimizations are now mostly migrated to to the pass manager. With only a few to be migrated as subsequent PRs.
---------------------------------------------------------------- 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] With regards, Apache Git Services
