@@ -25,47 +25,58 @@ target_include_directories(
25
25
26
26
target_include_directories (extension_training PUBLIC ${EXECUTORCH_ROOT} /.. )
27
27
target_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
+ )
31
37
32
38
list (TRANSFORM _train_xor__srcs PREPEND "${EXECUTORCH_ROOT} /" )
33
39
add_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} )
37
41
target_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
40
49
)
41
50
target_compile_options (train_xor PUBLIC ${_common_compile_options} )
42
51
43
52
if (EXECUTORCH_BUILD_PYBIND )
44
53
# 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
52
56
)
53
57
54
58
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
+ )
58
64
endif ()
59
65
60
66
# 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
+ )
62
71
63
72
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
+ )
65
76
target_link_libraries (_training_lib PRIVATE ${_pybind_training_dep_libs} )
66
77
67
78
install (TARGETS _training_lib
68
- LIBRARY DESTINATION executorch/extension/training/pybindings
79
+ LIBRARY DESTINATION executorch/extension/training/pybindings
69
80
)
70
81
endif ()
71
82
0 commit comments