Skip to main content

PartitionOutputPublisher

eredu_runtime::partitioned_execution

Trait PartitionOutputPublisher 

Source 
pub trait PartitionOutputPublisher<B, G, R, I>
where
    B: CommunicationBackend,
    G: Borrow<B::CommunicationGroup>,
    R: Borrow<B::CommunicationRoute>,
    I: CommunicationTensorMetadata<B>,
{
    const ENABLED: bool;

    // Required method
    fn publish(
        &mut self,
        communication: &PartitionCommunication<B, G, R, I>,
        value: B::Tensor,
        publication: PartitionOutputPublication,
        phase: DistributedExecutionPhase,
        executor: &B::Executor,
    ) -> Result<B::Tensor, PartitionExecutionError>;
}
Expand description

Additive policy for optional root-owned output publication.

Required Associated Constants§

Source

const ENABLED: bool

Whether this policy admits selected output publication.

Required Methods§

Source

fn publish( &mut self, communication: &PartitionCommunication<B, G, R, I>, value: B::Tensor, publication: PartitionOutputPublication, phase: DistributedExecutionPhase, executor: &B::Executor, ) -> Result<B::Tensor, PartitionExecutionError>

Publishes source data on root and destination storage on other members.

Dyn Compatibility§

This trait is not dyn compatible.

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

Implementors§