Skip to content

Commit 2a98470

Browse files
authored
fix(build): regenerate communication bridge for header changes (#57)
1 parent 31a44f3 commit 2a98470

1 file changed

Lines changed: 17 additions & 0 deletions

File tree

‎src/CMakeLists.txt‎

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,21 @@ set(AUTOGEN_WARNING [[/*
1616

1717
file(GLOB CORE_SRCS "*.cc")
1818
file(GLOB_RECURSE BASE_IMPL_SRCS "base/*.cc")
19+
file(GLOB_RECURSE BRIDGE_DEP_HEADERS CONFIGURE_DEPENDS
20+
"base/*.h"
21+
"backends/*.h"
22+
"devices/*.h"
23+
"devices/*.cuh"
24+
)
25+
26+
string(REPLACE ";" "\n" BRIDGE_DEPENDENCY_CONTENT "${BRIDGE_DEP_HEADERS}")
27+
set(BRIDGE_DEPENDENCY_MANIFEST
28+
"${CMAKE_CURRENT_BINARY_DIR}/bridge_dependencies.txt"
29+
)
30+
file(GENERATE
31+
OUTPUT "${BRIDGE_DEPENDENCY_MANIFEST}"
32+
CONTENT "${BRIDGE_DEPENDENCY_CONTENT}\n"
33+
)
1934

2035
target_sources(infiniccl PRIVATE
2136
${CORE_SRCS}
@@ -298,7 +313,9 @@ add_custom_command(
298313
"${BACK_STR}"
299314
DEPENDS "${PROJECT_SOURCE_DIR}/scripts/gen_bridge.py"
300315
"${PROJECT_SOURCE_DIR}/include/comm.h"
316+
"${BRIDGE_DEPENDENCY_MANIFEST}"
301317
${BASE_IMPL_SRCS}
318+
${BRIDGE_DEP_HEADERS}
302319
VERBATIM
303320
COMMENT "Generating InfiniCCL bridge and manifest files for Devices: [${DEV_STR}] Backends: [${BACK_STR}]..."
304321
)

0 commit comments

Comments
 (0)