melior 0.27.1

The rustic MLIR bindings in Rust
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

//! Passes and pass managers.

pub mod affine;
pub mod amdgpu;
pub mod arith;
pub mod arm_sme;
pub mod r#async;
pub mod bufferization;
pub mod conversion;
pub mod emitc;
pub mod external;
pub mod func;
pub mod gpu;
pub mod linalg;
pub mod llvm;
mod manager;
pub mod math;
pub mod memref;
pub mod ml_program;
pub mod nvgpu;
pub mod open_acc;
mod operation_manager;
pub mod scf;
pub mod shape;
pub mod shard;
pub mod sparse_tensor;
pub mod spirv;
pub mod tensor;
pub mod tosa;
pub mod transform;
pub mod transform_dialect;
pub mod vector;

pub use self::{
    external::{ExternalPass, RunExternalPass, create_external},
    manager::{PassIrPrintingOptions, PassManager},
    operation_manager::OperationPassManager,
};
use mlir_sys::MlirPass;

/// A pass.
pub struct Pass {
    raw: MlirPass,
}

impl Pass {
    /// Creates a pass from a raw function.
    ///
    /// # Safety
    ///
    /// A raw function must be valid.
    pub unsafe fn from_raw_fn(create_raw: unsafe extern "C" fn() -> MlirPass) -> Self {
        Self {
            raw: unsafe { create_raw() },
        }
    }

    /// Creates a pass from a raw object.
    ///
    /// # Safety
    ///
    /// A raw object must be valid.
    pub const unsafe fn from_raw(raw: MlirPass) -> Self {
        Self { raw }
    }

    /// Converts a pass into a raw object.
    pub const fn to_raw(&self) -> MlirPass {
        self.raw
    }

    #[doc(hidden)]
    pub unsafe fn __private_from_raw_fn(create_raw: unsafe extern "C" fn() -> MlirPass) -> Self {
        unsafe { Self::from_raw_fn(create_raw) }
    }
}