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

Exported Modules

module  Compute.IExecutionContext
module  Compute.ExecutionContextTraits
module  Compute.ExecutionContextTraits.Cpu
module  Compute.DeviceId
module  Compute.CpuExecutionContext
module  Compute.DeviceType

Typedefs

template<DeviceType TDeviceType>
using Mila::Dnn::Compute::ExecutionContext = typename ExecutionContextTraits<TDeviceType>::type
 The execution context for a device type.

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  Mila/Src/Dnn/Compute/ExecutionContext.ixx
 Execution context framework for compute operations and stream management.