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]>
pytorch · Oct 10, 2024 · 140fc57 · 140fc57
1 parent 59c83f9
commit 140fc57
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/docs/source/kernel-library-selective-build.md b/docs/source/kernel-library-selective-build.md
@@ -36,11 +36,16 @@ The basic flow looks like this:
 
 ## APIs
 
-We expose build macros for CMake, to allow users specifying op info:
+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:
 
-[gen_selected_ops](https://github.com/pytorch/executorch/blob/main/build/Codegen.cmake#L12)
-
-Build macros take the following inputs:
+```
+gen_selected_ops(
+  LIB_NAME         # the name of the selective build operator library to be generated
+  OPS_SCHEMA_YAML  # path to a yaml file containing operators to be selected
+  ROOT_OPS         # comma separated operator names to be selected
+  INCLUDE_ALL_OPS  # boolean flag to include all operators
+)
+```
 
 
 ### Select all ops