Skip to content

Commit 84498b2

Browse files
larryliu0820facebook-github-bot
authored andcommitted
Update kernel-library-selective-build.md (#5895)
Summary: Add more details to selective build API Pull Request resolved: #5895 Reviewed By: helunwencser Differential Revision: D63920010 Pulled By: larryliu0820 fbshipit-source-id: 39cce24dc26d86d27165875dc05a1d1d6154613b
1 parent acfcdd5 commit 84498b2

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

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

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,11 +36,16 @@ The basic flow looks like this:
3636

3737
## APIs
3838

39-
We expose build macros for CMake, to allow users specifying op info:
39+
We expose a CMake macro `[gen_selected_ops](https://github.com/pytorch/executorch/blob/main/build/Codegen.cmake#L12)`, to allow users specifying op info:
4040

41-
[gen_selected_ops](https://github.com/pytorch/executorch/blob/main/build/Codegen.cmake#L12)
42-
43-
Build macros take the following inputs:
41+
```
42+
gen_selected_ops(
43+
LIB_NAME # the name of the selective build operator library to be generated
44+
OPS_SCHEMA_YAML # path to a yaml file containing operators to be selected
45+
ROOT_OPS # comma separated operator names to be selected
46+
INCLUDE_ALL_OPS # boolean flag to include all operators
47+
)
48+
```
4449

4550

4651
### Select all ops

0 commit comments

Comments
 (0)