commit 04914bc2048dbb66b6dcf1b2c3a547d995d4784b
author MLIR Team <no-reply@google.com> Wed Aug 15 15:14:45 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 13:00:22 2019 -0700
tree 18e4ee466ebf642a17650a1b593d22f6cdc13716
parent 0497c4ba37d0d3128950aea79cea45e1e053eba3

AffineMap::isIdentity clean up from previous CL review.

PiperOrigin-RevId: 208891864

lib/IR/AffineMap.cpp

diff --git a/lib/IR/AffineMap.cpp b/lib/IR/AffineMap.cpp
index 8dc0d1f..3f640ba 100644
--- a/lib/IR/AffineMap.cpp
+++ b/lib/IR/AffineMap.cpp
@@ -31,10 +31,9 @@
   if (getNumDims() != getNumResults())
     return false;
   ArrayRef<AffineExpr *> results = getResults();
-  for (unsigned i = 0; i < getNumDims(); ++i) {
-    auto *expr = results[i];
-    if (!isa<AffineDimExpr>(expr) ||
-        cast<AffineDimExpr>(expr)->getPosition() != i)
+  for (unsigned i = 0, numDims = getNumDims(); i < numDims; ++i) {
+    auto *expr = dyn_cast<AffineDimExpr>(results[i]);
+    if (!expr || expr->getPosition() != i)
       return false;
   }
   return true;