|
Mila 0.13.48
Deep Neural Network Library
|
CUDA-specific execution context specialization. More...
#include <cuda_runtime.h>#include <cublasLt.h>#include <curand.h>#include <memory>#include <string>#include <format>#include <stdexcept>import Compute.DeviceType;import Core.RandomGenerator;import Compute.DeviceId;import Compute.IExecutionContext;import Compute.ExecutionContextTemplate;Classes | |
| class | Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda > |
| CUDA execution context specialization. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Typedefs | |
| using | Mila::Dnn::Compute::CudaExecutionContext = ExecutionContext<DeviceType::Cuda> |
CUDA-specific execution context specialization.