Skip to content

Commit c5cfad7

Browse files
BujSetTanish2101
authored andcommitted
Updating documentation for cmake dtype selective build (pytorch#12112)
Updating documentation for cmake dtype selective build with model API. Table at `selective_build_model_doc/docs/source/kernel-library-selective-build.md` now looks like: ![image](https://github.com/user-attachments/assets/68b7c8fe-dfc5-4ae4-9756-5599680357bf) Fixes pytorch#12065 cc @mergennachin @byjlw
1 parent 51a1a50 commit c5cfad7

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

docs/source/kernel-library-selective-build.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,4 +125,4 @@ register_custom_kernels(); // Registers only the "custom" kernels
125125
This avoids relying on static initialization and enables you to register only the kernels you want.
126126

127127
### Compatibility
128-
If `--lib-name` is not passed, the default behavior remains unchanged, the codegen script will generate a general `RegisterKernels.cpp` and `register_all_kernels()` function.
128+
If `--lib-name` is not passed, the default behavior remains unchanged, the codegen script will generate a general `RegisterKernels.cpp` and `register_all_kernels()` function.

examples/selective_build/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,12 @@ cd executorch
1212
bash examples/selective_build/test_selective_build.sh cmake
1313
```
1414

15-
Check out `CMakeLists.txt` for demo of 3 selective build APIs:
15+
Check out `CMakeLists.txt` for demo of selective build APIs:
1616
1. `SELECT_ALL_OPS`: Select all ops from the dependency kernel libraries, register all of them into ExecuTorch runtime.
1717
2. `SELECT_OPS_LIST`: Only select operators from a list.
1818
3. `SELECT_OPS_YAML`: Only select operators from a yaml file.
19+
4. `SELECT_OPS_FROM_MODEL`: Only select operators from a from an exported model pte.
20+
5. `DTYPE_SELECTIVE_BUILD`: Enable rebuild of `portable_kernels` to use dtype selection. Currently only supported for `SELECTED_OPS_FROM_MODEL` API and `portable_kernels` lib.
1921

2022
Other configs:
2123
- `MAX_KERNEL_NUM=N`: Only allocate memory for N operators.
22-
23-
We have one more API incoming: only select from an exported model file (.pte).

0 commit comments

Comments
 (0)