Enum LayerwiseTraversalRuntime 

pub enum LayerwiseTraversalRuntime<Direct, Partitioned> {
    Direct(Direct),
    Partitioned(Partitioned),
}

Neutral choice between an ordinary local traversal and the same traversal installed in a selected partition runtime.

Concrete backends construct one of these alternatives but do not retain a parallel execution enum or select a separate forward implementation.

Variants

Direct(Direct)

Replicated local traversal.

Partitioned(Partitioned)

Architecture-selected partition traversal.

Implementations

impl<Direct, Partitioned> LayerwiseTraversalRuntime<Direct, Partitioned>

pub const fn direct(runtime: Direct) -> Self

Wraps an ordinary local layered runtime.

pub const fn partitioned(runtime: Partitioned) -> Self

Wraps an architecture-selected partition runtime.

impl<A, B, S, ExecutionPolicy, G, R, I, T, U, V> LayerwiseTraversalRuntime<LayerwiseRuntime<A, B, S, ExecutionPolicy>, Box<PartitionedTextRuntime<A, B, S, (), LayerwiseTraversalPartitionExecutor<A, B, S, ExecutionPolicy>, G, R, I, T, U, V>>>

where B: CommunicationBackend<Executor = <<B as NeuralBackend>::Tensor as Tensor>::Context>, S: RuntimeState<B>, A: ParallelLayeredArchitecture<B, S>, ExecutionPolicy: LayerwisePolicy<B, A::Unit>, G: Borrow<B::CommunicationGroup>, R: Borrow<B::CommunicationRoute>, I: CommunicationTensorMetadata<B>, T: PartitionBoundaryTransport<B, G, R, I>, U: PartitionOutputPublisher<B, G, R, I>, V: PartitionCommitAgreement<B, G, R, I>, B::ParallelContext: Sized, A::Error: Display, ExecutionPolicy::Error: Display,

pub fn forward_with_traversal_hook<'a, H>( &mut self, input: A::Input<'a>, state: &mut S, context: &<B::Tensor as Tensor>::Context, hook: &mut H, ) -> PartitionedTraversalResult<B::Tensor, A::ForwardContext, A::Error, ExecutionPolicy::Error>

where H: LayeredTraversalHook<B, A::ForwardContext, A::Error> + ?Sized,

Runs the selected local traversal without exposing its realization kind to concrete composition.

pub fn policy(&self) -> &ExecutionPolicy

Borrows the selected residency policy independently of realization kind.

pub fn architecture(&self) -> &A

Borrows the selected local architecture independently of realization kind.