Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
CudaTensorOps.Random.ixx File Reference

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

Detailed Description

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.