vyre-primitives 0.4.1

Compositional primitives for vyre — marker types (always on) + Tier 2.5 LEGO substrate (feature-gated per domain).
Documentation 
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

//! Value-set interval boundary propagation as Vyre IR.
//!
//! The primitive computes conservative `[min, max]` interval merges over u32
//! pairs. Backend crates decide how to lower min/max; this module only owns the
//! substrate-neutral program shape.

use std::sync::Arc;

use vyre_foundation::ir::model::expr::Ident;
use vyre_foundation::ir::{BufferAccess, BufferDecl, DataType, Expr, Node, Program};

/// Canonical op id for interval merge.
pub const OP_ID: &str = "vyre-primitives::math::interval_merge";

/// Build the per-lane interval merge body.
///
/// Buffers use pair layout: `mins_out[i] = min(mins_a[i], mins_b[i])` and
/// `maxs_out[i] = max(maxs_a[i], maxs_b[i])`.
#[must_use]
pub fn interval_merge_body(
    mins_a: &str,
    maxs_a: &str,
    mins_b: &str,
    maxs_b: &str,
    mins_out: &str,
    maxs_out: &str,
    lane_count: u32,
) -> Vec<Node> {
    let lane = Expr::gid_x();
    vec![Node::if_then(
        Expr::lt(lane.clone(), Expr::u32(lane_count)),
        vec![
            Node::let_bind("interval_min_a", Expr::load(mins_a, lane.clone())),
            Node::let_bind("interval_max_a", Expr::load(maxs_a, lane.clone())),
            Node::let_bind("interval_min_b", Expr::load(mins_b, lane.clone())),
            Node::let_bind("interval_max_b", Expr::load(maxs_b, lane.clone())),
            Node::store(
                mins_out,
                lane.clone(),
                Expr::min(Expr::var("interval_min_a"), Expr::var("interval_min_b")),
            ),
            Node::store(
                maxs_out,
                lane,
                Expr::max(Expr::var("interval_max_a"), Expr::var("interval_max_b")),
            ),
        ],
    )]
}

/// Build an interval merge Program.
#[must_use]
#[allow(clippy::too_many_arguments)]
pub fn interval_merge_program(
    mins_a: &str,
    maxs_a: &str,
    mins_b: &str,
    maxs_b: &str,
    mins_out: &str,
    maxs_out: &str,
    lane_count: u32,
) -> Program {
    Program::wrapped(
        vec![
            BufferDecl::storage(mins_a, 0, BufferAccess::ReadOnly, DataType::U32)
                .with_count(lane_count.max(1)),
            BufferDecl::storage(maxs_a, 1, BufferAccess::ReadOnly, DataType::U32)
                .with_count(lane_count.max(1)),
            BufferDecl::storage(mins_b, 2, BufferAccess::ReadOnly, DataType::U32)
                .with_count(lane_count.max(1)),
            BufferDecl::storage(maxs_b, 3, BufferAccess::ReadOnly, DataType::U32)
                .with_count(lane_count.max(1)),
            BufferDecl::storage(mins_out, 4, BufferAccess::ReadWrite, DataType::U32)
                .with_count(lane_count.max(1)),
            BufferDecl::storage(maxs_out, 5, BufferAccess::ReadWrite, DataType::U32)
                .with_count(lane_count.max(1)),
        ],
        [256, 1, 1],
        vec![Node::Region {
            generator: Ident::from(OP_ID),
            source_region: None,
            body: Arc::new(interval_merge_body(
                mins_a, maxs_a, mins_b, maxs_b, mins_out, maxs_out, lane_count,
            )),
        }],
    )
}

/// CPU oracle for interval merge.
#[must_use]
pub fn cpu_interval_merge(
    mins_a: &[u32],
    maxs_a: &[u32],
    mins_b: &[u32],
    maxs_b: &[u32],
) -> (Vec<u32>, Vec<u32>) {
    let len = mins_a
        .len()
        .min(maxs_a.len())
        .min(mins_b.len())
        .min(maxs_b.len());
    let mut mins = Vec::with_capacity(len);
    let mut maxs = Vec::with_capacity(len);
    for i in 0..len {
        mins.push(mins_a[i].min(mins_b[i]));
        maxs.push(maxs_a[i].max(maxs_b[i]));
    }
    (mins, maxs)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn interval_merge_program_is_ir_not_target_text() {
        let program = interval_merge_program("amin", "amax", "bmin", "bmax", "omin", "omax", 16);
        let dump = format!("{program:#?}");
        assert!(dump.contains("Min"));
        assert!(dump.contains("Max"));
        assert!(!dump.contains("subgroupMin"));
        assert!(!dump.contains("vec2<u32>"));
    }

    #[test]
    fn cpu_interval_merge_is_conservative() {
        let (mins, maxs) = cpu_interval_merge(&[10, 0, 7], &[20, 3, 9], &[4, 2, 8], &[18, 5, 12]);
        assert_eq!(mins, vec![4, 0, 7]);
        assert_eq!(maxs, vec![20, 5, 12]);
    }
}