2 Commits

Author	SHA1	Message	Date
Larry Liu	7568484d54	[torchgen] Add CI job to cover custom ops registration for Executorch (#91291) ... As titled. To register a custom op into Executorch, we need: * `custom_ops.yaml`, defines the operator schema and the corresponding native function. * `custom_ops.cpp`, defines the kernel. * `RegisterDispatchKeyCustomOps.cpp`, a template to register operator into PyTorch. Added a new test for custom ops. The custom op `custom::add_3.out` takes 3 tensors and add them together. The test makes sure it is registered correctly and then verifies the outcome is correct. Differential Revision: [D42204263](https://our.internmc.facebook.com/intern/diff/D42204263/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/91291 Approved by: https://github.com/ezyang	2023-01-14 02:30:54 +00:00
Larry Liu	909b7ca92a	[torchgen] Move Executorch codegen logic into torchgen (#90806) ... ## Codegen entry point Main logic and Executorch codegen entry: `gen_executorch.py`. `RegisterCodegenUnboxedKernels.cpp`: ```cpp register_operators({ Operator( "aten::add.out", [](EValue** stack) { EValue& self = *stack[0]; EValue& other = *stack[1]; EValue& alpha = *stack[2]; EValue& out = *stack[3]; const at::Tensor & self_base = self.to<at::Tensor>(); const at::Tensor & other_base = other.to<at::Tensor>(); const at::Scalar & alpha_base = alpha.to<at::Scalar>(); at::Tensor & out_base = out.to<at::Tensor>(); EXECUTORCH_SCOPE_PROF("native_call_add.out"); torch::executor::aten::add_outf(self_base, other_base, alpha_base, out_base); }) ); ``` `Functions.h`: ```cpp namespace torch { namespace executor { namespace aten { // aten::add_outf(Tensor self, Tensor other, Scalar alpha, *, Tensor(a!) out) -> Tensor(a!) TORCH_API inline at::Tensor & add_outf(const at::Tensor & self, const at::Tensor & other, at::Scalar alpha, at::Tensor & out) { return at::add_outf(self, other, alpha, out); } } // namespace aten } // namespace executor } // namespace torch ``` * Unit tests: `test_executorch_gen.py` CI job in next PR. Pull Request resolved: https://github.com/pytorch/pytorch/pull/90806 Approved by: https://github.com/ezyang	2022-12-19 21:58:43 +00:00