Enum BrickId 

#[repr(u8)]
pub enum BrickId {
    RmsNorm = 0,
    LayerNorm = 1,
    QkvProjection = 2,
    RopeEmbedding = 3,
    AttentionScore = 4,
    AttentionSoftmax = 5,
    AttentionOutput = 6,
    OutputProjection = 7,
    GateProjection = 8,
    UpProjection = 9,
    Activation = 10,
    DownProjection = 11,
    Embedding = 12,
    LmHead = 13,
    Sampling = 14,
    SpMV = 15,
    SpMM = 16,
    FormatConvert = 17,
    FFT1D = 18,
    FFT2D = 19,
    LUFactorize = 20,
    QRFactorize = 21,
    SVDCompute = 22,
}

Well-known brick types for O(1) lookup on hot path.

PAR-200: Eliminates string allocation and HashMap hashing during profiling. Use BrickId::Custom with string fallback for unknown brick types.

Example

use trueno::brick::BrickId;

let brick = BrickId::RmsNorm;
assert_eq!(brick.category(), trueno::brick::BrickCategory::Norm);
assert_eq!(brick.name(), "RmsNorm");

Variants

RmsNorm = 0

RMS normalization layer

LayerNorm = 1

Layer normalization

QkvProjection = 2

Q/K/V projection (combined or separate)

RopeEmbedding = 3

Rotary position embedding

AttentionScore = 4

Attention score computation (Q @ K^T)

AttentionSoftmax = 5

Attention softmax

AttentionOutput = 6

Attention output (scores @ V)

OutputProjection = 7

Output projection after attention

GateProjection = 8

Gate projection (for gated FFN)

UpProjection = 9

Up projection

Activation = 10

SiLU/GELU/ReLU activation

DownProjection = 11

Down projection

Embedding = 12

Token embedding lookup

LmHead = 13

Language model head (logits)

Sampling = 14

Token sampling

SpMV = 15

Sparse matrix-vector multiply (cuSPARSE parity)

SpMM = 16

Sparse matrix-dense matrix multiply

FormatConvert = 17

Sparse format conversion

FFT1D = 18

1D Fast Fourier Transform

FFT2D = 19

2D Fast Fourier Transform

LUFactorize = 20

LU factorization

QRFactorize = 21

QR factorization

SVDCompute = 22

Singular Value Decomposition

Implementations

impl BrickId

pub const COUNT: usize = 23

Number of well-known brick types.

pub const ALL: [BrickId; 23]

All BrickId variants in order, for safe index-based iteration.

Eliminates need for transmute::<u8, BrickId> in array initialization.

pub fn validate_index(index: usize) -> bool

Validate that a raw u8 is within the BrickId range.

pub fn category(self) -> BrickCategory

Get the category for hierarchical aggregation.

pub const fn name(self) -> &'static str

Get the string name of this brick.

pub fn from_str(s: &str) -> Option<Self>

Try to parse a string into a BrickId.

Trait Implementations

impl Clone for BrickId

fn clone(&self) -> BrickId

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BrickId

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

Formats the value using the given formatter.

impl Display for BrickId

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

Formats the value using the given formatter.

impl Hash for BrickId

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for BrickId

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for BrickId

impl Eq for BrickId

impl StructuralPartialEq for BrickId