AffineMap::isIdentity clean up from previous CL review.

PiperOrigin-RevId: 208891864
This commit is contained in:
MLIR Team 2018-08-15 15:14:45 -07:00 committed by jpienaar
parent d6c4c748d7
commit 2487f2dc73
1 changed files with 3 additions and 4 deletions

View File

@ -31,10 +31,9 @@ bool AffineMap::isIdentity() const {
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;