pytorch/test/edge/templates
Mengwei Liu 0721a394b6 [executorch][kernel reg] Allow kernel manual registration (#110086)
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
2023-09-27 16:04:20 +00:00
..
Functions.h
NativeFunctions.h
RegisterCodegenUnboxedKernels.cpp Add support for event_tracer in codegen layer (#109990) 2023-09-27 09:09:03 +00:00
RegisterDispatchKeyCustomOps.cpp
RegisterKernels.h [executorch][kernel reg] Allow kernel manual registration (#110086) 2023-09-27 16:04:20 +00:00
RegisterSchema.cpp