mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
0721a394b6
Summary: Exposing a codegen mode for generating a hook for user to register their kernels. If we pass `--manual-registration` flag to `gen_executorch.py`, we will generate the following files: 1. RegisterKernels.h which declares a `register_all_kernels()` API inside `torch::executor` namespace. 2. RegisterKernelsEverything.cpp which implements `register_all_kernels()` by defining an array of generated kernels. This way user can depend on the library declared by `executorch_generated_lib` macro (with `manual_registration=True`) and be able to include `RegisterKernels.h`. Then they can manually call `register_all_kernels()` instead of relying on C++ static initialization mechanism which is not available in some embedded systems. Test Plan: Rely on the unit test: ``` buck2 test fbcode//executorch/runtime/kernel/test:test_kernel_manual_registration ``` Reviewed By: cccclai Differential Revision: D49439673 Pull Request resolved: https://github.com/pytorch/pytorch/pull/110086 Approved by: https://github.com/cccclai |
||
|---|---|---|
| .. | ||
| templates | ||
| CMakeLists.txt | ||
| custom_ops.cpp | ||
| custom_ops.yaml | ||
| Evalue.h | ||
| event_tracer_hooks.h | ||
| event_tracer.h | ||
| kernel_runtime_context.h | ||
| operator_registry.cpp | ||
| operator_registry.h | ||
| selected_operators.yaml | ||
| test_main.cpp | ||
| test_operator_registration.cpp | ||