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
37
38
39

use crate::ir::DataType;
use crate::ir::Program;
use crate::ops::AlgebraicLaw;
use crate::ops::OpSpec;

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

impl ReduceAll {
    /// Declarative operation specification.
    pub const SPEC: OpSpec = OpSpec::composition(
        "reductions.reduce_all",
        INPUTS,
        OUTPUTS,
        LAWS,
        Self::program,
    );

    /// Build the canonical sequential reduction program.
    #[must_use]
    pub fn program() -> Program {
        crate::ops::reductions::programs::reduce_bool_program(true, crate::ir::Expr::and)
    }
}

pub const INPUTS: &[DataType] = &[DataType::Bool];

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

pub const OUTPUTS: &[DataType] = &[DataType::Bool];

/// Boolean all reduction.
#[derive(Debug, Clone, Copy, Default)]
pub struct ReduceAll;