cairo-native 0.9.0-rc.5

A compiler to convert Cairo's IR Sierra code to MLIR and execute it.
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

//! # Memory allocation external bindings
//!
//! This metadata ensures that the bindings to the C function `realloc` exist in the current
//! compilation context.

use melior::{
    dialect::llvm,
    ir::{
        attribute::{FlatSymbolRefAttribute, StringAttribute, TypeAttribute},
        operation::OperationBuilder,
        r#type::IntegerType,
        BlockLike, Identifier, Location, Module, Operation, Region, Value,
    },
    Context, Error,
};

/// Memory allocation `realloc` metadata.
#[derive(Debug)]
pub struct ReallocBindingsMeta;

impl ReallocBindingsMeta {
    /// Register the bindings to the `realloc` C function and return the metadata.
    pub fn new(context: &Context, module: &Module) -> Self {
        module.body().append_operation(llvm::func(
            context,
            StringAttribute::new(context, "realloc"),
            TypeAttribute::new(llvm::r#type::function(
                llvm::r#type::pointer(context, 0),
                &[
                    llvm::r#type::pointer(context, 0),
                    IntegerType::new(context, 64).into(),
                ],
                false,
            )),
            Region::new(),
            &[(
                Identifier::new(context, "sym_visibility"),
                StringAttribute::new(context, "private").into(),
            )],
            Location::unknown(context),
        ));
        module.body().append_operation(llvm::func(
            context,
            StringAttribute::new(context, "free"),
            TypeAttribute::new(llvm::r#type::function(
                llvm::r#type::void(context),
                &[llvm::r#type::pointer(context, 0)],
                false,
            )),
            Region::new(),
            &[(
                Identifier::new(context, "sym_visibility"),
                StringAttribute::new(context, "private").into(),
            )],
            Location::unknown(context),
        ));

        Self
    }

    /// Calls the `realloc` function, returns a op with 1 result: an opaque pointer.
    pub fn realloc<'c, 'a>(
        context: &'c Context,
        ptr: Value<'c, 'a>,
        len: Value<'c, 'a>,
        location: Location<'c>,
    ) -> Result<Operation<'c>, Error> {
        OperationBuilder::new("llvm.call", location)
            .add_attributes(&[(
                Identifier::new(context, "callee"),
                FlatSymbolRefAttribute::new(context, "realloc").into(),
            )])
            .add_operands(&[ptr, len])
            .add_results(&[llvm::r#type::pointer(context, 0)])
            .build()
    }

    /// Calls the `free` function.
    pub fn free<'c>(
        context: &'c Context,
        ptr: Value<'c, '_>,
        location: Location<'c>,
    ) -> Result<Operation<'c>, Error> {
        OperationBuilder::new("llvm.call", location)
            .add_attributes(&[(
                Identifier::new(context, "callee"),
                FlatSymbolRefAttribute::new(context, "free").into(),
            )])
            .add_operands(&[ptr])
            .build()
    }
}