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
106 lines
3.6 KiB
C++
106 lines
3.6 KiB
C++
// SPDX-FileCopyrightText: Copyright (c) 2008-2013, NVIDIA Corporation. All rights reserved.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
/*! \file random.h
|
|
* \brief Pseudo-random number generators.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <thrust/detail/config.h>
|
|
|
|
#if defined(_CCCL_IMPLICIT_SYSTEM_HEADER_GCC)
|
|
# pragma GCC system_header
|
|
#elif defined(_CCCL_IMPLICIT_SYSTEM_HEADER_CLANG)
|
|
# pragma clang system_header
|
|
#elif defined(_CCCL_IMPLICIT_SYSTEM_HEADER_MSVC)
|
|
# pragma system_header
|
|
#endif // no system header
|
|
#include <cuda/std/cstdint>
|
|
|
|
// RNGs
|
|
#include <thrust/random/discard_block_engine.h>
|
|
#include <thrust/random/linear_congruential_engine.h>
|
|
#include <thrust/random/linear_feedback_shift_engine.h>
|
|
#include <thrust/random/subtract_with_carry_engine.h>
|
|
#include <thrust/random/xor_combine_engine.h>
|
|
|
|
// distributions
|
|
#include <thrust/random/normal_distribution.h>
|
|
#include <thrust/random/uniform_int_distribution.h>
|
|
#include <thrust/random/uniform_real_distribution.h>
|
|
|
|
THRUST_NAMESPACE_BEGIN
|
|
|
|
/*! \addtogroup random Random Number Generation
|
|
* \{
|
|
*/
|
|
|
|
/*! \namespace thrust::random
|
|
* \brief \p thrust::random is the namespace which contains random number engine class templates,
|
|
* random number engine adaptor class templates, engines with predefined parameters,
|
|
* and random number distribution class templates. They are provided in a separate namespace
|
|
* for import convenience but are also aliased in the top-level \p thrust namespace for
|
|
* easy access.
|
|
*/
|
|
namespace random
|
|
{
|
|
/*! \addtogroup predefined_random Random Number Engines with Predefined Parameters
|
|
* \ingroup random
|
|
* \{
|
|
*/
|
|
|
|
/*! \typedef ranlux24
|
|
* \brief A random number engine with predefined parameters which implements the
|
|
* RANLUX level-3 random number generation algorithm.
|
|
* \note The 10000th consecutive invocation of a default-constructed object of type \p ranlux24
|
|
* shall produce the value \c 9901578 .
|
|
*/
|
|
using ranlux24 = discard_block_engine<ranlux24_base, 223, 23>;
|
|
|
|
/*! \typedef ranlux48
|
|
* \brief A random number engine with predefined parameters which implements the
|
|
* RANLUX level-4 random number generation algorithm.
|
|
* \note The 10000th consecutive invocation of a default-constructed object of type \p ranlux48
|
|
* shall produce the value \c 88229545517833 .
|
|
*/
|
|
using ranlux48 = discard_block_engine<ranlux48_base, 389, 11>;
|
|
|
|
/*! \typedef taus88
|
|
* \brief A random number engine with predefined parameters which implements
|
|
* L'Ecuyer's 1996 three-component Tausworthe random number generator.
|
|
*
|
|
* \note The 10000th consecutive invocation of a default-constructed object of type \p taus88
|
|
* shall produce the value \c 3535848941 .
|
|
*/
|
|
using taus88 =
|
|
xor_combine_engine<linear_feedback_shift_engine<std::uint32_t, 32U, 31U, 13U, 12U>,
|
|
0,
|
|
xor_combine_engine<linear_feedback_shift_engine<std::uint32_t, 32U, 29U, 2U, 4U>,
|
|
0,
|
|
linear_feedback_shift_engine<std::uint32_t, 32U, 28U, 3U, 17U>,
|
|
0>,
|
|
0>;
|
|
|
|
/*! \typedef default_random_engine
|
|
* \brief An implementation-defined "default" random number engine.
|
|
* \note \p default_random_engine is currently an alias for \p minstd_rand, and may change
|
|
* in a future version.
|
|
*/
|
|
using default_random_engine = minstd_rand;
|
|
|
|
/*! \} // end predefined_random
|
|
*/
|
|
} // namespace random
|
|
|
|
/*! \} // end random
|
|
*/
|
|
|
|
// import names into thrust::
|
|
using random::default_random_engine;
|
|
using random::ranlux24;
|
|
using random::ranlux48;
|
|
using random::taus88;
|
|
|
|
THRUST_NAMESPACE_END
|