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
54
55
56
57
58
59
60
61
62
63
64
65
66

//! Node encoder for the stable IR wire format.

use super::{put_expr, put_nodes};
use crate::ir::serial::wire::framing::{put_string, put_u8};
use crate::ir::serial::wire::Node;

/// Append the tag and payload for one control-flow or storage node.
///
/// # Errors
///
/// Returns an actionable error when a nested expression, block, or identifier
/// cannot be represented by the current wire schema.
#[inline]
pub fn put_node(out: &mut Vec<u8>, node: &Node) -> Result<(), String> {
    match node {
        Node::Let { name, value } => {
            put_u8(out, 0);
            put_string(out, name)?;
            put_expr(out, value)?;
        }
        Node::Assign { name, value } => {
            put_u8(out, 1);
            put_string(out, name)?;
            put_expr(out, value)?;
        }
        Node::Store {
            buffer,
            index,
            value,
        } => {
            put_u8(out, 2);
            put_string(out, buffer)?;
            put_expr(out, index)?;
            put_expr(out, value)?;
        }
        Node::If {
            cond,
            then,
            otherwise,
        } => {
            put_u8(out, 3);
            put_expr(out, cond)?;
            put_nodes(out, then)?;
            put_nodes(out, otherwise)?;
        }
        Node::Loop {
            var,
            from,
            to,
            body,
        } => {
            put_u8(out, 4);
            put_string(out, var)?;
            put_expr(out, from)?;
            put_expr(out, to)?;
            put_nodes(out, body)?;
        }
        Node::Return => put_u8(out, 5),
        Node::Block(nodes) => {
            put_u8(out, 6);
            put_nodes(out, nodes)?;
        }
        Node::Barrier => put_u8(out, 7),
    }
    Ok(())
}