![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|

Go to the source code of this file.
Namespaces | |
| cutlass | |
| cutlass::reference | |
| cutlass::reference::device | |
| cutlass::reference::device::kernel | |
Functions | |
| template<typename T > | |
| __global__ void | cutlass::reference::device::kernel::TensorInitializeUniform (Distribution dist, int64_t seed, int dim_contiguous, int dim_strided, T *tensor, int ldm) |
| Kernel to initialize tensor to uniform random distribution. More... | |
| template<typename T > | |
| __global__ void | cutlass::reference::device::kernel::TensorInitializeGaussian (Distribution dist, int64_t seed, int dim_contiguous, int dim_strided, T *tensor, int ldm) |
| Kernel to initialize tensor to uniform distribution. More... | |
| template<typename T > | |
| __global__ void | cutlass::reference::device::kernel::TensorInitializeLinear (Distribution dist, int64_t seed, int dim_contiguous, int dim_strided, T *tensor, int ldm) |
| Kernel to initialize tensor to an identity matrix. More... | |
| template<typename T > | |
| __global__ void | cutlass::reference::device::kernel::TensorInitializeIdentity (Distribution dist, int64_t seed, int dim_contiguous, int dim_strided, T *tensor, int ldm) |
| Kernel to initialize tensor to an identity matrix. More... | |
1.8.11