commit 1989cc15c31284a70ac1165e97bb3ad04a3b99bb
author MLIR Team <no-reply@google.com> Wed Aug 15 15:39:26 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 13:00:50 2019 -0700
tree e0909fe4189f5aa17871555d9754cd1cb12f8d93
parent 04914bc2048dbb66b6dcf1b2c3a547d995d4784b

Adds dealloc MLIR memory operation to StandardOps.

PiperOrigin-RevId: 208896071

lib/IR/StandardOps.cpp

diff --git a/lib/IR/StandardOps.cpp b/lib/IR/StandardOps.cpp
index e78290e..a1603ba 100644
--- a/lib/IR/StandardOps.cpp
+++ b/lib/IR/StandardOps.cpp
@@ -268,6 +268,28 @@
 }
 
 //===----------------------------------------------------------------------===//
+// DeallocOp
+//===----------------------------------------------------------------------===//
+
+void DeallocOp::print(OpAsmPrinter *p) const {
+  *p << "dealloc " << *getMemRef() << " : " << *getMemRef()->getType();
+}
+
+bool DeallocOp::parse(OpAsmParser *parser, OperationState *result) {
+  OpAsmParser::OperandType memrefInfo;
+  MemRefType *type;
+
+  return parser->parseOperand(memrefInfo) || parser->parseColonType(type) ||
+         parser->resolveOperand(memrefInfo, type, result->operands);
+}
+
+const char *DeallocOp::verify() const {
+  if (!isa<MemRefType>(getMemRef()->getType()))
+    return "operand must be a memref";
+  return nullptr;
+}
+
+//===----------------------------------------------------------------------===//
 // DimOp
 //===----------------------------------------------------------------------===//
 
@@ -466,7 +488,7 @@
 
 /// Install the standard operations in the specified operation set.
 void mlir::registerStandardOperations(OperationSet &opSet) {
-  opSet.addOperations<AddFOp, AffineApplyOp, AllocOp, ConstantOp, DimOp, LoadOp,
-                      ReturnOp, StoreOp>(
+  opSet.addOperations<AddFOp, AffineApplyOp, AllocOp, ConstantOp, DeallocOp,
+                      DimOp, LoadOp, ReturnOp, StoreOp>(
       /*prefix=*/"");
 }