@@ -25,47 +25,58 @@ target_include_directories(
2525
2626target_include_directories (extension_training PUBLIC ${EXECUTORCH_ROOT} /..)
2727target_compile_options (extension_training PUBLIC ${_common_compile_options} )
28- target_link_libraries (extension_training executorch_core kernels_util_all_deps
29- extension_data_loader extension_module_static extension_tensor extension_flat_tensor )
30-
28+ target_link_libraries (
29+ extension_training
30+ executorch_core
31+ kernels_util_all_deps
32+ extension_data_loader
33+ extension_module_static
34+ extension_tensor
35+ extension_flat_tensor
36+ )
3137
3238list (TRANSFORM _train_xor__srcs PREPEND "${EXECUTORCH_ROOT} /" )
3339add_executable (train_xor ${_train_xor__srcs} )
34- target_include_directories (
35- train_xor PUBLIC ${_common_include_directories}
36- )
40+ target_include_directories (train_xor PUBLIC ${_common_include_directories} )
3741target_link_libraries (
38- train_xor gflags executorch_core portable_ops_lib extension_tensor
39- extension_training program_schema
42+ train_xor
43+ gflags
44+ executorch_core
45+ portable_ops_lib
46+ extension_tensor
47+ extension_training
48+ program_schema
4049)
4150target_compile_options (train_xor PUBLIC ${_common_compile_options} )
4251
4352if (EXECUTORCH_BUILD_PYBIND)
4453 # Pybind library.
45- set (_pybind_training_dep_libs
46- ${TORCH_PYTHON_LIBRARY}
47- etdump
48- executorch
49- util
50- torch
51- extension_training
54+ set (_pybind_training_dep_libs ${TORCH_PYTHON_LIBRARY} etdump executorch util
55+ torch extension_training
5256 )
5357
5458 if (EXECUTORCH_BUILD_XNNPACK)
55- # need to explicitly specify XNNPACK and xnnpack-microkernels-prod
56- # here otherwise uses XNNPACK and microkernel-prod symbols from libtorch_cpu
57- list (APPEND _pybind_training_dep_libs xnnpack_backend XNNPACK xnnpack-microkernels-prod)
59+ # need to explicitly specify XNNPACK and xnnpack-microkernels-prod here
60+ # otherwise uses XNNPACK and microkernel-prod symbols from libtorch_cpu
61+ list (APPEND _pybind_training_dep_libs xnnpack_backend XNNPACK
62+ xnnpack-microkernels-prod
63+ )
5864 endif ()
5965
6066 # pybind training
61- pybind11_add_module(_training_lib SHARED ${CMAKE_CURRENT_SOURCE_DIR} /pybindings/_training_lib.cpp)
67+ pybind11_add_module(
68+ _training_lib SHARED
69+ ${CMAKE_CURRENT_SOURCE_DIR} /pybindings/_training_lib.cpp
70+ )
6271
6372 target_include_directories (_training_lib PRIVATE ${TORCH_INCLUDE_DIRS} )
64- target_compile_options (_training_lib PUBLIC -Wno-deprecated-declarations -fPIC -frtti -fexceptions)
73+ target_compile_options (
74+ _training_lib PUBLIC -Wno-deprecated-declarations -fPIC -frtti -fexceptions
75+ )
6576 target_link_libraries (_training_lib PRIVATE ${_pybind_training_dep_libs} )
6677
6778 install (TARGETS _training_lib
68- LIBRARY DESTINATION executorch/extension/training/pybindings
79+ LIBRARY DESTINATION executorch/extension/training/pybindings
6980 )
7081endif ()
7182
0 commit comments