dylanyunlong/project_6
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix: use cccl_preload::cub namespace — CUB_WRAPPED_NAMESPACE requires it

Browse Source 
CUB_DISABLE_NAMESPACE_MAGIC requires CUB_WRAPPED_NAMESPACE.
CUB_WRAPPED_NAMESPACE=cccl_preload wraps cub into cccl_preload::cub.
Source must use cccl_preload::cub::CachingDeviceAllocator.
This commit is contained in:
Claude
2026-08-13 10:36:27 +00:00
parent c1e23615b5
commit 3ce5bff10f
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
qwen3_6_scripts/cccl_preload/build_cccl_preload.sh
View File

@@ -75,6 +75,7 @@ COMMON_FLAGS=(
# CUB needs this for non-NVCC compilers # CUB needs this for non-NVCC compilers
-D__CUDA_ARCH_LIST__=700 -D__CUDA_ARCH_LIST__=700
-DCUB_DISABLE_NAMESPACE_MAGIC -DCUB_DISABLE_NAMESPACE_MAGIC
-DCUB_WRAPPED_NAMESPACE=cccl_preload
) )
if [[ "${CXX}" == *clang++* ]]; then if [[ "${CXX}" == *clang++* ]]; then

4
qwen3_6_scripts/cccl_preload/cccl_allocator_preload.cu
View File

@@ -41,8 +41,8 @@ static constexpr unsigned int ALLOC_MAX_BIN = 32; /* 4 GB */
static constexpr size_t ALLOC_MAX_CACHED = (size_t)8 * 1024 * 1024 * 1024; /* 8GB */ static constexpr size_t ALLOC_MAX_CACHED = (size_t)8 * 1024 * 1024 * 1024; /* 8GB */
/* ---- Global allocator singleton ---- */ /* ---- Global allocator singleton ---- */
static cub::CachingDeviceAllocator& get_allocator() { static cccl_preload::cub::CachingDeviceAllocator& get_allocator() {
static cub::CachingDeviceAllocator instance( static cccl_preload::cub::CachingDeviceAllocator instance(
ALLOC_BIN_GROWTH, ALLOC_BIN_GROWTH,
ALLOC_MIN_BIN, ALLOC_MIN_BIN,
ALLOC_MAX_BIN, ALLOC_MAX_BIN,