Update

[ghstack-poisoned]

Author: GregoryComer

CMakeLists.txt

@@ -354,6 +354,9 @@ add_library(executorch_no_prim_ops ALIAS executorch_core)
 # A list of all configured backends.
 set(_executorch_backends "")
 
+# A list of all configured extensions.
+set(_executorch_extensions "")
+
 target_link_libraries(executorch_core PRIVATE program_schema)
 if(ANDROID)
   target_link_libraries(executorch_core PUBLIC log)
@@ -549,14 +552,17 @@ endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_APPLE)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/apple)
+  list(APPEND _executorch_extensions apple_extension)
 endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_DATA_LOADER)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/data_loader)
+  list(APPEND _executorch_extensions extension_data_loader)
 endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_FLAT_TENSOR)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/flat_tensor)
+  list(APPEND _executorch_extensions extension_flat_tensor)
 endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_MODULE)
@@ -567,26 +573,32 @@ if(EXECUTORCH_BUILD_EXTENSION_MODULE)
     FILES_MATCHING
     PATTERN "*.h"
   )
+list(APPEND _executorch_extensions extension_module_static)
 endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_LLM)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/llm/tokenizers)
+  list(APPEND _executorch_extensions tokenizers)
 endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_LLM_APPLE)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/llm/apple)
+  list(APPEND _executorch_extensions extension_llm_apple)
 endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_LLM_RUNNER)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/llm/runner)
+  list(APPEND _executorch_extensions extension_llm_runner)
 endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_RUNNER_UTIL)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/runner_util)
+  list(APPEND _executorch_extensions extension_runner_util)
 endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_TENSOR)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/tensor)
+  list(APPEND _executorch_extensions extension_tensor)
 endif()
 
 if(EXECUTORCH_BUILD_PTHREADPOOL AND EXECUTORCH_BUILD_CPUINFO)
@@ -688,6 +700,7 @@ endif()
 
 if(EXECUTORCH_BUILD_EXTENSION_TRAINING)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/training)
+  list(APPEND _executorch_extensions extension_training)
 endif()
 
 if(EXECUTORCH_BUILD_KERNELS_LLM)
@@ -706,11 +719,15 @@ if(EXECUTORCH_BUILD_VULKAN)
 endif()
 
 # Top-level interface targets.
-add_library(executorch_backends INTERFACE)
 
 # A target containing all configured backends.
+add_library(executorch_backends INTERFACE)
 target_link_libraries(executorch_backends INTERFACE ${_executorch_backends})
 
+# A target containing all configured extensions.
+add_library(executorch_extensions INTERFACE)
+target_link_libraries(executorch_extensions INTERFACE ${_executorch_extensions})
+
 if(EXECUTORCH_BUILD_EXECUTOR_RUNNER)
   # Baseline libraries that executor_runner will link against.
   set(_executor_runner_libs executorch gflags)

tools/cmake/executorch-config.cmake

@@ -73,20 +73,23 @@ set(backend_lib_list
     vulkan_backend
 )
 
-set(lib_list
-    flatccrt
-    etdump
-    bundled_program
+set(extension_lib_list
     extension_data_loader
     extension_flat_tensor
-    portable_ops_lib
-    custom_ops
     extension_module
     extension_module_static
     extension_runner_util
     extension_tensor
     extension_threadpool
     extension_training
+)
+
+set(lib_list
+    flatccrt
+    etdump
+    bundled_program
+    portable_ops_lib
+    custom_ops
     cpuinfo
     pthreadpool
     optimized_kernels
@@ -100,7 +103,7 @@ set(lib_list
     quantized_ops_aot_lib
 )
 
-list(APPEND lib_list ${backend_lib_list})
+list(APPEND lib_list ${backend_lib_list} ${extension_lib_list})
 
 foreach(lib ${lib_list})
   # Name of the variable which stores result of the find_library search
@@ -213,3 +216,12 @@ foreach(lib ${backend_lib_list})
     target_link_options(executorch_backends INTERFACE ${lib})
   endif()
 endforeach()
+
+# An interface target containing all available extensions.
+add_library(executorch_extensions INTERFACE)
+
+foreach(lib ${extension_lib_list})
+  if(TARGET ${lib})
+    target_link_options(executorch_extensions INTERFACE ${lib})
+  endif()
+endforeach()