diff --git a/CMakeLists.txt b/CMakeLists.txt index 928e7b7b7..baa9b5828 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -15,8 +15,10 @@ #============================================================================= cmake_minimum_required(VERSION 3.23.1 FATAL_ERROR) +set(rapids-cmake-repo bdice/rapids-cmake) +set(rapids-cmake-branch rapids_cpm_cccl) if(NOT EXISTS ${CMAKE_CURRENT_BINARY_DIR}/CUCO_RAPIDS.cmake) - file(DOWNLOAD https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-23.12/RAPIDS.cmake + file(DOWNLOAD https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-24.02/RAPIDS.cmake ${CMAKE_CURRENT_BINARY_DIR}/CUCO_RAPIDS.cmake) endif() include(${CMAKE_CURRENT_BINARY_DIR}/CUCO_RAPIDS.cmake) @@ -74,8 +76,7 @@ rapids_find_package( rapids_cpm_init() -include(cmake/thirdparty/get_thrust.cmake) -include(cmake/thirdparty/get_libcudacxx.cmake) +include(cmake/thirdparty/get_cccl.cmake) ################################################################################################### # - cuco target --------------------------------------------------------------------------------- @@ -84,7 +85,7 @@ add_library(cuco::cuco ALIAS cuco) target_include_directories(cuco INTERFACE INTERFACE $ $) -target_link_libraries(cuco INTERFACE libcudacxx::libcudacxx CUDA::toolkit $) +target_link_libraries(cuco INTERFACE libcudacxx::libcudacxx CUDA::toolkit $) target_compile_features(cuco INTERFACE cxx_std_17 cuda_std_17) ################################################################################################### @@ -128,21 +129,13 @@ structures tailored for efficient use with GPUs. ]=]) -set(code_string -[=[ -if(NOT TARGET cuco::Thrust) -thrust_create_target(cuco::Thrust FROM_OPTIONS) -endif() -]=]) - # build directory cuco-config generation rapids_export( BUILD cuco EXPORT_SET cuco-exports GLOBAL_TARGETS cuco NAMESPACE cuco:: - DOCUMENTATION doc_string - FINAL_CODE_BLOCK code_string) + DOCUMENTATION doc_string) if(INSTALL_CUCO) install(DIRECTORY include/cuco/ DESTINATION include/cuco) diff --git a/cmake/thirdparty/get_thrust.cmake b/cmake/thirdparty/get_cccl.cmake similarity index 71% rename from cmake/thirdparty/get_thrust.cmake rename to cmake/thirdparty/get_cccl.cmake index a79617807..27e7d81e3 100644 --- a/cmake/thirdparty/get_thrust.cmake +++ b/cmake/thirdparty/get_cccl.cmake @@ -1,5 +1,5 @@ # ============================================================================= -# Copyright (c) 2021-2022, NVIDIA CORPORATION. +# Copyright (c) 2021-2023, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except # in compliance with the License. You may obtain a copy of the License at @@ -12,11 +12,10 @@ # the License. # ============================================================================= -# Use CPM to find or clone thrust -function(find_and_configure_thrust) - include(${rapids-cmake-dir}/cpm/thrust.cmake) - rapids_cpm_thrust(NAMESPACE cuco - BUILD_EXPORT_SET cuco-exports) +# Use CPM to find or clone CCCL +function(find_and_configure_cccl) + include(${rapids-cmake-dir}/cpm/cccl.cmake) + rapids_cpm_cccl(BUILD_EXPORT_SET cuco-exports) endfunction() -find_and_configure_thrust() +find_and_configure_cccl()