|
Mila
Deep Neural Network Library
|
CPU execution context specialization. More...
Public Member Functions | |
| CpuExecutionContext (const CpuExecutionContext &)=delete | |
| CpuExecutionContext (CpuExecutionContext &&)=delete | |
| CpuExecutionContext (DeviceId device_id=Device::Cpu()) | |
| Constructs CPU execution context. | |
| DeviceId | getDeviceId () const noexcept override |
| Gets the device identifier. | |
| CpuExecutionContext & | operator= (const CpuExecutionContext &)=delete |
| CpuExecutionContext & | operator= (CpuExecutionContext &&)=delete |
| void | synchronize () override |
| Synchronizes CPU execution (no-op). | |
| Public Member Functions inherited from Mila::Dnn::Compute::IExecutionContext | |
| const ActivationObserver & | getActivationObserver () const noexcept |
| bool | hasActivationObserver () const noexcept |
| virtual void | releaseLoadStaging () noexcept |
| Free the device buffer a load staged full-precision weights through. | |
| virtual void | reserveScratch (std::size_t bytes) |
| Reserve the forward scratch buffer at its final size. | |
| void | setActivationObserver (ActivationObserver observer) |
| Install the observer components publish their activations to. | |
CPU execution context specialization.
Minimal execution context for CPU operations. CPU execution does not require streams or library handles, so this implementation provides the interface without additional overhead.
|
inlineexplicit |
Constructs CPU execution context.
| device_id | CPU device identifier (typically Device::Cpu()). |
|
inlinenodiscardoverridevirtualnoexcept |
Gets the device identifier.
Implements Mila::Dnn::Compute::IExecutionContext.
|
inlineoverridevirtual |
Synchronizes CPU execution (no-op).
CPU operations are synchronous, so this is a no-op.
Implements Mila::Dnn::Compute::IExecutionContext.