From: Nicolas Vasilache <ntv@google.com>
Date: Tue, 2 Jun 2020 22:04:10 +0000 (-0400)
Subject: [mlir] NFC - Add AffineMinMaxOpBase getDim/SymbolOperands
X-Git-Tag: llvmorg-12-init~4360
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=7de45878059a4e38dd6ceba43a058cee5d283b97;p=platform%2Fupstream%2Fllvm.git

[mlir] NFC - Add AffineMinMaxOpBase getDim/SymbolOperands
---

diff --git a/mlir/include/mlir/Dialect/Affine/IR/AffineOps.td b/mlir/include/mlir/Dialect/Affine/IR/AffineOps.td
index 3366ef4..9329f3d 100644
--- a/mlir/include/mlir/Dialect/Affine/IR/AffineOps.td
+++ b/mlir/include/mlir/Dialect/Affine/IR/AffineOps.td
@@ -460,6 +460,14 @@ class AffineMinMaxOpBase<string mnemonic, list<OpTrait> traits = []> :
     static StringRef getMapAttrName() { return "map"; }
     AffineMap getAffineMap() { return map(); }
     ValueRange getMapOperands() { return operands(); }
+    ValueRange getDimOperands() {
+      return OperandRange{operands().begin(),
+                        operands().begin() + map().getNumDims()};
+    }
+    ValueRange getSymbolOperands() {
+      return OperandRange{operands().begin() + map().getNumDims(),
+                        operands().end()};
+    }
   }];
   let verifier = [{ return ::verifyAffineMinMaxOp(*this); }];
   let printer = [{ return ::printAffineMinMaxOp(p, *this); }];