56fd68e7dd
CCCL (CUDA C++ Core Libraries) provides: - CUB: device/block/warp-level GPU primitives (reduce, scan, sort, topk) - Thrust: high-level parallel algorithms (transform_reduce, sort, scan) - libcudacxx: CUDA C++ standard library (atomics, barriers, memory) - cudax: experimental features (memory resources, allocators) - Tuning policies: per-SM hardware-specific algorithm parameters Competition optimization vectors mapped to CCCL: - Output TPS (83% weight): warp_reduce, block_reduce, device_topk - Input TPS (14% weight): device_scan, block_load, prefetch - Cache TPS (3% weight): prefix caching strategy patterns - Memory (0.9 util): pooled/cached/buddy allocators Source: https://github.com/NVIDIA/cccl (shallow clone, HEAD only) License: Apache-2.0
60 lines
1.7 KiB
CMake
60 lines
1.7 KiB
CMake
cccl_get_cudatoolkit()
|
|
|
|
set(cccl_revision "")
|
|
find_package(Git)
|
|
if (GIT_FOUND)
|
|
execute_process(
|
|
COMMAND ${GIT_EXECUTABLE} describe
|
|
WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
|
|
OUTPUT_VARIABLE cccl_revision
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
)
|
|
|
|
if (cccl_revision STREQUAL "")
|
|
# Sometimes, there is no tag (shallow copy)
|
|
execute_process(
|
|
COMMAND ${GIT_EXECUTABLE} rev-parse --short HEAD
|
|
WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
|
|
OUTPUT_VARIABLE cccl_revision
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
)
|
|
endif()
|
|
endif()
|
|
|
|
# Sometimes this script is used outside of a Git repository.
|
|
# In this case, we read the revision from cccl/cccl_version instead.
|
|
if ("${cccl_revision}" STREQUAL "")
|
|
file(READ "${CMAKE_SOURCE_DIR}/cccl_version" cccl_revision)
|
|
string(STRIP "${cccl_revision}" cccl_revision)
|
|
string(REPLACE "\n" "" cccl_revision "${cccl_revision}")
|
|
endif()
|
|
message(STATUS "Git revision: ${cccl_revision}")
|
|
|
|
function(get_meta_path meta_path)
|
|
set(meta_path "${CMAKE_BINARY_DIR}/cccl_meta_bench.csv" PARENT_SCOPE)
|
|
endfunction()
|
|
|
|
function(create_benchmark_registry)
|
|
get_meta_path(meta_path)
|
|
|
|
set(ctk_version "${CUDAToolkit_VERSION}")
|
|
message(STATUS "CTK version: ${ctk_version}")
|
|
|
|
file(REMOVE "${meta_path}")
|
|
file(APPEND "${meta_path}" "ctk_version,${ctk_version}\n")
|
|
file(APPEND "${meta_path}" "cccl_revision,${cccl_revision}\n")
|
|
endfunction()
|
|
|
|
function(register_cccl_tuning bench_name ranges)
|
|
get_meta_path(meta_path)
|
|
if ("${ranges}" STREQUAL "")
|
|
file(APPEND "${meta_path}" "${bench_name}\n")
|
|
else()
|
|
file(APPEND "${meta_path}" "${bench_name},${ranges}\n")
|
|
endif()
|
|
endfunction()
|
|
|
|
function(register_cccl_benchmark bench_name)
|
|
register_cccl_tuning("${bench_name}" "")
|
|
endfunction()
|