Struct ComputeFacts 

pub struct ComputeFacts {
    pub stages: usize,
    pub reductions: usize,
    pub per_column_operands: usize,
    pub per_row_operands: usize,
    pub sfpu: BTreeSet<SfpuKind>,
}

What a program’s compute stages do, in the vocabulary of the algebra rather than of a backend.

Fields

stages: usize

Compute stages in the program.

reductions: usize

Stages that fold an axis away — the row reductions.

per_column_operands: usize

Stages applying a value that is constant DOWN THE ROWS: a per-COLUMN vector, which is what a normalization’s weight and bias are. The access map omits the row axis.

per_row_operands: usize

Stages applying a value that is constant ACROSS A ROW: a per-ROW scalar, which is what a reduction’s result is when it is broadcast back. The access map omits the column axis.

sfpu: BTreeSet<SfpuKind>

Every SFPU transform the program applies, anywhere in a body (including nested inside arithmetic, which is where fusion puts them).

Implementations

impl ComputeFacts

pub fn applies_a_per_column_operand(&self) -> bool

Whether the program applies a per-column vector — a weight or a bias.

pub fn rescales_by_a_per_row_scalar(&self) -> bool

Whether the program broadcasts a per-row scalar back across its row — a rescale.

pub fn uses(&self, kind: SfpuKind) -> bool

Whether the program applies a given SFPU transform anywhere.

Trait Implementations

impl Clone for ComputeFacts

fn clone(&self) -> ComputeFacts

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ComputeFacts

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

Formats the value using the given formatter.

impl Default for ComputeFacts

fn default() -> ComputeFacts

Returns the “default value” for a type.

impl PartialEq for ComputeFacts

fn eq(&self, other: &ComputeFacts) -> 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 Eq for ComputeFacts

impl StructuralPartialEq for ComputeFacts