pytorch/torch/_export
Brian Hirsh 98ece75043 [aot autograd] merge all outputs of funtionalization analysis into single metadata (#95991)
This makes the next PR in the stack cleaner: having the top level entry point to aot autograd perform the functionalization analysis pass once, and plumb the metadata everywhere else that we need it.

I put it in a separate PR because I recently learned that this function is used in fbcode, so I'll need to fix up internals when I land this PR.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/95991
Approved by: https://github.com/ezyang
2023-03-08 16:22:54 +00:00
..
__init__.py [aot autograd] merge all outputs of funtionalization analysis into single metadata (#95991) 2023-03-08 16:22:54 +00:00
logical_schema.py
workflow.py [export] Add a data type for representing export workflow information. (#95013) 2023-02-17 16:28:17 +00:00