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

Functions

template<DeviceType TDeviceType>
std::shared_ptr< DeviceContext > Mila::Dnn::Compute::CreateCompatibleContext ()
 Creates a device context compatible with the specified device type.
 
template<DeviceType TDeviceType>
std::shared_ptr< DeviceContext > Mila::Dnn::Compute::ValidateContext (std::shared_ptr< DeviceContext > context)
 Validates that the provided context is compatible with the specified device type.
 

Files

file  /home/runner/work/Mila/Mila/Mila/Src/Dnn/Compute/Devices/DeviceContextHelpers.ixx
 Utility functions for device context validation.