|
Mila 0.13.48
Deep Neural Network Library
|
CUDA random initialization partition for tensor buffers. More...
#include <cuda_runtime.h>#include <curand.h>#include <stdexcept>#include <string>#include "Kernels/Random.h"import Core.RandomGenerator;import Compute.DeviceType;import Dnn.TensorDataTypeTraits;import Compute.DeviceId;import Dnn.TensorDataType;import Cuda.Error;import Dnn.ITensor;import Compute.IExecutionContext;import Compute.ExecutionContext;import Dnn.Tensor;Classes | |
| struct | Mila::Dnn::Compute::Cuda::RandomOps |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
| namespace | Mila::Dnn::Compute::Cuda |
CUDA random initialization partition for tensor buffers.
Provides device-dispatched normal and uniform random initialization using cuRAND. Reuses the cached cuRAND generator from CudaExecutionContext when provided, avoiding per-call generator creation overhead.