Trait JitRuntime

pub trait JitRuntime: Runtime<Device = Self::JitDevice, Server = Self::JitServer> {
    type JitDevice: DeviceOps;
    type JitServer: ComputeServer<Kernel = Box<dyn CubeTask<Self::Compiler>>, Feature = Feature>;
}

Just-in-Time runtime extending the cube runtime.

Required Associated Types

type JitDevice: DeviceOps

The device that should also implement [burn_tensor::backend::DeviceOps].

type JitServer: ComputeServer<Kernel = Box<dyn CubeTask<Self::Compiler>>, Feature = Feature>

The cube server with the [JitAutotuneKey].

Object Safety

This trait is not object safe.

Implementors

impl<R> JitRuntime for R

where R: Runtime, <R as Runtime>::Device: DeviceOps,

type JitDevice = <R as Runtime>::Device

type JitServer = <R as Runtime>::Server