Skip to content

Commit 140fc57

Browse files
Update kernel-library-selective-build.md (#6121)
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 (cherry picked from commit 84498b2) Co-authored-by: Mengwei Liu <[email protected]>
1 parent 59c83f9 commit 140fc57

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)