Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Compute.ExecutionContext Module Reference

Exported Modules

module  Compute.IExecutionContext
module  Compute.DeviceType
module  Compute.ExecutionContextTemplate
module  Compute.DeviceId

Typedefs

using Mila::Dnn::Compute::CpuExecutionContext = ExecutionContext<DeviceType::Cpu>

Functions

template<DeviceType TDeviceType>
const ExecutionContext< TDeviceType > * Mila::Dnn::Compute::cast_context_ (const IExecutionContext *ctx) noexcept
template<DeviceType TDeviceType>
ExecutionContext< TDeviceType > * Mila::Dnn::Compute::cast_context_ (IExecutionContext *ctx) noexcept
template<DeviceType TDeviceType>
ExecutionContext< TDeviceType > * Mila::Dnn::Compute::validateExecutionContext_ (IExecutionContext *context, const std::string &op_name)

Files

file  /__w/Mila/Mila/Mila/Src/Dnn/Compute/ExecutionContext.ixx
 Templated execution context framework for compute operations and stream management.
file  /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cpu/CpuExecutionContext.ixx
 CPU-specific execution context specialization.
file  /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/CudaExecutionContext.ixx
 CUDA-specific execution context specialization.