benchmark
abd8dd23 - Refactor convert_frame._compile_inner to return compiled bytecode + output graph. [3/n] (#160855)

Commit
120 days ago
Refactor convert_frame._compile_inner to return compiled bytecode + output graph. [3/n] (#160855) Summary: We are refactoring dynamo code for convert frame so that we can have modularized pieces sharable between different compiler frontends (e.g. torch.compile, precompile and torch.export). This PR adds a new helper function compile_frame() which takes a bytecode and a transform function and return compiled bytecode + output graph as DynamoOutput type. X-link: https://github.com/pytorch/pytorch/pull/160855 Approved by: https://github.com/tugsbayasgalan ghstack dependencies: #160814, #160815 Reviewed By: tugsbayasgalan Differential Revision: D80430802 fbshipit-source-id: b52f2c57be2cf8bc3a84ef747d5489814a988375
Author
Parents
Loading