Skip to content

Commit d00b24c

Browse files
authored
[easy] cmake-format extension/training/CMakeLists.txt (#12451)
It wasn't format-clean.
1 parent 7990a2c commit d00b24c

File tree

1 file changed

+32
-21
lines changed

1 file changed

+32
-21
lines changed

extension/training/CMakeLists.txt

Lines changed: 32 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -25,47 +25,58 @@ target_include_directories(
2525

2626
target_include_directories(extension_training PUBLIC ${EXECUTORCH_ROOT}/..)
2727
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+
)
3137

3238
list(TRANSFORM _train_xor__srcs PREPEND "${EXECUTORCH_ROOT}/")
3339
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})
3741
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
4049
)
4150
target_compile_options(train_xor PUBLIC ${_common_compile_options})
4251

4352
if(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
)
7081
endif()
7182

0 commit comments

Comments
 (0)