rzou f8fcc54f70 Add torch.library.impl_abstract (#109912) ... Changelog: - torch.library.impl_abstract optionally accepts a torch.library.Library object. If passed in, then the lifetime of the registration is tied to the Library object. - we've also changed torch.library.impl_abstract to work on all operators, including overloads. - we refactored the `torch._custom_ops.*` and `torch._custom_op.*` impl_abstract APIs and put them under torch._library. This is the final resting place for them. I will follow-up with deleting all the `torch._custom_ops.*` stuff later. - There is a new "SimpleOperatorRegistry" where we actually collect the abstract_impl. We will expand this to also hold the other torch._custom_ops.* APIs when we move those to torch.library NB: Previously we had designed `impl_abstract` assuming a very high-level Python-only custom op API. We've revisited that since; now, impl_abstract works for all custom ops, no matter python or C++, no matter the schema. The new refactored design reflects this better. Test Plan: - existing and new tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/109912 Approved by: https://github.com/ezyang		2023-09-26 01:59:50 +00:00
..
__init__.py
autograd.py	custom ops: don't error if autograd input is a tensor subclass (#109248)	2023-09-20 04:37:31 +00:00
functional.py	Add API to construct the functional variant of an op (#102293)	2023-06-02 13:36:50 +00:00
impl.py	Add torch.library.impl_abstract (#109912)	2023-09-26 01:59:50 +00:00