Crate burn_router

Burn multi-backend router.

Modules

duo

Module containing the essential types for multi-backend operations.

quad

Module containing the essential types for multi-backend operations.

trio

Module containing the essential types for multi-backend operations.

Macros

binary_bool_ops

binary_float_cmp_ops

binary_float_ops

binary_int_cmp_ops

binary_int_ops

int_float_ops

reduce_float2int_dim_ops

reduce_float_dim_ops

reduce_int_dim_ops

scalar_float2int_ops

scalar_float_cmp_ops

scalar_float_dim_ops

scalar_float_ops

scalar_int_cmp_ops

scalar_int_dim_ops

scalar_int_ops

unary_float_ops

unary_int_ops

Structs

BackendRouter

A backend that forwards the tensor operations to the appropriate backend (given multiple backends).

ByteBridge

Simply transfers tensors between backends via the underlying tensor data.

DirectChannel

A local channel with direct connection to the backend runner clients.

RouterTensor

Tensor primitive for the router backend.

Runner

A runner is responsible for executing tensor operations for a given intermediate backend.

RunnerContext

A runner’s context contains a handle container to manage (i.e., fetch and update) existing tensors.

Traits

MultiBackendBridge

Allows tensors to be transferred between multiple backends.

RunnerChannel

Defines the connection channel and operations for a setup with multiple backend runner clients.

RunnerClient

Define how to interact with the runner.

Functions

get_client

Get the client for the given device

Type Aliases

Client

Type alias for <R as RunnerChannel>::Client.

DirectByteChannel

A local channel with a simple byte bridge between backends. It transfers tensors between backends via the underlying tensor data.

Router

Router backend.

TensorHandle

Type alias for <Br as MultiBackendBridge>::TensorHandle.