GlmVitBlock

metrale_model_layers::layers::glm_vit

Struct GlmVitBlock

pub struct GlmVitBlock {Show 14 fields
    pub norm1_w: DevicePtr,
    pub qkv_w: DevicePtr,
    pub qkv_b: DevicePtr,
    pub q_norm_w: DevicePtr,
    pub k_norm_w: DevicePtr,
    pub proj_w: DevicePtr,
    pub proj_b: DevicePtr,
    pub norm2_w: DevicePtr,
    pub gate_w: DevicePtr,
    pub gate_b: DevicePtr,
    pub up_w: DevicePtr,
    pub up_b: DevicePtr,
    pub down_w: DevicePtr,
    pub down_b: DevicePtr,
}

Expand description

2026-09-25: One ViT block’s BF16 weights.

Fields§

§norm1_w: DevicePtr§qkv_w: DevicePtr§qkv_b: DevicePtr§q_norm_w: DevicePtr

2026-09-25: Per-head RMSNorm weight for Q, [head_dim]. The kernel applies it after the QKV split and before the rotary transform.

§k_norm_w: DevicePtr§proj_w: DevicePtr§proj_b: DevicePtr§norm2_w: DevicePtr§gate_w: DevicePtr§gate_b: DevicePtr§up_w: DevicePtr§up_b: DevicePtr§down_w: DevicePtr§down_b: DevicePtr

Auto Trait Implementations§

impl Freeze for GlmVitBlock

impl RefUnwindSafe for GlmVitBlock

impl Send for GlmVitBlock

impl Sync for GlmVitBlock

impl Unpin for GlmVitBlock

impl UnwindSafe for GlmVitBlock

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T> Instrument for T

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided [Span], returning an Instrumented wrapper. Read more

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> Same for T

type Output = T

Should always be Self

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V

impl<T> WithSubscriber for T

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a [WithDispatch] wrapper. Read more

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a [WithDispatch] wrapper. Read more

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,