Richard Zou d1c092ae1b Update impl_abstract_pystub to be less boilerplatey (#113182) ... Summary: We've made the following changes: - The new way to use the API is `m.impl_abstract_pystub(module, context)`. Every subsequent m.def of an op inside the TORCH_LIBRARY block gives the op the `impl_abstract_pystub`. - Added a mechanism to determine if an operator was defined in Python or C++. Library.define in Python appends the op to a global set, which is analogous to what we do for tracking Library.impl. - If someone does `torch.library.impl_abstract` in Python for an operator, then we require that it has an `impl_abstract_pystub` specified and we also check that the module in the `impl_abstract_pystub` is the same as the module where the call to `torch.library.impl_abstract` exists. - Unfortunately we can't check the "context" (which is the buck target on buck-based systems) because buck sits above us. bypass-github-export-checks Test Plan: - existing tests Differential Revision: D51080493 Pull Request resolved: https://github.com/pytorch/pytorch/pull/113182 Approved by: https://github.com/ezyang		2023-11-08 00:39:00 +00:00
..
CMakeLists.txt
model.py	[BE] Enable ruff's UP rules and autoformat test/ (#105434)	2023-07-19 20:36:06 +00:00
my_custom_ops.py	Update impl_abstract_pystub to be less boilerplatey (#113182)	2023-11-08 00:39:00 +00:00
my_custom_ops2.py	Update impl_abstract_pystub to be less boilerplatey (#113182)	2023-11-08 00:39:00 +00:00
op.cpp	Update impl_abstract_pystub to be less boilerplatey (#113182)	2023-11-08 00:39:00 +00:00
op.h
pointwise.py	Update impl_abstract_pystub to be less boilerplatey (#113182)	2023-11-08 00:39:00 +00:00
test_custom_ops.cpp
test_custom_ops.py	Update impl_abstract_pystub to be less boilerplatey (#113182)	2023-11-08 00:39:00 +00:00