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

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

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

impl ReduceSumU32 {
    /// Declarative operation specification.
    pub const SPEC: OpSpec = OpSpec::composition(
        "reductions.reduce_sum_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, crate::ir::Expr::add)
    }
}

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

/// Wrapping u32 sum reduction.
#[derive(Debug, Clone, Copy, Default)]
pub struct ReduceSumU32;