Struct burn::backend::wgpu::WgpuRuntime

pub struct WgpuRuntime<C = WgslCompiler>(/* private fields */)
where
    C: WgpuCompiler;

Runtime that uses the [wgpu] crate with the wgsl compiler. This is used in the Wgpu backend. For advanced configuration, use init_sync to pass in runtime options or to select a specific graphics API.

Trait Implementations

impl<C> Debug for WgpuRuntime<C>

where C: Debug + WgpuCompiler,

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

Formats the value using the given formatter.

impl Runtime for WgpuRuntime

type Compiler = WgslCompiler

The compiler used to compile the inner representation into tokens.

type Server = WgpuServer<WgslCompiler>

The compute server used to run kernels and perform autotuning.

type Channel = MutexComputeChannel<WgpuServer<WgslCompiler>>

The channel used to communicate with the compute server.

type Device = WgpuDevice

The device used to retrieve the compute client.

fn client( device: &<WgpuRuntime as Runtime>::Device, ) -> ComputeClient<<WgpuRuntime as Runtime>::Server, <WgpuRuntime as Runtime>::Channel>

Retrieve the compute client from the runtime device.

fn name() -> &'static str

The runtime name.

fn supported_line_sizes() -> &'static [u8]

Returns the supported line sizes for the current runtime’s compiler.

fn require_array_lengths() -> bool

Return true if global input array lengths should be added to kernel info.