[ReuseIR] adjust clone

SchrodingerZhu · Sep 18, 2024 · 46fd76e · 46fd76e
1 parent aabbd69
commit 46fd76e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/reuse-mlir/include/ReuseIR/IR/ReuseIROps.td b/reuse-mlir/include/ReuseIR/IR/ReuseIROps.td
@@ -319,14 +319,15 @@ def CloneOp : ReuseIR_Op<"clone"> {
     it will increase the reference count of sub fields if necessary.
   }];
 
-  // TODO: Add type restrictions
   let arguments = (ins 
-    Arg<ReuseIR_RefType, "clonable object", [MemRead, MemAlloc<DefaultResource>]>:$object,
+    Arg<ReuseIR_RefType, "clonable object">:$object,
     OptionalAttr<IndexAttr>:$tag
   );
 
+  let results = (outs Res<AnyType, "Cloned Object">:$result);
+
   let assemblyFormat = [{
-    `(` $object `:` type($object) `)` (`tag` `(` $tag^ `)`)? attr-dict
+    `(` $object `:` type($object) `)` (`tag` `(` $tag^ `)`)? attr-dict `:` type($result)
   }];
 }