xrcf 0.8.0

eXtensible and Reusable Compiler Framework
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

use crate::frontend::Parse;
use crate::frontend::Parser;
use crate::frontend::ParserDispatch;
use crate::frontend::TokenKind;
use crate::ir::Block;
use crate::ir::Op;
use crate::ir::Operation;
use crate::ir::OperationName;
use crate::ir::Region;
use crate::shared::Shared;
use crate::shared::SharedExt;
use anyhow::Result;
use std::fmt::Display;
use std::fmt::Formatter;

// See `include/mlir/IR/BuiltinOps.h` and goto definition of
// `mlir/IR/BuiltinOps.h.inc`.
pub struct ModuleOp {
    operation: Shared<Operation>,
}

impl Op for ModuleOp {
    fn operation_name() -> OperationName {
        OperationName::new("module".to_string())
    }
    fn new(operation: Shared<Operation>) -> Self {
        Self { operation }
    }
    fn as_any(&self) -> &dyn std::any::Any {
        self
    }
    fn operation(&self) -> &Shared<Operation> {
        &self.operation
    }
    fn display(&self, f: &mut Formatter<'_>, indent: i32) -> std::fmt::Result {
        write!(f, "{}", crate::ir::spaces(indent))?;
        self.operation().rd().display(f, indent)
    }
}

impl Display for ModuleOp {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.operation().rd())
    }
}

impl ModuleOp {
    pub fn get_body_region(&self) -> Result<Option<Shared<Region>>> {
        Ok(self.operation().rd().region())
    }
    pub fn first_op(&self) -> Result<Shared<dyn Op>> {
        let region = match self.get_body_region()? {
            Some(region) => region,
            None => return Err(anyhow::anyhow!("Expected 1 region in module, got 0")),
        };
        let block = match region.rd().blocks().into_iter().next() {
            Some(block) => block,
            None => return Err(anyhow::anyhow!("Expected 1 block in module, got 0")),
        };
        match block.rd().ops.first() {
            #[allow(clippy::needless_return)]
            None => return Err(anyhow::anyhow!("Expected 1 op, got 0")),
            #[allow(clippy::needless_return)]
            Some(op) => return Ok(op.clone()),
        };
    }
}

impl Parse for ModuleOp {
    fn op<T: ParserDispatch>(
        parser: &mut Parser<T>,
        parent: Option<Shared<Block>>,
    ) -> Result<Shared<dyn Op>> {
        let mut operation = Operation::default();
        let operation_name = parser.expect(TokenKind::BareIdentifier)?;
        assert!(operation_name.lexeme == "module");
        operation.set_name(ModuleOp::operation_name());
        let operation = Shared::new(operation.into());
        let op = ModuleOp {
            operation: operation.clone(),
        };
        let op = Shared::new(op.into());

        let region = parser.parse_region(op.clone());
        let mut operation = operation.wr();
        operation.set_region(Some(region?));
        operation.set_parent(parent.clone());

        Ok(op)
    }
}