Add BlockApi trait.

mlir-rs · Dec 9, 2024 · ad7998f · ad7998f
1 parent 73a214d
commit ad7998f
Show file tree

Hide file tree

Showing 19 changed files with 242 additions and 93 deletions.
diff --git a/macro/tests/operand.rs b/macro/tests/operand.rs
@@ -1,6 +1,6 @@
 mod utility;
 
-use melior::ir::{Block, Location, Type};
+use melior::ir::{block::BlockApi, Block, Location, Type};
 use utility::*;
 
 melior_macro::dialect! {

diff --git a/melior/src/dialect/arith.rs b/melior/src/dialect/arith.rs
@@ -174,6 +174,7 @@ mod tests {
         dialect::func,
         ir::{
             attribute::{StringAttribute, TypeAttribute},
+            block::BlockApi,
             r#type::FunctionType,
             Attribute, Block, Location, Module, Region, Type,
         },

diff --git a/melior/src/dialect/cf.rs b/melior/src/dialect/cf.rs
@@ -5,6 +5,7 @@ use crate::{
         attribute::{
             DenseElementsAttribute, DenseI32ArrayAttribute, IntegerAttribute, StringAttribute,
         },
+        block::BlockApi,
         operation::OperationBuilder,
         r#type::RankedTensorType,
         Block, Identifier, Location, Operation, Type, Value,

diff --git a/melior/src/dialect/func.rs b/melior/src/dialect/func.rs
@@ -87,7 +87,7 @@ pub fn r#return<'c>(operands: &[Value<'c, '_>], location: Location<'c>) -> Opera
 mod tests {
     use super::*;
     use crate::{
-        ir::{Block, Module, Type},
+        ir::{block::BlockApi, Block, Module, Type},
         test::create_test_context,
     };
 

diff --git a/melior/src/dialect/index.rs b/melior/src/dialect/index.rs
@@ -75,6 +75,7 @@ mod tests {
         dialect::func,
         ir::{
             attribute::{StringAttribute, TypeAttribute},
+            block::BlockApi,
             r#type::{FunctionType, IntegerType},
             Block, Location, Module, Region, Type,
         },

diff --git a/melior/src/dialect/llvm.rs b/melior/src/dialect/llvm.rs
@@ -381,6 +381,7 @@ mod tests {
         },
         ir::{
             attribute::{IntegerAttribute, StringAttribute, TypeAttribute},
+            block::BlockApi,
             r#type::{FunctionType, IntegerType},
             Block, Module, Region,
         },

diff --git a/melior/src/dialect/memref.rs b/melior/src/dialect/memref.rs
@@ -324,6 +324,7 @@ mod tests {
         dialect::{func, index},
         ir::{
             attribute::{DenseElementsAttribute, StringAttribute, TypeAttribute},
+            block::BlockApi,
             r#type::{FunctionType, IntegerType, RankedTensorType},
             Block, Module, Region, Type,
         },

diff --git a/melior/src/dialect/ods.rs b/melior/src/dialect/ods.rs
@@ -140,6 +140,7 @@ mod tests {
         dialect,
         ir::{
             attribute::{IntegerAttribute, StringAttribute, TypeAttribute},
+            block::BlockApi,
             r#type::{FunctionType, IntegerType},
             Block, Location, Module, Region, Type,
         },

diff --git a/melior/src/dialect/scf.rs b/melior/src/dialect/scf.rs
@@ -114,6 +114,7 @@ mod tests {
         dialect::{arith, func},
         ir::{
             attribute::{FloatAttribute, IntegerAttribute, StringAttribute, TypeAttribute},
+            block::BlockApi,
             r#type::{FunctionType, IntegerType, Type},
             Attribute, Block, Module,
         },

diff --git a/melior/src/helpers/builtin.rs b/melior/src/helpers/builtin.rs
@@ -1,5 +1,5 @@
 use crate::{
-    ir::{Block, Operation, Value},
+    ir::{block::BlockApi, Block, Operation, Value},
     Error,
 };
 

diff --git a/melior/src/helpers/llvm.rs b/melior/src/helpers/llvm.rs
@@ -6,6 +6,7 @@ use crate::{
         attribute::{
             DenseI32ArrayAttribute, DenseI64ArrayAttribute, IntegerAttribute, TypeAttribute,
         },
+        block::BlockApi,
         r#type::IntegerType,
         Attribute, Block, Location, Type, Value, ValueLike,
     },