pytorch
4b9d79c0 - Don't support kwargs at runtime in aot_module_simplified

Commit
2 years ago
Don't support kwargs at runtime in aot_module_simplified The preexisting logic here added in https://github.com/pytorch/functorch/pull/970 was very peculiar: if top_kwargs was non-empty, then the inner compiled function supports kwargs. Naively, this would leave you to expect that there is some sort of correlation between top_kwargs and kwargs. But in fact, they're completely unrelated! top_kwargs is the AOTAutograd configuration knobs (e.g., fw_compiler/bw_compiler), but kwargs is the RUNTIME kwargs that are to be passed to the compiled function. But (1) we don't support this (the function to be compiled only takes a list of tensors) and (2) even if we did support it, conditioning on whether or not you had passed AOTAutograd configuration kwargs to support kwargs at runtime is bonkers. So delete it. Signed-off-by: Edward Z. Yang <ezyangfb.com> ghstack-source-id: 0ffdf43cf86642d15f4e8ec3c30bc675fe8e2bec Pull Request resolved: https://github.com/pytorch/pytorch/pull/89664
Author
Parents
Loading