Properly introduce selective build options

Summary:
Refactor options to do selective build.
* `EXECUTORCH_SELECT_ALL_OPS` by default `ON`, will be set to `OFF` if any of the other options is on.
* `EXECUTORCH_SELECT_OPS_LIST` by default `OFF`
* `EXECUTORCH_SELECT_OPS_YAML` by default `OFF`.

This allows these options to be used outside of selective build examples.

Differential Revision: D49935141

Author: larryliu0820

CMakeLists.txt

@@ -94,15 +94,19 @@ option(BUILD_SELECTIVE_BUILD_TEST
 
 option(EXECUTORCH_BUILD_SIZE_TEST "Whether to build size test" OFF)
 
-if(BUILD_SELECTIVE_BUILD_TEST)
-  option(SELECT_ALL_OPS
-         "Whether to register all ops defined in portable kernel library." OFF)
-
-  # Option to register op list
-  option(SELECT_OPS_LIST "Register the following list of ops" OFF)
-
-  # Option to register ops from yaml file
-  option(SELECT_OPS_YAML "Register all the ops from a given yaml file" OFF)
+# Selective build options.
+option(EXECUTORCH_SELECT_ALL_OPS
+      "Whether to register all ops defined in portable kernel library." ON)
+
+# Option to register op list
+option(EXECUTORCH_SELECT_OPS_LIST "Register the following list of ops" OFF)
+
+# Option to register ops from yaml file
+option(EXECUTORCH_SELECT_OPS_YAML
+      "Register all the ops from a given yaml file" OFF)
+# Do not enable select all ops if any of the other select options is on.
+if(EXECUTORCH_SELECT_OPS_LIST OR EXECUTORCH_SELECT_OPS_YAML)
+  set(EXECUTORCH_SELECT_ALL_OPS OFF)
 endif()
 
 # Build xnn_executor_runner which depends on XNNPACK

examples/selective_build/CMakeLists.txt

@@ -27,13 +27,7 @@ include(${EXECUTORCH_ROOT}/build/Codegen.cmake)
 # library
 #
 set(_kernel_lib)
-if(SELECT_ALL_OPS)
-  gen_selected_ops("" "" "${SELECT_ALL_OPS}")
-  list(APPEND _kernel_lib portable_kernels)
-elseif(SELECT_OPS_LIST)
-  gen_selected_ops("" "${SELECT_OPS_LIST}" "")
-  list(APPEND _kernel_lib portable_kernels)
-elseif(SELECT_OPS_YAML)
+if(EXECUTORCH_SELECT_OPS_YAML)
   set(_custom_ops_yaml ${EXECUTORCH_ROOT}/examples/custom_ops/custom_ops.yaml)
   gen_selected_ops("${_custom_ops_yaml}" "" "")
   set(kernel_sources
@@ -48,6 +42,13 @@ elseif(SELECT_OPS_YAML)
 
   list(APPEND _kernel_lib custom_kernels)
 endif()
+
+gen_selected_ops(
+  "${_custom_ops_yaml}"
+  "${EXECUTORCH_SELECT_OPS_LIST}"
+  "${EXECUTORCH_SELECT_ALL_OPS}")
+list(APPEND _kernel_lib portable_kernels)
+
 generate_bindings_for_kernels(${EXECUTORCH_ROOT}/kernels/portable/functions.yaml
                               "${_custom_ops_yaml}")
 gen_operators_lib("select_build_lib" ${_kernel_lib} executorch)

kernels/portable/CMakeLists.txt

@@ -39,7 +39,11 @@ list(FILTER _portable_kernels__srcs EXCLUDE REGEX "test/*.cpp")
 list(FILTER _portable_kernels__srcs EXCLUDE REGEX "codegen")
 # Generate C++ bindings to register kernels into both PyTorch (for AOT) and
 # Executorch (for runtime). Here select all ops in functions.yaml
-gen_selected_ops("${CMAKE_CURRENT_LIST_DIR}/functions.yaml" "" "")
+if(EXECUTORCH_SELECT_OPS_YAML)
+  set(_yaml_file "${CMAKE_CURRENT_LIST_DIR}/functions.yaml")
+endif()
+gen_selected_ops(
+  "${_yaml_file}" "${EXECUTORCH_SELECT_OPS_LIST}" "${EXECUTORCH_SELECT_ALL_OPS}")
 # Expect gen_selected_ops output file to be selected_operators.yaml
 generate_bindings_for_kernels(${CMAKE_CURRENT_SOURCE_DIR}/functions.yaml "")
 message("Generated files ${gen_command_sources}")