Skip to main content

VariableAllToAllBackend

Trait VariableAllToAllBackend 

Source
pub trait VariableAllToAllBackend: CommunicationBackend {
    // Required method
    fn variable_all_to_all(
        value: Self::Tensor,
        counts: &CommunicationPeerCounts,
        axis: usize,
        group: &Self::CommunicationGroup,
        executor: &Self::Executor,
    ) -> Result<Submission<Self::Tensor, Self::CommunicationCompletion>, Self::CommunicationError>;
}
Expand description

Variable-count exchange on an opaque communication group.

Required Methods§

Source

fn variable_all_to_all( value: Self::Tensor, counts: &CommunicationPeerCounts, axis: usize, group: &Self::CommunicationGroup, executor: &Self::Executor, ) -> Result<Submission<Self::Tensor, Self::CommunicationCompletion>, Self::CommunicationError>

Exchanges exact per-peer partitions on axis and returns exact completion.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors§