Struct RecognizedProgram 

pub struct RecognizedProgram {
    pub stages: Vec<RecognizedStage>,
    pub unrecognized: Vec<(usize, String)>,
    pub unlifted_writes: Vec<(usize, String, &'static str)>,
    pub declared: Vec<DeclaredTensor>,
    pub symbols: HashMap<Symbol, String>,
    pub shapes: HashMap<String, (usize, usize)>,
    pub mentions: HashMap<String, BTreeSet<usize>>,
    pub opaque: Vec<usize>,
    pub quiet: BTreeSet<usize>,
}

The walk’s result: the stages in program order, the tensor statements it could not lift (by top-level index and binder), the declared tensors, and the names of every symbol the specs mention.

Fields

stages: Vec<RecognizedStage>

unrecognized: Vec<(usize, String)>

unlifted_writes: Vec<(usize, String, &'static str)>

Top-level writes to a declared tensor that no section computes — (statement, tensor, why). Each is a reason the program is NOT lowered: dropping it would leave the tensor holding values the program never had.

declared: Vec<DeclaredTensor>

symbols: HashMap<Symbol, String>

shapes: HashMap<String, (usize, usize)>

Every tensor the program binds, by name, as (rows, cols) — the parser’s table; the desugared statements carry only flat lengths.

mentions: HashMap<String, BTreeSet<usize>>

Every name the program’s top-level statements touch, by name, with the statements that touch it — a Show, a scalar read and a stage alike. What reads a value is here, so a pass can know a value nothing else reads.

opaque: Vec<usize>

The top-level statements that run foreign code, which may touch any name.

quiet: BTreeSet<usize>

The top-level statements that only declare a tensor or a scalar, or give a declared tensor a literal value before any op reads it: the statements that make the program’s inputs, which observe nothing.

Implementations

impl RecognizedProgram

pub fn shape_of(&self, name: &str, n: usize) -> (usize, usize)

The (rows, cols) the program gives tensor name, when that accounts for the n elements a recognizer counted — and the 1 × n row otherwise.

pub fn name(&self, s: Symbol) -> &str

The name of a symbol one of this program’s specs mentions.

Trait Implementations

impl Clone for RecognizedProgram

fn clone(&self) -> RecognizedProgram

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RecognizedProgram

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

Formats the value using the given formatter.

impl Default for RecognizedProgram

fn default() -> RecognizedProgram

Returns the “default value” for a type.