Skip to content

Commit 733eb9e

Browse files
committed
Rearranging if logic
1 parent e4c4908 commit 733eb9e

File tree

1 file changed

+27
-28
lines changed

1 file changed

+27
-28
lines changed

tools/cmake/Codegen.cmake

Lines changed: 27 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -238,42 +238,41 @@ function(gen_operators_lib)
238238

239239
set(_common_compile_options -Wno-deprecated-declarations -ffunction-sections -fdata-sections -Os)
240240

241-
if("${portable_kernels_check}" IN_LIST GEN_KERNEL_LIBS)
242-
list(REMOVE_ITEM GEN_KERNEL_LIBS ${portable_kernels_check})
243-
244-
# Define portable kernels sources (same as in kernels/portable/CMakeLists.txt)
245-
file(GLOB_RECURSE _portable_kernels_srcs
246-
"${EXECUTORCH_ROOT}/kernels/portable/cpu/*.cpp"
247-
)
248-
list(FILTER _portable_kernels_srcs EXCLUDE REGEX "test/*.cpp")
249-
list(FILTER _portable_kernels_srcs EXCLUDE REGEX "codegen")
250-
251-
# Build kernels_util_all_deps, since later selected_portable_kernels depends on it
252-
list(TRANSFORM _kernels_util_all_deps__srcs PREPEND "${EXECUTORCH_ROOT}/")
253-
add_library(selected_kernels_util_all_deps ${_kernels_util_all_deps__srcs})
254-
target_link_libraries(selected_kernels_util_all_deps PRIVATE executorch_core)
255-
target_include_directories(selected_kernels_util_all_deps PUBLIC ${_common_include_directories})
256-
target_compile_definitions(selected_kernels_util_all_deps PUBLIC C10_USING_CUSTOM_GENERATED_MACROS)
257-
target_compile_options(selected_kernels_util_all_deps PUBLIC ${_common_compile_options})
258-
259-
# Build selected_portable_kernels
260-
add_library(selected_portable_kernels ${_portable_kernels_srcs})
261-
target_link_libraries(selected_portable_kernels PRIVATE executorch_core selected_kernels_util_all_deps)
262-
target_compile_options(selected_portable_kernels PUBLIC ${_common_compile_options})
263-
target_include_directories(selected_portable_kernels PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/${GEN_LIB_NAME}/)
264-
265-
# Add dependency on generated header if dtype selective build is enabled
266-
if(GEN_DTYPE_SELECTIVE_BUILD)
241+
if(GEN_DTYPE_SELECTIVE_BUILD)
242+
if("${portable_kernels_check}" IN_LIST GEN_KERNEL_LIBS)
243+
list(REMOVE_ITEM GEN_KERNEL_LIBS ${portable_kernels_check})
244+
245+
# Define portable kernels sources (same as in kernels/portable/CMakeLists.txt)
246+
file(GLOB_RECURSE _portable_kernels_srcs
247+
"${EXECUTORCH_ROOT}/kernels/portable/cpu/*.cpp"
248+
)
249+
list(FILTER _portable_kernels_srcs EXCLUDE REGEX "test/*.cpp")
250+
list(FILTER _portable_kernels_srcs EXCLUDE REGEX "codegen")
251+
252+
# Build kernels_util_all_deps, since later selected_portable_kernels depends on it
253+
list(TRANSFORM _kernels_util_all_deps__srcs PREPEND "${EXECUTORCH_ROOT}/")
254+
add_library(selected_kernels_util_all_deps ${_kernels_util_all_deps__srcs})
255+
target_link_libraries(selected_kernels_util_all_deps PRIVATE executorch_core)
256+
target_include_directories(selected_kernels_util_all_deps PUBLIC ${_common_include_directories})
257+
target_compile_definitions(selected_kernels_util_all_deps PUBLIC C10_USING_CUSTOM_GENERATED_MACROS)
258+
target_compile_options(selected_kernels_util_all_deps PUBLIC ${_common_compile_options})
259+
260+
# Build selected_portable_kernels
261+
add_library(selected_portable_kernels ${_portable_kernels_srcs})
262+
target_link_libraries(selected_portable_kernels PRIVATE executorch_core selected_kernels_util_all_deps)
263+
target_compile_options(selected_portable_kernels PUBLIC ${_common_compile_options})
264+
target_include_directories(selected_portable_kernels PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/${GEN_LIB_NAME}/)
265+
267266
# Make sure the header is generated before compiling the library
268267
add_dependencies(selected_portable_kernels ${GEN_LIB_NAME})
269268
# Create a custom target for the header to ensure proper dependency tracking
270269
add_custom_target(selected_portable_kernels_header DEPENDS ${_opvariant_h})
271270
add_dependencies(selected_portable_kernels selected_portable_kernels_header)
272271
# Apply the compile definition for dtype selective build
273272
target_compile_definitions(selected_portable_kernels PRIVATE EXECUTORCH_SELECTIVE_BUILD_DTYPE=1)
274-
endif()
275273

276-
target_link_libraries(${GEN_LIB_NAME} PUBLIC selected_portable_kernels)
274+
target_link_libraries(${GEN_LIB_NAME} PUBLIC selected_portable_kernels)
275+
endif()
277276
endif()
278277

279278
if(GEN_KERNEL_LIBS)

0 commit comments

Comments
 (0)