Implement getSCEVAtScope for SCEV cast expressions.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@70422 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Analysis/ScalarEvolution.cpp b/lib/Analysis/ScalarEvolution.cpp
index 257186f..8ecdd11 100644
--- a/lib/Analysis/ScalarEvolution.cpp
+++ b/lib/Analysis/ScalarEvolution.cpp
@@ -2707,8 +2707,31 @@
return UnknownValue;
}
- //assert(0 && "Unknown SCEV type!");
- return UnknownValue;
+ if (SCEVZeroExtendExpr *Cast = dyn_cast<SCEVZeroExtendExpr>(V)) {
+ SCEVHandle Op = getSCEVAtScope(Cast->getOperand(), L);
+ if (Op == UnknownValue) return Op;
+ if (Op == Cast->getOperand())
+ return Cast; // must be loop invariant
+ return getZeroExtendExpr(Op, Cast->getType());
+ }
+
+ if (SCEVSignExtendExpr *Cast = dyn_cast<SCEVSignExtendExpr>(V)) {
+ SCEVHandle Op = getSCEVAtScope(Cast->getOperand(), L);
+ if (Op == UnknownValue) return Op;
+ if (Op == Cast->getOperand())
+ return Cast; // must be loop invariant
+ return getSignExtendExpr(Op, Cast->getType());
+ }
+
+ if (SCEVTruncateExpr *Cast = dyn_cast<SCEVTruncateExpr>(V)) {
+ SCEVHandle Op = getSCEVAtScope(Cast->getOperand(), L);
+ if (Op == UnknownValue) return Op;
+ if (Op == Cast->getOperand())
+ return Cast; // must be loop invariant
+ return getTruncateExpr(Op, Cast->getType());
+ }
+
+ assert(0 && "Unknown SCEV type!");
}
/// getSCEVAtScope - Return a SCEV expression handle for the specified value