Struct burn_compute::ComputeRuntime

pub struct ComputeRuntime<Device, Server: ComputeServer, Channel> { /* private fields */ }

The compute type has the responsibility to retrieve the correct compute client based on the given device.

Implementations

impl<Device, Server, Channel> ComputeRuntime<Device, Server, Channel>

where Device: Hash + PartialEq + Eq + Clone + Debug, Server: ComputeServer, Channel: ComputeChannel<Server>,

pub const fn new() -> Self

Create a new compute.

pub fn client<Init>( &self, device: &Device, init: Init ) -> ComputeClient<Server, Channel>

where Init: Fn() -> ComputeClient<Server, Channel>,

Get the compute client for the given device.

Provide the init function to create a new client if it isn’t already initialized.

pub fn register(&self, device: &Device, client: ComputeClient<Server, Channel>)

Register the compute client for the given device.

Note

This function is mostly useful when the creation of the compute client can’t be done synchronously and require special context.

Panics

If a client is already registered for the given device.