Struct MutexFusionClient

pub struct MutexFusionClient<R: FusionRuntime> { /* private fields */ }

Use a mutex to communicate with the fusion server.

Trait Implementations

impl<R> Clone for MutexFusionClient<R>

where R: FusionRuntime,

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<R> FusionClient<R> for MutexFusionClient<R>

where R: FusionRuntime<FusionClient = Self> + 'static,

fn new(device: FusionDevice<R>) -> Self

Create a new client for the given device.

fn register<O>( &self, streams: OperationStreams, repr: OperationIr, operation: O, )

where O: Operation<R> + 'static,

Register a new tensor operation intermediate representation.

fn drain(&self)

Register all lazy computation.

fn tensor_uninitialized( &self, shape: Vec<usize>, dtype: DType, ) -> FusionTensor<R>

Create a new fusion tensor, but with no resources allocated to it.

fn device(&self) -> &FusionDevice<R>

Get the current device used by all operations handled by this client.

fn register_tensor( &self, handle: FusionHandle<R>, shape: Vec<usize>, stream: StreamId, dtype: DType, ) -> FusionTensor<R>

Create a tensor with the given handle and shape.

fn read_tensor_float<B>( self, tensor: TensorIr, stream: StreamId, ) -> impl Future<Output = TensorData> + Send

where B: FusionBackend<FusionRuntime = R>,

Read the values contained by a float tensor.

fn read_tensor_int<B>( self, tensor: TensorIr, id: StreamId, ) -> impl Future<Output = TensorData> + Send

where B: FusionBackend<FusionRuntime = R>,

Read the values contained by an int tensor.

fn read_tensor_bool<B>( self, tensor: TensorIr, stream: StreamId, ) -> impl Future<Output = TensorData> + Send

where B: FusionBackend<FusionRuntime = R>,

Read the values contained by a bool tensor.

fn read_tensor_quantized<B>( self, tensor: TensorIr, stream: StreamId, ) -> impl Future<Output = TensorData> + Send

where B: FusionBackend<FusionRuntime = R>,

Read the values contained by a quantized tensor.

fn change_client_float<B>( &self, tensor: TensorIr, client: Self, stream: StreamId, ) -> FusionTensor<R>

where B: FusionBackend<FusionRuntime = R>,

Change the client of the given float tensor.

fn change_client_int<B>( &self, tensor: TensorIr, client: Self, stream: StreamId, ) -> FusionTensor<R>

where B: FusionBackend<FusionRuntime = R>,

Change the client of the given int tensor.

fn change_client_bool<B>( &self, tensor: TensorIr, client: Self, stream: StreamId, ) -> FusionTensor<R>

where B: FusionBackend<FusionRuntime = R>,

Change the client of the given bool tensor.

fn change_client_quantized<B>( &self, tensor: TensorIr, client: Self, stream: StreamId, ) -> FusionTensor<R>

where B: FusionBackend<FusionRuntime = R>,

Change the client of the given quantized tensor.

fn resolve_tensor_float<B>( &self, tensor: FusionTensor<R>, ) -> B::FloatTensorPrimitive

where B: FusionBackend<FusionRuntime = R>,

Resolve the given float tensor to a primitive tensor.

fn resolve_tensor_int<B>( &self, tensor: FusionTensor<R>, ) -> B::IntTensorPrimitive

where B: FusionBackend<FusionRuntime = R>,

Resolve the given int tensor to a primitive tensor.

fn resolve_tensor_bool<B>( &self, tensor: FusionTensor<R>, ) -> B::BoolTensorPrimitive

where B: FusionBackend<FusionRuntime = R>,

Resolve the given bool tensor to a primitive tensor.