Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::CpuExecutionContext Class Referenceexport

CPU execution context specialization. More...

Inheritance diagram for Mila::Dnn::Compute::CpuExecutionContext:
Mila::Dnn::Compute::IExecutionContext

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.

Detailed Description

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.

Constructor & Destructor Documentation

◆ CpuExecutionContext()

Mila::Dnn::Compute::CpuExecutionContext::CpuExecutionContext ( DeviceId device_id = Device::Cpu())
inlineexplicit

Constructs CPU execution context.

Parameters
device_idCPU device identifier (typically Device::Cpu()).

Member Function Documentation

◆ getDeviceId()

DeviceId Mila::Dnn::Compute::CpuExecutionContext::getDeviceId ( ) const
inlinenodiscardoverridevirtualnoexcept

Gets the device identifier.

Returns
DeviceId The CPU device identifier.

Implements Mila::Dnn::Compute::IExecutionContext.

◆ synchronize()

void Mila::Dnn::Compute::CpuExecutionContext::synchronize ( )
inlineoverridevirtual

Synchronizes CPU execution (no-op).

CPU operations are synchronous, so this is a no-op.

Implements Mila::Dnn::Compute::IExecutionContext.


The documentation for this class was generated from the following file: