bb-compiler 0.3.4

Compiler pipeline for the bytesandbrains framework — Compiler driver, CompilerPass trait, canonical pass list, BuildError.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

//! Shared helpers for compiler passes that emit FunctionProtos.
//!
//! [`hash_node_bodies`] produces a stable u64 over a node sequence
//! so identical bodies converge to the same FunctionProto name and
//! dedup at link time.

use std::collections::hash_map::DefaultHasher;
use std::hash::{Hash, Hasher};

use bb_ir::proto::onnx::NodeProto;

/// Stable short hash of a node sequence. Used to disambiguate two
/// fused regions with the same binding but different bodies, and to
/// converge identical bodies onto the same FunctionProto name (the
/// linker dedupes at register time).
pub(crate) fn hash_node_bodies(nodes: &[NodeProto]) -> u64 {
    let mut h = DefaultHasher::new();
    for node in nodes {
        node.op_type.hash(&mut h);
        node.domain.hash(&mut h);
        node.overload.hash(&mut h);
        node.input.len().hash(&mut h);
        node.output.len().hash(&mut h);
        for inp in &node.input {
            inp.hash(&mut h);
        }
        for out in &node.output {
            out.hash(&mut h);
        }
        for attr in &node.attribute {
            attr.name.hash(&mut h);
        }
    }
    h.finish()
}