Add extension/data_loader targets to cmake build. (#521)

Summary:
Pull Request resolved: #521

We need to be able to build a separate lib for data loaders.

Reviewed By: cccclai

Differential Revision: D49742389

fbshipit-source-id: 884e28244aa8c484e0b184c11643d578ee689b62

Author: shoumikhin

CMakeLists.txt

@@ -92,8 +92,7 @@ option(REGISTER_QUANTIZED_OPS
 option(BUILD_SELECTIVE_BUILD_TEST
        "Whether to build binary for demo selective build" OFF)
 
-option(EXECUTORCH_BUILD_SIZE_TEST
-       "Whether to build size test" OFF)
+option(EXECUTORCH_BUILD_SIZE_TEST "Whether to build size test" OFF)
 
 if(BUILD_SELECTIVE_BUILD_TEST)
   option(SELECT_ALL_OPS
@@ -290,7 +289,13 @@ if(EXECUTORCH_BUILD_EXECUTOR_RUNNER)
   target_compile_options(executor_runner PUBLIC ${_common_compile_options})
 endif()
 
-# Add XNNPACK subdirectory
+option(EXECUTORCH_BUILD_EXTENSION_DATA_LOADER
+       "Build the extension/data_loader directory" OFF)
+if(EXECUTORCH_BUILD_EXTENSION_DATA_LOADER)
+  add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/data_loader)
+endif()
+
+option(EXECUTORCH_BUILD_XNNPACK "Build the backends/xnnpack directory" OFF)
 if(EXECUTORCH_BUILD_XNNPACK)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/backends/xnnpack)
 endif()

build/Utils.cmake

@@ -40,6 +40,11 @@ function(executorch_print_configuration_summary)
   )
   message(
     STATUS "  REGISTER_EXAMPLE_CUSTOM_OPS   : ${REGISTER_EXAMPLE_CUSTOM_OPS}")
+  message(
+    STATUS
+      "  EXECUTORCH_BUILD_EXTENSION_DATA_LOADER : ${EXECUTORCH_BUILD_EXTENSION_DATA_LOADER}"
+  )
+  message(STATUS "  EXECUTORCH_BUILD_XNNPACK : ${EXECUTORCH_BUILD_XNNPACK}")
 endfunction()
 
 # This is the funtion to use -Wl, --whole-archive to link static library NB:

build/cmake_deps.toml

@@ -118,3 +118,17 @@ deps = [
   "xnnpack_backend",
   "portable_kernels",
 ]
+
+[targets.extension_data_loader]
+buck_targets = [
+  "//extension/data_loader:buffer_data_loader",
+  "//extension/data_loader:file_data_loader",
+  "//extension/data_loader:mmap_data_loader",
+  "//extension/data_loader:shared_ptr_data_loader",
+]
+filters = [
+  ".cpp$",
+]
+deps = [
+  "executorch",
+]

extension/data_loader/CMakeLists.txt

@@ -0,0 +1,21 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+# Please this file formatted by running:
+# ~~~
+# cmake-format --first-comment-is-literal=True CMakeLists.txt
+# ~~~
+
+cmake_minimum_required(VERSION 3.19)
+
+# Source root directory for executorch.
+if(NOT EXECUTORCH_ROOT)
+  set(EXECUTORCH_ROOT ${CMAKE_CURRENT_SOURCE_DIR}/../..)
+endif()
+
+list(TRANSFORM _extension_data_loader__srcs PREPEND "${EXECUTORCH_ROOT}/")
+add_library(extension_data_loader ${_extension_data_loader__srcs})
+target_include_directories(extension_data_loader PUBLIC ${EXECUTORCH_ROOT}/..)