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.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<R: Runtime> JitRuntime for R

where R::Device: DeviceOps,

type JitDevice = <R as Runtime>::Device

type JitServer = <R as Runtime>::Server