Struct DeviceFusionRuntime 

pub struct DeviceFusionRuntime<R: DeviceRuntime> { /* private fields */ }

Available on crate feature fusion only.

Fusion runtime for JIT runtimes.

Trait Implementations

impl<R: Debug + DeviceRuntime> Debug for DeviceFusionRuntime<R>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<R: DeviceRuntime> FusionRuntime for DeviceFusionRuntime<R>

type OptimizationState = RudaOptimizationState

The state that can be serialized for an optimization.

type Optimization = RudaOptimization<R>

Optimization type for the backend.

type FusionHandle = RudaFusionHandle<R>

Handle used to store tensor dynamically.

type FusionDevice = <R as DeviceRuntime>::RudaDevice

Device used by the runtime.

fn fusers(device: R::Device) -> Vec<Box<dyn OperationFuser<Self::Optimization>>>

The list of fusers that will be used to optimize the computational graph.

impl<R> Optimization<DeviceFusionRuntime<R>> for RudaOptimization<R>

where R: DeviceRuntime,

fn execute( &mut self, context: &mut Context<<DeviceFusionRuntime<R> as FusionRuntime>::FusionHandle>, execution: &OrderedExecution<DeviceFusionRuntime<R>>, )

Execute the optimization.

fn to_state(&self) -> RudaOptimizationState

Returns the state that can be serialized.

fn from_state(device: &R::Device, state: RudaOptimizationState) -> Self

Create the optimization from the state.