Struct PlacementPlan 

pub struct PlacementPlan { /* private fields */ }

Exact logical checkpoint placement for one rank.

Implementations

impl PlacementPlan

pub const fn new(rank: PlacementRank) -> Self

Creates a strict plan in which every checkpoint source must be named.

pub fn replicated(rank: PlacementRank) -> Self

Creates a plan that replicates every checkpoint source.

pub const fn rank(&self) -> PlacementRank

Returns the selected logical rank.

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

Sets the placement for otherwise unnamed checkpoint sources.

pub fn insert(&mut self, source: impl Into<String>, placement: TensorPlacement)

Adds or replaces one exact source placement.

pub fn insert_expected( &mut self, source: impl Into<String>, expected_source_shape: impl Into<Vec<usize>>, placement: TensorPlacement, ) -> Result<(), PlacementPlanError>

Adds one source placement with an exact pre-selection shape.

pub fn insert_quantized_companions( &mut self, prefix: &str, placement: TensorPlacement, has_biases: bool, )

Adds a packed weight and its scale and optional bias companions together.

pub fn placement(&self, source: &str) -> Option<&TensorPlacement>

Returns an explicit placement by exact source name.

pub fn validate(&self) -> Result<(), PlacementPlanError>

Validates all geometry available before checkpoint access.

pub fn potentially_local( &self, source: &str, ) -> Result<bool, PlacementPlanError>

Returns whether a named source can require materialization on this rank.

pub fn resolve( &self, source: &str, shape: &[usize], ) -> Result<ResolvedTensorPlacement, PlacementPlanError>

Resolves one named source against its admitted logical shape.

pub fn validate_loaded_sources( &self, loaded: &BTreeSet<String>, unexpected: Vec<String>, ) -> Result<(), PlacementPlanError>

Verifies exact locally required and unexpected source coverage.

Trait Implementations

impl Clone for PlacementPlan

fn clone(&self) -> PlacementPlan

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PlacementPlan

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

Formats the value using the given formatter.