Skip to main content

forward_graph

Function forward_graph 

Source
pub fn forward_graph(
    stmts: &[Stmt<'_>],
    interner: &Interner,
    q8: &HashMap<String, Q8Ref>,
    moe: &HashMap<String, MoeWeights>,
    kind: Q8MatKind,
) -> Result<TensorGraph, ForwardError>
Expand description

The forward of a program as the shared table fuses it under the forward’s shim (ForwardShim): every statement the front end lifts (crate::spatial::passes::recognize_stages::walk), or the program refused whole, each refusal naming its statement. The front end decides what a statement IS (an op, a declaration, a fill, a write no op computes); kernel_refusal decides whether this lane’s kernels compute it. q8 and moe name the weights it reads in Q8_0.