Fix

mlir-rs · Oct 14, 2023 · 32150cb · 32150cb
1 parent a6a6c77
commit 32150cb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/macro/src/dialect/operation/builder.rs b/macro/src/dialect/operation/builder.rs
@@ -142,7 +142,7 @@ impl<'o> OperationBuilder<'o> {
             #[doc = #doc]
             pub struct #builder_ident <'c, #(#iter_arguments),* > {
                 builder: ::melior::ir::operation::OperationBuilder<'c>,
-                context: ::melior::ContextRef<'c>,
+                context: &'c ::melior::Context,
                 #(#phantom_fields),*
             }
 
@@ -180,9 +180,9 @@ impl<'o> OperationBuilder<'o> {
 
         quote! {
             impl<'c> #builder_ident<'c, #(#arguments),*> {
-                pub fn new(location: ::melior::ir::Location<'c>) -> Self {
+                pub fn new(context: &'c ::melior::Context, location: ::melior::ir::Location<'c>) -> Self {
                     Self {
-                        context: location.context(),
+                        context,
                         builder: ::melior::ir::operation::OperationBuilder::new(&context, #name, location),
                         #(#phantoms),*
                     }

diff --git a/melior/src/dialect/ods.rs b/melior/src/dialect/ods.rs
@@ -215,7 +215,7 @@ mod tests {
             let i64_type = IntegerType::new(&context, 64);
 
             block.append_operation(
-                llvm::AllocaOpBuilder::new(location)
+                llvm::AllocaOpBuilder::new(&context, location)
                     .alignment(IntegerAttribute::new(8, i64_type.into()))
                     .elem_type(TypeAttribute::new(i64_type.into()))
                     .array_size(alloca_size)