lift-ast 0.4.4

LIFT compiler framework AST: lexer, parser, and AST for the .lif source language
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

use crate::token::Span;

#[derive(Debug, Clone)]
pub struct Program {
    pub dialect_directives: Vec<DialectDirective>,
    pub modules: Vec<ModuleDecl>,
}

#[derive(Debug, Clone)]
pub struct DialectDirective {
    pub name: String,
    pub span: Span,
}

#[derive(Debug, Clone)]
pub struct ModuleDecl {
    pub name: String,
    pub functions: Vec<FuncDecl>,
    pub span: Span,
}

#[derive(Debug, Clone)]
pub struct FuncDecl {
    pub name: String,
    pub params: Vec<ParamDecl>,
    pub returns: Vec<TypeExpr>,
    pub body: Vec<Statement>,
    pub span: Span,
}

#[derive(Debug, Clone)]
pub struct ParamDecl {
    pub name: String,
    pub ty: TypeExpr,
    pub span: Span,
}

#[derive(Debug, Clone)]
pub enum Statement {
    OpAssign(OpAssign),
    Return(ReturnStmt),
}

#[derive(Debug, Clone)]
pub struct OpAssign {
    pub results: Vec<String>,
    pub op_name: String,
    pub operands: Vec<Operand>,
    pub attrs: Vec<(String, AttrValue)>,
    pub type_sig: Option<TypeSignature>,
    pub span: Span,
}

#[derive(Debug, Clone)]
pub struct ReturnStmt {
    pub values: Vec<Operand>,
    pub span: Span,
}

#[derive(Debug, Clone)]
pub enum Operand {
    Value(String),
    FuncRef(String),
    Literal(LiteralValue),
}

#[derive(Debug, Clone)]
pub enum LiteralValue {
    Integer(i64),
    Float(f64),
    Bool(bool),
    String(String),
}

#[derive(Debug, Clone)]
pub enum AttrValue {
    Integer(i64),
    Float(f64),
    Bool(bool),
    String(String),
    Array(Vec<AttrValue>),
}

#[derive(Debug, Clone)]
pub struct TypeSignature {
    pub inputs: Vec<TypeExpr>,
    pub outputs: Vec<TypeExpr>,
}

#[derive(Debug, Clone)]
pub enum TypeExpr {
    Tensor(TensorTypeExpr),
    Qubit,
    Bit,
    Hamiltonian(usize),
    Scalar(ScalarTypeExpr),
    Void,
    Index,
    Function(Box<TypeSignature>),
}

#[derive(Debug, Clone)]
pub struct TensorTypeExpr {
    pub shape: Vec<DimExpr>,
    pub dtype: String,
}

#[derive(Debug, Clone)]
pub enum DimExpr {
    Constant(usize),
    Symbolic(String),
    Dynamic,
}

#[derive(Debug, Clone)]
pub struct ScalarTypeExpr {
    pub name: String,
}