Enum LayerShapes 

pub enum LayerShapes {
    Uniform,
    PerLayer(Vec<LayerShape>),
}

Every layer’s shape, or the statement that they all agree.

Uniform is not PerLayer(vec![same; n]): the fused Metal stacks, the CUDA resident KV and the slot-file format each hold ONE geometry, and “is this model uniform” is a question they ask through Self::is_uniform rather than by comparing entries.

Variants

Uniform

Every layer is ModelConfig::{n_heads, n_kv_heads, moe.expert_ffn_dim}.

PerLayer(Vec<LayerShape>)

One entry per layer, at least two of which differ.

Implementations

impl LayerShapes

pub fn is_uniform(&self) -> bool

pub fn replicated(self, n_loops: usize) -> Self

The same shapes for n_loops passes over the layers, as nanbeige.cpp:24-26 copies each physical layer’s arrays to every logical slot (crate::layer_loops). Uniform stays uniform.

pub fn resolve( arch: &str, heads: &[u64], kv_heads: &[u64], ffn: Option<&[u64]>, expert_ffn_dim: usize, recurrent: Option<&RecurrentMask>, ) -> Result<Self, LoadError>

Builds the table from the three per-layer arrays, collapsing to Uniform when nothing varies and refusing a varying file for an architecture llama.cpp itself reads at layer 0.

ffn may be absent (a MoE file that declares only expert_feed_forward_length); then every layer takes expert_ffn_dim.

recurrent is crate::gdn::recurrent_layers’ answer: the layers that run a recurrent block AND which block, on an architecture whose head counts are uniform and say nothing about either.

Trait Implementations

impl Clone for LayerShapes

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LayerShapes

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

Formats the value using the given formatter.

impl Default for LayerShapes

fn default() -> Self

Returns the “default value” for a type.

impl Eq for LayerShapes

impl PartialEq for LayerShapes

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for LayerShapes