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
40
41
42
43
44
45
46
47
48
49
50
51
52
53

//! Generated operation registry.
//!
//! `build.rs` scans the operation source tree and writes the registry into
//! `$OUT_DIR`; this file only exposes that generated list to lookup helpers.

use crate::ops::OpSpec;

include!(concat!(env!("OUT_DIR"), "/ops_registry.rs"));

#[cfg(test)]
/// Test-only operation used to verify recursive call-depth validation.
pub static TEST_OP_1: OpSpec = OpSpec::composition_inlinable("test.op1", &[], &[], &[], || {
    crate::ir::Program::new(
        vec![],
        [1, 1, 1],
        vec![crate::ir::Node::let_bind(
            "x",
            crate::ir::Expr::call("test.op2", vec![]),
        )],
    )
});

#[cfg(test)]
/// Test-only operation used to verify recursive call-depth validation.
pub static TEST_OP_2: OpSpec = OpSpec::composition_inlinable("test.op2", &[], &[], &[], || {
    crate::ir::Program::new(
        vec![],
        [1, 1, 1],
        vec![crate::ir::Node::let_bind(
            "x",
            crate::ir::Expr::call("test.op1", vec![]),
        )],
    )
});

#[cfg(test)]
static TEST_REGISTRY: &[&OpSpec] = &[&TEST_OP_1, &TEST_OP_2];

/// Iterate over every registered `OpSpec`.
pub fn registry() -> impl Iterator<Item = &'static OpSpec> {
    #[cfg(test)]
    {
        TEST_REGISTRY
            .iter()
            .copied()
            .chain(GENERATED_REGISTRY.iter().copied())
    }

    #[cfg(not(test))]
    {
        GENERATED_REGISTRY.iter().copied()
    }
}