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 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>