Struct PartitionState 

pub struct PartitionState { /* private fields */ }

Rank-local mutable-state geometry and its architecture-global layer range.

Implementations

impl PartitionState

pub fn new( layout: StateLayout, global_layer_offset: usize, ) -> Result<Self, ArchitecturePartitionError>

Attaches a local state layout at one architecture-global layer offset.

pub const fn layout(&self) -> &StateLayout

Returns the exact rank-local state layout.

pub const fn global_layer_offset(&self) -> usize

Returns the first architecture-global layer represented by the layout.

pub fn global_layers(&self) -> Range<usize>

Returns the architecture-global state-layer range.

pub fn prompt_cache_identity<B, M>( &self, architecture: &M, topology: PromptCacheTopology, ) -> Result<PromptCacheModelIdentity, ArchitecturePartitionError>

where B: NeuralBackend, M: ArchitectureParameters<B>, M::DefinitionError: Display,

Derives prompt-cache identity from this canonical state partition.

Trait Implementations

impl Clone for PartitionState

fn clone(&self) -> PartitionState

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PartitionState

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

Formats the value using the given formatter.

impl Eq for PartitionState

impl PartialEq for PartitionState

fn eq(&self, other: &PartitionState) -> bool

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for PartitionState