pytorch/torch/_library
rzou d0aad93249 Refactor can_auto_functionalize (#115134)
In preparation for the next PR up in the stack, which is going to update
"can_auto_functionalize" to support more operators than just ones that
return nothing. We are unable to auto-generate FakeTensor kernels for
operators that do not return nothing, but we are able to generate
functionalization kernels for operators that return something.

Test Plan:
Existing tests

Pull Request resolved: https://github.com/pytorch/pytorch/pull/115134
Approved by: https://github.com/bdhirsh
ghstack dependencies: #114955, #114956
2023-12-05 22:43:06 +00:00
..
__init__.py
abstract_impl.py [custom op] Use canonical API to constrain unbacked values (#108372) 2023-10-10 05:14:28 +00:00
simple_registry.py
utils.py Refactor can_auto_functionalize (#115134) 2023-12-05 22:43:06 +00:00