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
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
# -*- Python -*- vim: set ft=python ts=4 sw=4 expandtab tw=79:
|
|
# Configuration file for the 'lit' test runner.
|
|
import os
|
|
import site
|
|
|
|
site.addsitedir(os.path.join(os.path.dirname(os.path.dirname(__file__)), 'utils'))
|
|
|
|
|
|
# Tell pylint that we know config and lit_config exist somewhere.
|
|
if 'PYLINT_IMPORT' in os.environ:
|
|
config = object()
|
|
lit_config = object()
|
|
|
|
# name: The name of this test suite.
|
|
config.name = 'libcu++'
|
|
|
|
# suffixes: A list of file extensions to treat as test files.
|
|
config.suffixes = ['.pass.cpp', '.fail.cpp', '.runfail.cpp', '.sh.cpp']
|
|
|
|
# test_source_root: The root path where tests are located.
|
|
config.test_source_root = os.path.dirname(__file__)
|
|
|
|
loaded_site_cfg = getattr(config, 'loaded_site_config', False)
|
|
if not loaded_site_cfg:
|
|
import libcudacxx.test.config
|
|
libcudacxx.test.config.loadSiteConfig(lit_config, config, 'libcudacxx_site_config',
|
|
'LIBCUDACXX_SITE_CONFIG')
|
|
|
|
# Infer the test_exec_root from the libcudacxx_object root.
|
|
obj_root = getattr(config, 'libcudacxx_obj_root', None)
|
|
|
|
# Check that the test exec root is known.
|
|
if obj_root is None:
|
|
obj_root = getattr(config, 'libcudacxx_obj_root', None)
|
|
if obj_root is None:
|
|
import tempfile
|
|
obj_root = tempfile.mkdtemp(prefix='libcudacxx-testsuite-')
|
|
lit_config.warning('Creating temporary directory for object root: %s' %
|
|
obj_root)
|
|
|
|
config.test_exec_root = os.path.join(obj_root, 'test')
|
|
|
|
cfg_variant = getattr(config, 'configuration_variant', 'libcudacxx')
|
|
if cfg_variant:
|
|
lit_config.note('Using configuration variant: %s' % cfg_variant)
|
|
|
|
# Load the Configuration class from the module name <cfg_variant>.test.config.
|
|
config_module_name = '.'.join([cfg_variant, 'test', 'config'])
|
|
config_module = __import__(config_module_name, fromlist=['Configuration'])
|
|
|
|
configuration = config_module.Configuration(lit_config, config)
|
|
configuration.configure()
|
|
configuration.print_config_info()
|
|
config.test_format = configuration.get_test_format()
|