Skip to content

Commit

Permalink
Update to CCCL 2.2.0.
Browse files Browse the repository at this point in the history
  • Loading branch information
bdice committed Dec 7, 2023
1 parent 7fae640 commit ccb64ce
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 19 deletions.
17 changes: 5 additions & 12 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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 ---------------------------------------------------------------------------------
Expand Down Expand Up @@ -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)
Expand Down
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -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()

0 comments on commit ccb64ce

Please sign in to comment.