vyre-primitives 0.7.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

//! Tier 2.5 reduction primitives  -  `count`/`min`/`max`/`sum` over
//! bitsets and fixed-width u32 ValueSets.
//!
//! Scalar reductions use one grid-stride workgroup and global atomics
//! so the baseline primitive is parallel instead of serial lane-0
//! scaffolding. Higher-level workgroup-tree reductions still compose
//! these where a caller needs per-workgroup partials or f32 support.

/// `reduce_all` - emit `1` when every lane in a u32 ValueSet is non-zero.
pub mod all;
/// `reduce_any` - emit `1` when any lane in a u32 ValueSet is non-zero.
pub mod any;
mod atomic_scalar;
pub mod count;
pub mod count_non_zero;
pub mod gather;
pub mod histogram;
mod indexed_move;
/// Unsigned maximum over a u32 ValueSet.
pub mod max;
/// Unsigned minimum over a u32 ValueSet.
pub mod min;
pub mod multi_block_prefix_scan;
pub mod radix_sort;
pub mod range_counts;
pub mod scatter;
pub mod segment_reduce;
pub mod sum;
pub mod workgroup_any;
pub mod workgroup_tree;

#[cfg(any(test, feature = "cpu-parity"))]
fn collect_cpu_reference<T>(
    operation: &str,
    fill: impl FnOnce(&mut Vec<T>) -> Result<(), String>,
) -> Vec<T> {
    let mut out = Vec::new();
    fill(&mut out).unwrap_or_else(|error| {
        panic!("vyre-primitives {operation} CPU reference failed: {error}")
    });
    out
}