Struct MoeTopkCapture 

pub struct MoeTopkCapture {
    pub num_experts: usize,
    /* private fields */
}

Shared capture buffer — one entry per MoE router TopK in schedule order.

Fields

num_experts: usize

Implementations

impl MoeTopkCapture

pub fn new(num_experts: usize) -> Arc<MoeTopkCapture>

pub fn clear(&self)

pub fn push_topk_f32(&self, data: &[f32], axis_dim: usize)

Record one router TopK output (outer * k f32-encoded expert ids).

pub fn take_layers(&self) -> Vec<Vec<u32>>

Trait Implementations

impl Debug for MoeTopkCapture

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

Formats the value using the given formatter.