Compare commits
7 Commits
e1cdbfc167
...
19946d4c72
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
19946d4c72 | ||
|
|
188d3654ef | ||
|
|
0ac118911d | ||
|
|
8d6f9eaeb0 | ||
|
|
967d572073 | ||
|
|
502ea2fc96 | ||
|
|
327c2c9044 |
@@ -72,8 +72,7 @@ COMMON_FLAGS=(
|
||||
# Suppress CCCL warnings that don't affect correctness
|
||||
-Wno-unused-function
|
||||
-Wno-unknown-pragmas
|
||||
# CUB needs these for non-NVCC compilers
|
||||
-D_CCCL_COMPILER_GCC=1
|
||||
# Non-NVCC: disable arch magic, wrap namespace to avoid ODR conflicts
|
||||
-D__CUDA_ARCH_LIST__=700
|
||||
-DCUB_DISABLE_NAMESPACE_MAGIC
|
||||
-DCUB_WRAPPED_NAMESPACE=cccl_preload
|
||||
|
||||
@@ -41,8 +41,10 @@ static constexpr unsigned int ALLOC_MAX_BIN = 32; /* 4 GB */
|
||||
static constexpr size_t ALLOC_MAX_CACHED = (size_t)8 * 1024 * 1024 * 1024; /* 8GB */
|
||||
|
||||
/* ---- Global allocator singleton ---- */
|
||||
static cub::CachingDeviceAllocator& get_allocator() {
|
||||
static cub::CachingDeviceAllocator instance(
|
||||
using CubAllocator = CUB_NS_QUALIFIER::CachingDeviceAllocator;
|
||||
|
||||
static CubAllocator& get_allocator() {
|
||||
static CubAllocator instance(
|
||||
ALLOC_BIN_GROWTH,
|
||||
ALLOC_MIN_BIN,
|
||||
ALLOC_MAX_BIN,
|
||||
|
||||
Reference in New Issue
Block a user