I want to know if there are any benefits to using torch.export.exported_program
instead of torch.fx.Graph
when building an IRModule in TVM. Additionally, I noticed that from_fx
supports custom mappings, whereas from_imported_program
does not.
Are there specific advantages of using exported_program
over fx.Graph
? Also, in what situations would I need to use custom mappings? Are they necessary in most cases, or only for specific scenarios?
Any insights would be appreciated!