Skip to content

Commit

Permalink
Only cufft offers a static_nocallback version of the library (#5703)
Browse files Browse the repository at this point in the history
Authors:
  - Robert Maynard (https://github.com/robertmaynard)

Approvers:
  - Bradley Dice (https://github.com/bdice)

URL: #5703
  • Loading branch information
robertmaynard authored Jan 9, 2024
1 parent b8daf81 commit c8b9901
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions cpp/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -141,8 +141,10 @@ endif()
# - compiler options ---------------------------------------------------------

set(_ctk_static_suffix "")
set(_ctk_fft_static_suffix "")
if(CUDA_STATIC_RUNTIME)
set(_ctk_static_suffix "_static_nocallback")
set(_ctk_static_suffix "_static")
set(_ctk_fft_static_suffix "_static_nocallback")
endif()

if (NOT DISABLE_OPENMP)
Expand Down Expand Up @@ -612,7 +614,7 @@ if(BUILD_CUML_CPP_LIBRARY)
list(APPEND _cuml_cpp_private_libs
raft::raft
$<TARGET_NAME_IF_EXISTS:GPUTreeShap::GPUTreeShap>
$<$<BOOL:${LINK_CUFFT}>:CUDA::cufft${_ctk_static_suffix}>
$<$<BOOL:${LINK_CUFFT}>:CUDA::cufft${_ctk_fft_static_suffix}>
${TREELITE_LIBS}
${OpenMP_CXX_LIB_NAMES}
$<$<OR:$<BOOL:${BUILD_CUML_STD_COMMS}>,$<BOOL:${BUILD_CUML_MPI_COMMS}>>:NCCL::NCCL>
Expand Down

0 comments on commit c8b9901

Please sign in to comment.