pytorch
83781af9 - Simplify aot_module_simplified by removing top_args/top_kwargs

Commit
2 years ago
Simplify aot_module_simplified by removing top_args/top_kwargs This makes good on Chillee's CR comment at https://github.com/pytorch/functorch/pull/660/files/af30d351cc93dfafb5a94dbcb32983c5ef65fd6a#r843315222 which was never done in the original PR. There is no logic change, just unpack the args/kwargs at the top level and remove the inner function indirection. Signed-off-by: Edward Z. Yang <ezyangfb.com> ghstack-source-id: 26883e7249d72df5c8866373989c4e39127ae2ee Pull Request resolved: https://github.com/pytorch/pytorch/pull/89666
Author
Parents
Loading