chunit-quic commented on PR #16531: URL: https://github.com/apache/tvm/pull/16531#issuecomment-1931031334
Hi community, This is a PoC to support ExportedProgram, which is a new format in pytorch 2.0. May we know do you have any plan to support ExportedProgram? Or maybe this PR look fine to be a start point? :D ExportedProgram is a new format in pytorch2.0. After exporting to ExportedProgram, a model represented by Module or functional will be flatten and represented by core aten op. One of its benefits is that, we can focus on writing convert function for those core aten ops. It can decrease the effort to support different Module and functional. We are wondering do you have any plan to support this format as well? if there is no such plan, perhaps this PR can be a simple reference to support ExportedProgram conversion. If it looks good to you, we can submit a formal one based on your comments, and add adding this to converter. It will be really nice to have your thoughts. Thanks! ref: https://pytorch.org/executorch/stable/tutorials/export-to-executorch-tutorial.html https://pytorch.org/docs/stable/export.html#exporting-a-pytorch-model https://pytorch.org/docs/stable/torch.compiler_ir.html Best, Joey -- 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]
