Ameya Keshava Mallya
35 行
1.2 KiB
CMake
35 行
1.2 KiB
CMake
# Find all C source files in current directory
|
|
set(SRC_FILES
|
|
${CMAKE_CURRENT_SOURCE_DIR}/plugin.cc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/print_event.cc
|
|
)
|
|
|
|
# Create shared library
|
|
add_library(nccl-profiler-example SHARED ${SRC_FILES})
|
|
|
|
# Set include directories
|
|
target_include_directories(nccl-profiler-example PRIVATE
|
|
${CMAKE_CURRENT_SOURCE_DIR}/nccl
|
|
${CUDAToolkit_INCLUDE_DIRS}
|
|
)
|
|
|
|
# Set output name to match Makefile
|
|
set_target_properties(nccl-profiler-example PROPERTIES
|
|
OUTPUT_NAME "nccl-profiler-example"
|
|
PREFIX "lib"
|
|
POSITION_INDEPENDENT_CODE ON
|
|
LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib
|
|
)
|
|
|
|
add_custom_command(TARGET nccl-profiler-example POST_BUILD
|
|
COMMAND ${CMAKE_COMMAND} -E make_directory ${CMAKE_BINARY_DIR}/test/unit/plugins
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_BINARY_DIR}/lib/libnccl-profiler-example.so ${CMAKE_BINARY_DIR}/test/unit/plugins
|
|
)
|
|
|
|
# Add custom target for clean (equivalent to Makefile clean target)
|
|
add_custom_target(clean-profiler-lib
|
|
COMMAND ${CMAKE_COMMAND} -E remove -f ${CMAKE_BINARY_DIR}/lib/libnccl-profiler-example.so
|
|
COMMAND ${CMAKE_COMMAND} -E remove -f ${CMAKE_BINARY_DIR}/test/unit/plugins/libnccl-profiler-example.so
|
|
COMMENT "Cleaning libnccl-profiler-example.so"
|
|
)
|