Struct burn_compute::client::ComputeClient

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

The ComputeClient is the entry point to require tasks from the ComputeServer. It should be obtained for a specific device via the Compute struct.

Implementations

impl<Server, Channel> ComputeClient<Server, Channel>where Server: ComputeServer, Channel: ComputeChannel<Server>,

pub fn new(channel: Channel, tuner: Arc<Mutex<Tuner<Server, Channel>>>) -> Self

Create a new client.

pub fn read(&self, handle: &Handle<Server>) -> Reader<Vec<u8>>

Given a handle, returns owned resource as bytes.

pub fn create(&self, data: &[u8]) -> Handle<Server>

Given a resource, stores it and returns the resource handle.

pub fn empty(&self, size: usize) -> Handle<Server>

Reserves size bytes in the storage, and returns a handle over them.

pub fn execute(&self, kernel: Server::Kernel, handles: &[&Handle<Server>])

Executes the kernel over the given handles.

pub fn sync(&self)

Wait for the completion of every task in the server.

pub fn execute_autotune( &self, autotune_operation_set: Box<dyn AutotuneOperationSet<Server::AutotuneKey>> )

Executes the fastest kernel in the autotune operation, using (cached) runtime benchmarks

Trait Implementations

impl<S, C> Clone for ComputeClient<S, C>where S: ComputeServer, C: ComputeChannel<S>,

fn clone(&self) -> Self

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<Server: Debug + ComputeServer, Channel: Debug> Debug for ComputeClient<Server, Channel>

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

Formats the value using the given formatter.