vyre 0.4.0

GPU compute intermediate representation with a standard operation library
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

use crate::ir::{Expr, Program};
use crate::ops::AlgebraicLaw;
use crate::ops::{OpSpec, U32_INPUTS, U32_OUTPUTS};

// WGSL lowering marker for `reductions.reduce_max_u32`.
//
// No special handling is needed. The generic Category A lowerer emits the
// sequential reduction composition.

impl ReduceMaxU32 {
    /// Declarative operation specification.
    pub const SPEC: OpSpec = OpSpec::composition(
        "reductions.reduce_max_u32",
        U32_INPUTS,
        U32_OUTPUTS,
        LAWS,
        Self::program,
    );

    /// Build the canonical sequential reduction program.
    #[must_use]
    pub fn program() -> Program {
        crate::ops::reductions::programs::reduce_u32_program(0, |acc, value| {
            Expr::select(Expr::ge(acc.clone(), value.clone()), acc, value)
        })
    }
}

pub const LAWS: &[AlgebraicLaw] = &[
    AlgebraicLaw::Associative,
    AlgebraicLaw::Identity { element: 0 },
];

/// u32 maximum reduction.
#[derive(Debug, Clone, Copy, Default)]
pub struct ReduceMaxU32;