Struct ReplicatedTextMaterializationTask

Source

pub struct ReplicatedTextMaterializationTask { /* private fields */ }

Expand description

Exact backend work item for one selected logical parameter.

This value joins architecture-owned topology and artifact facts with the authoritative selected lowering. A materializer may batch these tasks, but it must not replace them with one model-wide transform choice.

Implementations§

Source §

impl ReplicatedTextMaterializationTask

Source

pub fn from_exact_source( name: impl Into<String>, physical_source: ReplicatedTextPhysicalSource, aliases: Vec<String>, physical_shape: Vec<usize>, logical_shape: Vec<usize>, role: ReplicatedTextParameterRole, owner: ReplicatedTextParameterOwner, executable: LinearFormat, lowering: WeightLoweringKind, lowering_descriptor: WeightLoweringDescriptor, ) -> Result<Self, ReplicatedTextContractError>

Creates one exact, source-backed materialization task selected outside the ordinary replicated-text session lifecycle.

This is used by architecture-owned auxiliary modules which share the same physical lowering contract but do not own a text session.

Source

pub fn with_output_companions( self, companions: Vec<ReplicatedTextOutputCompanion>, ) -> Result<Self, ReplicatedTextContractError>

Attaches the architecture’s exact selected packed-output companions.

Source

pub fn name(&self) -> &str

Returns the canonical logical parameter identity.

Source

pub fn sources(&self) -> &[String]

Returns every admitted physical source identity.

Source

pub fn physical_sources(&self) -> &[ReplicatedTextPhysicalSource]

Returns exact shard and translated-output provenance.

Source

pub fn aliases(&self) -> &[String]

Returns every architecture-admitted alias.

Source

pub fn shared_source_keys(&self) -> &BTreeSet<String>

Returns architecture-declared physical sources shared by logical outputs.

Source

pub fn permitted_native_source_dtypes(&self) -> &[RecipeDtype]

Returns exact source dtypes explicitly permitted for native binding.

Source

pub const fn source_encoding(&self) -> &SourceTensorEncoding

Returns the exact admitted source encoding.

Source

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

Returns the admitted physical source geometry.

Source

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

Returns the architecture-declared logical geometry.

Source

pub const fn role(&self) -> ReplicatedTextParameterRole

Returns the architecture-owned semantic parameter role.

Source

pub const fn owner(&self) -> &ReplicatedTextParameterOwner

Returns the architecture-owned module location.

Source

pub const fn presence(&self) -> &ReplicatedTextParameterPresence

Returns the exact admitted presence or derivation.

Source

pub const fn executable(&self) -> LinearFormat

Returns the selected executable format.

Source

pub const fn lowering(&self) -> WeightLoweringKind

Returns the selected backend lowering mechanism.

Source

pub const fn lowering_descriptor(&self) -> &WeightLoweringDescriptor

Returns the complete geometry-bearing lowering request.

Source

pub const fn derived_recipe(&self) -> Option<&DerivedWeightRecipe>

Returns the architecture-owned derivation, when this output is derived.

Source

pub const fn derived_output(&self) -> Option<&RecipeMetadata>

Returns admission-time metadata for the derived output.

Source

pub fn output_companions(&self) -> &[ReplicatedTextOutputCompanion]

Returns exact output companion identities declared by the architecture.

Source

pub fn source_recipe( &self, ) -> Result<DerivedWeightRecipe, ReplicatedTextContractError>

Returns the exact source recipe selected for this task.

Direct tasks are represented as a full selection of their single admitted source. Derived tasks return the architecture-owned recipe without reconstructing it from a checkpoint catalog.