Struct cubecl_core::channel::MutexComputeChannel

pub struct MutexComputeChannel<Server> { /* private fields */ }

The MutexComputeChannel ensures thread-safety by locking the server on every operation

Implementations

impl<Server> MutexComputeChannel<Server>

where Server: ComputeServer,

pub fn new(server: Server) -> MutexComputeChannel<Server>

Create a new mutex compute channel.

Trait Implementations

impl<S> Clone for MutexComputeChannel<S>

fn clone(&self) -> MutexComputeChannel<S>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<Server> ComputeChannel<Server> for MutexComputeChannel<Server>

where Server: ComputeServer,

fn read( &self, handle: Binding<Server>, ) -> Pin<Box<dyn Future<Output = Vec<u8>> + Send>>

Given a binding, returns owned resource as bytes

fn get_resource( &self, binding: Binding<Server>, ) -> <<Server as ComputeServer>::Storage as ComputeStorage>::Resource

Given a resource handle, return the storage resource.

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

Given a resource as bytes, stores it and returns the resource handle

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

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

unsafe fn execute( &self, kernel: <Server as ComputeServer>::Kernel, count: <Server as ComputeServer>::DispatchOptions, handles: Vec<Binding<Server>>, kind: ExecutionMode, )

Executes the kernel over the given bindings.

fn sync(&self, sync_type: SyncType)

Perform some synchronization of commands on the server.

impl<Server> Debug for MutexComputeChannel<Server>

where Server: Debug,

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

Formats the value using the given formatter.