![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <tensor_fill.h>

Classes | |
| struct | Params |
| Parameters structure. More... | |
Public Types | |
| using | FloatType = typename std::conditional<(sizeof(Element) > 4), double, float >::type |
| using | IntType = typename std::conditional<(sizeof(Element) > 4), int64_t, int >::type |
Public Member Functions | |
| CUTLASS_DEVICE | RandomGaussianFunc (Params const ¶ms) |
| Device-side initialization of RNG. More... | |
| CUTLASS_DEVICE Element | operator() () |
| Compute random value and update RNG state. More... | |
Public Attributes | |
| Params | params |
| Parameters object. More... | |
| curandState_t | rng_state |
| RNG state object. More... | |
| using cutlass::reference::device::detail::RandomGaussianFunc< Element >::FloatType = typename std::conditional<(sizeof(Element) > 4), double, float>::type |
| using cutlass::reference::device::detail::RandomGaussianFunc< Element >::IntType = typename std::conditional<(sizeof(Element) > 4), int64_t, int>::type |
|
inline |
|
inline |
| Params cutlass::reference::device::detail::RandomGaussianFunc< Element >::params |
| curandState_t cutlass::reference::device::detail::RandomGaussianFunc< Element >::rng_state |
1.8.11