Add option specify a list of ops

Summary: This is needed to generate `portable_ops_lib` with select number of ops for quick testing when we can't afford to link all the ops for size reasons. Arm Baremetal builds is one example.

Differential Revision: D49899554

fbshipit-source-id: 9d35e7bfd1c1975da640e62493fc744898ad6714

Author: digantdesai

CMakeLists.txt

@@ -104,13 +104,13 @@ option(BUILD_SELECTIVE_BUILD_TEST
 
 option(EXECUTORCH_BUILD_SIZE_TEST "Whether to build size test" OFF)
 
+# Option to register op list
+option(SELECT_OPS_LIST "Register the following list of ops" 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)
 endif()

kernels/portable/CMakeLists.txt

@@ -37,9 +37,19 @@ file(GLOB_RECURSE _portable_kernels__srcs
      "${CMAKE_CURRENT_SOURCE_DIR}/cpu/*.cpp")
 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 a filterlist is provided only generate wrappers for those
+# Else for all - this is the default behavior.
+if(SELECT_OPS_LIST)
+  message("Selecting only ${SELECT_OPS_LIST} op(s)!")
+  gen_selected_ops("" "${SELECT_OPS_LIST}" "")
+else()
+  # Generate C++ bindings to register kernels into both PyTorch (for AOT) and
+  # Executorch (for runtime). Here select all ops in functions.yaml
+  message("Selecting all ops")
+  gen_selected_ops("${CMAKE_CURRENT_LIST_DIR}/functions.yaml" "" "")
+endif()
+
 # 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}")