Skip to main content

ShardProtocol

Trait ShardProtocol 

Source
pub trait ShardProtocol: Send + 'static {
    type Msg: Send + 'static;

    // Required method
    fn make_step(
        input_slice: Vec<f32>,
        reply: Sender<Result<ShardStepResult, GpuError>>,
    ) -> Self::Msg;
}

Required Associated Types§

Source

type Msg: Send + 'static

Required Methods§

Source

fn make_step( input_slice: Vec<f32>, reply: Sender<Result<ShardStepResult, GpuError>>, ) -> Self::Msg

Per-shard step: takes a slice of input + the local weight shard, returns (partial_output, partial_loss, partial_grad_norm).

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§