From a3c9bc7a6250a2c5f0f40e1fa3d97bfc8d1a9cd3 Mon Sep 17 00:00:00 2001 From: Mengwei Liu Date: Fri, 4 Oct 2024 14:36:14 -0700 Subject: [PATCH] Update kernel-library-selective-build.md Add more details to selective build API --- docs/source/kernel-library-selective-build.md | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/docs/source/kernel-library-selective-build.md b/docs/source/kernel-library-selective-build.md index 68d999bbc00..1a7562942e0 100644 --- 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