Struct LocalTensorLayout 

pub struct LocalTensorLayout<P = TensorPlacement> { /* private fields */ }

Rank-local shape and placement for one planned physical tensor.

Implementations

impl<P> LocalTensorLayout<P>

pub fn new( logical_name: impl Into<String>, role: ParameterRole, global_shape: Vec<usize>, local_shape: Vec<usize>, placement: P, logical_units: Option<usize>, logical_range: Option<Range<usize>>, fell_back_to_replication: bool, ) -> Self

Creates one validated-planner output entry.

pub fn logical_name(&self) -> &str

Returns the logical parameter group name.

pub const fn role(&self) -> ParameterRole

Returns the semantic parameter role.

pub fn global_shape(&self) -> &[usize]

Returns the checkpoint-global shape.

pub fn local_shape(&self) -> &[usize]

Returns the shape materialized on this rank.

pub const fn placement(&self) -> &P

Returns the backend-realized placement.

pub fn additional_placements(&self) -> &[TensorPlacement]

Returns independent checkpoint-global selections applied before the primary placement.

This is used when distinct parallel axes own distinct tensor axes, such as EP selection of packed experts followed by TP selection of each expert matrix. Empty means the primary placement is complete.

pub fn with_additional_placement(self, placement: TensorPlacement) -> Self

Adds one exact checkpoint-global selection preceding the primary placement.

pub fn logical_range(&self) -> Option<&Range<usize>>

Returns the rank-local range in the parameter group’s semantic domain.

pub const fn logical_units(&self) -> Option<usize>

Returns the size of the complete semantic partition domain.

pub const fn fell_back_to_replication(&self) -> bool

Returns whether permissive planning replicated an unsupported shard.

Trait Implementations

impl<P: Clone> Clone for LocalTensorLayout<P>

fn clone(&self) -> LocalTensorLayout<P>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<P: Debug> Debug for LocalTensorLayout<P>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<P: Eq> Eq for LocalTensorLayout<P>

impl<P: PartialEq> PartialEq for LocalTensorLayout<P>

fn eq(&self, other: &LocalTensorLayout<P>) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl<P: PartialEq> StructuralPartialEq for LocalTensorLayout<P>