feat(cccl): integrate missing CCCL directories — python/, ci/, .agent/, docs/, test/

Sparse-checkout from NVIDIA/cccl main branch to complete cccl_upstream:

Added:
- python/cuda_cccl/ (226 files) — Python bindings for device-level algorithms
  Critical for muh toolchain: cuda.compute.reduce_into, scan, radix_sort, etc.
  Includes 204 .py files with full test coverage for all 27 algorithms
- ci/ (163 files) — Build/test infrastructure
  build_cub.sh, test_cub.sh, build_and_test_targets.sh, matrix.yaml
  Directly maps to our [INFRA-CI] and [INFRA-BUILD] items
- .agent/skills/ (7 files) — NVIDIA's own agent skills for CCCL
  cccl-style/SKILL.md, cccl-test/SKILL.md, sass-diff/SKILL.md
- docs/ (491 files) — Official CCCL documentation
  CI references, CMake guides, Python compute docs, libcudacxx PTX docs
- test/ (12 files) — Top-level integration tests (cuda_smoke, stdpar)
- Root configs: .clang-format, .clang-tidy, CONTRIBUTING.md, pyproject.toml
- CLAUDE.md symlink → AGENTS.md (NVIDIA's standard)

cccl_upstream now mirrors full NVIDIA/cccl structure:
  Before: 42M (cub + thrust + libcudacxx + cudax + c + examples + benchmarks)
  After:  53M (+python +ci +docs +.agent +test +configs)

This completes the CCCL base needed for:
- [muh-bench] items: ci/util/build_and_test_targets.sh for targeted builds
- [CCCL-verify] items: python/cuda_cccl/tests/ as reference implementations
- [CCCL-test] items: ci/test_cub.sh, ci/test_thrust.sh
- Agent workflow: .agent/skills/ for consistent style and test patterns
This commit is contained in:
muh-bot
2026-08-07 02:34:33 +00:00
parent 3f97dca7ad
commit 2a7ca101d7
908 changed files with 121615 additions and 0 deletions

View File

@@ -0,0 +1,166 @@
# Test the CMake packages for CCCL and all subprojects.
#
# Parameters:
# - CCCL_ROOT [Path] Root of the CCCL repo, or an installation root.
# - COMPONENTS [StringList] {Thrust CUB libcudacxx} Which CCCL subprojects
# should be found.
# - PACKAGE_TYPE [String] {CCCL | NATIVE | SUBDIR}:
# - CCCL -> `find_package(CCCL COMPONENTS <subproject>)`
# - NATIVE -> `find_package(<subproject>)`
# - SUBDIR -> `set(CCCL_REQUIRED_COMPONENTS <subproject>)`
# `add_subdirectory(${cccl_root})`
cmake_minimum_required(VERSION 3.21)
project(CCCLTestExport LANGUAGES CXX)
include(CTest)
enable_testing()
set(CCCL_ROOT "" CACHE PATH "Root of the CCCL repo, or an installation root.")
set(
COMPONENTS
""
CACHE STRING
"DEFAULT for no components, or semi-colon delimited list of Thrust, CUB, and/or libcudacxx."
)
set(
PACKAGE_TYPE
""
CACHE STRING
"CCCL: Find CCCL with subpackages as components; NATIVE: Find subpackages directly; SUBDIR: add_subdirectory(${CCCL_ROOT}"
)
set_property(CACHE PACKAGE_TYPE PROPERTY STRINGS CCCL NATIVE SUBDIR)
message(STATUS "CCCL_ROOT=${CCCL_ROOT}")
message(STATUS "COMPONENTS=${COMPONENTS}")
message(STATUS "PACKAGE_TYPE=${PACKAGE_TYPE}")
function(do_find_package pkg_name pkg_prefix)
list(
APPEND arg_list
REQUIRED
${ARGN}
NO_DEFAULT_PATH
HINTS
"${pkg_prefix}"
)
list(JOIN arg_list " " arg_str)
message(STATUS "Executing: find_package(${pkg_name} ${arg_str})")
find_package(${pkg_name} ${arg_list})
if (NOT ${pkg_name}_FOUND)
message(FATAL_ERROR "Failed: find_package(${pkg_name} ${arg_str})")
endif()
# Re-execute find_package to ensure that repeated calls don't break:
find_package(${pkg_name} ${arg_list})
endfunction()
# Run find package with the requested configuration:
if (PACKAGE_TYPE STREQUAL "CCCL")
if (COMPONENTS STREQUAL "DEFAULT")
do_find_package(CCCL "${CCCL_ROOT}")
else()
do_find_package(CCCL "${CCCL_ROOT}" COMPONENTS ${COMPONENTS})
endif()
elseif (PACKAGE_TYPE STREQUAL "NATIVE")
if (COMPONENTS STREQUAL "DEFAULT")
message(
FATAL_ERROR
"COMPONENTS=DEFAULT incompatible with PACKAGE_TYPE=NATIVE"
)
endif()
foreach (component IN LISTS COMPONENTS)
do_find_package(${component} "${CCCL_ROOT}")
endforeach()
elseif (PACKAGE_TYPE STREQUAL "SUBDIR")
if (COMPONENTS STREQUAL "DEFAULT")
set(CCCL_REQUIRED_COMPONENTS)
else()
set(CCCL_REQUIRED_COMPONENTS ${COMPONENTS})
endif()
add_subdirectory("${CCCL_ROOT}" "${CMAKE_CURRENT_BINARY_DIR}/subdir")
else()
message(FATAL_ERROR "Invalid PACKAGE_TYPE: ${PACKAGE_TYPE}")
endif()
if (COMPONENTS STREQUAL "DEFAULT")
set(COMPONENTS libcudacxx CUB Thrust)
if (CCCL_ENABLE_UNSTABLE)
list(APPEND COMPONENTS cudax)
endif()
endif()
foreach (component IN LISTS COMPONENTS)
set(test_target version_check.${component})
set(component_target "${component}::${component}")
add_executable(${test_target} version_check.cxx)
target_compile_features(${test_target} PUBLIC cxx_std_17)
target_link_libraries(${test_target} PRIVATE ${component_target})
add_test(NAME ${test_target} COMMAND ${test_target})
if (component STREQUAL "libcudacxx")
math(
EXPR
component_cmake_version
"(${LIBCUDACXX_VERSION_MAJOR} * 1000000) +
${LIBCUDACXX_VERSION_MINOR} * 1000 +
${LIBCUDACXX_VERSION_PATCH}"
)
target_compile_definitions(
${test_target}
PRIVATE
"VERSION_HEADER=cuda/std/version"
"VERSION_MACRO=_LIBCUDACXX_CUDA_API_VERSION"
"EXPECTED_VERSION=${component_cmake_version}"
)
elseif (component STREQUAL "CUB")
math(
EXPR
component_cmake_version
"(${CUB_VERSION_MAJOR} * 100000) +
${CUB_VERSION_MINOR} * 100 +
${CUB_VERSION_PATCH}"
)
target_compile_definitions(
${test_target}
PRIVATE
"VERSION_HEADER=cub/version.cuh"
"VERSION_MACRO=CUB_VERSION"
"EXPECTED_VERSION=${component_cmake_version}"
)
elseif (component STREQUAL "Thrust")
math(
EXPR
component_cmake_version
"(${THRUST_VERSION_MAJOR} * 100000) +
${THRUST_VERSION_MINOR} * 100 +
${THRUST_VERSION_PATCH}"
)
target_compile_definitions(
${test_target}
PRIVATE
"VERSION_HEADER=thrust/version.h"
"VERSION_MACRO=THRUST_VERSION"
"EXPECTED_VERSION=${component_cmake_version}"
)
elseif (component STREQUAL "cudax")
math(
EXPR
component_cmake_version
"(${CUDAX_VERSION_MAJOR} * 1000000) +
${CUDAX_VERSION_MINOR} * 1000 +
${CUDAX_VERSION_PATCH}"
)
target_compile_definitions(
${test_target}
PRIVATE
"VERSION_HEADER=cuda/experimental/version.cuh"
"VERSION_MACRO=CUDAX_VERSION"
"EXPECTED_VERSION=${component_cmake_version}"
)
else()
message(
FATAL_ERROR
"Valid COMPONENTS are (case-sensitive): Thrust;CUB;libcudacxx;cudax"
)
endif()
endforeach()

View File

@@ -0,0 +1,20 @@
// Compile with:
// -DVERSION_HEADER=include/path/for/version.h
// -DEXPECTED_VERSION=XXYYZZ
// -DVERSION_MACRO=PROJECT_VERSION
#define HEADER <VERSION_HEADER>
#include HEADER
#include <cstdio>
#define DETECTED_VERSION VERSION_MACRO
int main()
{
printf("Expected version: %d\n"
"Detected version: %d\n",
EXPECTED_VERSION,
VERSION_MACRO);
return EXPECTED_VERSION == DETECTED_VERSION ? 0 : 1;
}