Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 11 additions & 5 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,11 @@ if(NOT TARGET zstd)
rapids_cpm_find(zstd
1.5.7
GLOBAL_TARGETS
zstd
zstd::libzstd_static
BUILD_EXPORT_SET
rtcx-exports
INSTALL_EXPORT_SET
rtcx-exports
CPM_ARGS
GIT_REPOSITORY
https://github.com/facebook/zstd.git
Expand Down Expand Up @@ -59,7 +63,7 @@ endif()

if(NOT TARGET nvtx3::nvtx3-cpp)
include(${rapids-cmake-dir}/cpm/nvtx3.cmake)
rapids_cpm_nvtx3()
rapids_cpm_nvtx3(BUILD_EXPORT_SET rtcx-exports INSTALL_EXPORT_SET rtcx-exports)
endif()

add_library(rtcx STATIC hash.cpp rtcx.cpp AlgorithmLauncher.cpp AlgorithmPlanner.cpp
Expand All @@ -82,13 +86,15 @@ if(RTCX_STATIC_LINK_NVRTC)
target_link_libraries(rtcx PRIVATE CUDA::nvrtc_static CUDA::nvrtc_builtins_static)
target_compile_definitions(rtcx PRIVATE RTCX_STATIC_LINK_LIBNVRTC=1)
else()
target_link_libraries(rtcx PRIVATE CUDA::nvrtc CUDA::nvrtc_builtins)
target_compile_definitions(rtcx PRIVATE RTCX_STATIC_LINK_LIBNVRTC=0)
endif()

if(RTCX_STATIC_LINK_NVJITLINK)
target_link_libraries(rtcx PRIVATE CUDA::nvJitLink_static)
target_compile_definitions(rtcx PRIVATE RTCX_STATIC_LINK_LIBNVJITLINK=1)
else()
target_link_libraries(rtcx PRIVATE CUDA::nvJitLink)
target_compile_definitions(rtcx PRIVATE RTCX_STATIC_LINK_LIBNVJITLINK=0)
endif()

Expand Down Expand Up @@ -141,9 +147,9 @@ if(RTCX_INSTALL)
install(FILES embed.cmake DESTINATION ${lib_dir}/cmake/rtcx)
install(DIRECTORY embed/ DESTINATION ${lib_dir}/cmake/rtcx/embed)
install(FILES AlgorithmLauncher.hpp AlgorithmPlanner.hpp FragmentEntry.hpp nvjitlink_checker.hpp
NVRTCLTOFRagmentCompiler.hpp DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/rtcx)
install(FILES generate_jit_lto_kernels.cmake register_fatbin.cpp.in
DESTINATION ${lib_dir}/cmake/rtcx)
NVRTCLTOFragmentCompiler.hpp DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/rtcx)
install(FILES generate_jit_lto_kernels.cmake compute_matrix_product.cmake
compute_matrix_product.py register_fatbin.cpp.in DESTINATION ${lib_dir}/cmake/rtcx)
install(DIRECTORY libcxx/ DESTINATION ${CMAKE_INSTALL_DATADIR}/rtcx/libcxx)

rapids_export(INSTALL rtcx EXPORT_SET rtcx-exports GLOBAL_TARGETS rtcx NAMESPACE rtcx::
Expand Down