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::{Expr, Program};
use crate::ops::AlgebraicLaw;
use crate::ops::{OpSpec, U32_INPUTS, U32_OUTPUTS};

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

/// Argmax over u32 values, returning `u32::MAX` for an empty input.
#[derive(Debug, Clone, Copy, Default)]
pub struct ArgmaxU32;

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

    /// Build the canonical sequential argmax program.
    #[must_use]
    pub fn program() -> Program {
        crate::ops::reductions::programs::arg_u32_program(Expr::gt)
    }
}

pub const LAWS: &[AlgebraicLaw] = &[AlgebraicLaw::Bounded {
    lo: 0,
    hi: u32::MAX,
}];