pytorch
0df239e5 - [FX] Make arg normalization a method on Node and not a pass (also augment tests to be exhaustive) (#55992)

Commit
3 years ago
[FX] Make arg normalization a method on Node and not a pass (also augment tests to be exhaustive) (#55992) Summary: Commandeered from https://github.com/pytorch/pytorch/pull/54563 Primary changes from first PR: 1. Refactored primary `normalize_function` logic into `operator_schemas.py` so that non-FX users can use it. 2. Refactored tests a bit, and added a path to call `normalize_function` directly. 3. Moved check for `boolean_dispatch` so that `torch.lu` also gets properly handled. Pull Request resolved: https://github.com/pytorch/pytorch/pull/55992 Reviewed By: mruberry Differential Revision: D27774396 Pulled By: Chillee fbshipit-source-id: 7f65632e1d608e4abd55aec5ccbfdc3f67f52b8e
Author
Parents
Loading