changes

Author: JacobSzwejbka

CMakeLists.txt

@@ -240,6 +240,17 @@ cmake_dependent_option(
   "NOT EXECUTORCH_BUILD_ARM_BAREMETAL" OFF
 )
 
+if(EXECUTORCH_BUILD_PYBIND)
+  set(EXECUTORCH_BUILD_EXTENSION_DATA_LOADER ON)
+  set(EXECUTORCH_BUILD_DEVTOOLS ON)
+endif()
+
+if(EXECUTORCH_BUILD_EXTENSION_TRAINING)
+  set(EXECUTORCH_BUILD_EXTENSION_TENSOR ON)
+  set(EXECUTORCH_BUILD_EXTENSION_DATA_LOADER ON)
+  set(EXECUTORCH_BUILD_EXTENSION_MODULE ON)
+endif()
+
 if(EXECUTORCH_BUILD_KERNELS_CUSTOM_AOT)
   set(EXECUTORCH_BUILD_EXTENSION_TENSOR ON)
   set(EXECUTORCH_BUILD_KERNELS_CUSTOM ON)
@@ -713,14 +724,6 @@ endif()
 if(EXECUTORCH_BUILD_PYBIND)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/third-party/pybind11)
 
-  if(NOT EXECUTORCH_BUILD_EXTENSION_DATA_LOADER)
-    add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/data_loader)
-  endif()
-
-  if(NOT EXECUTORCH_BUILD_DEVTOOLS)
-    add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/devtools)
-  endif()
-
   # find pytorch lib, to allow pybind to take at::Tensor as input/output
   find_package(Torch CONFIG REQUIRED)
   find_library(
@@ -816,7 +819,6 @@ if(EXECUTORCH_BUILD_PYBIND)
     target_compile_options(_training_lib PUBLIC ${_pybind_compile_options})
     target_link_libraries(_training_lib PRIVATE ${_pybind_training_dep_libs})
 
-    message(STATUS "Installing Training Lib")
     install(TARGETS _training_lib
             LIBRARY DESTINATION executorch/extension/training/pybindings
     )

setup.py

@@ -585,6 +585,7 @@ def run(self):
                 cmake_args += [
                     "-DEXECUTORCH_BUILD_EXTENSION_TRAINING=ON",
                 ]
+                build_args += ["--target", "_training_lib"]
             build_args += ["--target", "portable_lib"]
             # To link backends into the portable_lib target, callers should
             # add entries like `-DEXECUTORCH_BUILD_XNNPACK=ON` to the CMAKE_ARGS