mirror of
https://github.com/c64scene-ar/llvm-6502.git
synced 2024-08-23 02:29:18 +00:00
LoopVectorizer: Extend the induction variable to a larger type
In some case the loop exit count computation can overflow. Extend the type to prevent most of those cases. The problem is loops like: int main () { int a = 1; char b = 0; lbl: a &= 4; b--; if (b) goto lbl; return a; } The backedge count is 255. The induction variable type is i8. If we add one to 255 to get the exit count we overflow to zero. To work around this issue we extend the type of the induction variable to i32 in the case of i8 and i16. PR17532 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@195008 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
parent
23427207ea
commit
07a3c481c6
@ -1537,6 +1537,7 @@ InnerLoopVectorizer::createEmptyLoop(LoopVectorizationLegality *Legal) {
|
|||||||
const SCEV *ExitCount = SE->getBackedgeTakenCount(OrigLoop);
|
const SCEV *ExitCount = SE->getBackedgeTakenCount(OrigLoop);
|
||||||
assert(ExitCount != SE->getCouldNotCompute() && "Invalid loop count");
|
assert(ExitCount != SE->getCouldNotCompute() && "Invalid loop count");
|
||||||
|
|
||||||
|
ExitCount = SE->getNoopOrZeroExtend(ExitCount, IdxTy);
|
||||||
// Get the total trip count from the count by adding 1.
|
// Get the total trip count from the count by adding 1.
|
||||||
ExitCount = SE->getAddExpr(ExitCount,
|
ExitCount = SE->getAddExpr(ExitCount,
|
||||||
SE->getConstant(ExitCount->getType(), 1));
|
SE->getConstant(ExitCount->getType(), 1));
|
||||||
@ -2888,6 +2889,11 @@ static Type *convertPointerToIntegerType(DataLayout &DL, Type *Ty) {
|
|||||||
if (Ty->isPointerTy())
|
if (Ty->isPointerTy())
|
||||||
return DL.getIntPtrType(Ty);
|
return DL.getIntPtrType(Ty);
|
||||||
|
|
||||||
|
// It is possible that char's or short's overflow when we ask for the loop's
|
||||||
|
// trip count, work around this by changing the type size.
|
||||||
|
if (Ty->getScalarSizeInBits() < 32)
|
||||||
|
return Type::getInt32Ty(Ty->getContext());
|
||||||
|
|
||||||
return Ty;
|
return Ty;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -66,3 +66,45 @@ for.body:
|
|||||||
loopexit:
|
loopexit:
|
||||||
ret void
|
ret void
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
; Make sure that the loop exit count computation does not overflow for i8 and
|
||||||
|
; i16. The exit count of these loops is i8/i16 max + 1. If we don't cast the
|
||||||
|
; induction variable to a bigger type the exit count computation will overflow
|
||||||
|
; to 0.
|
||||||
|
; PR17532
|
||||||
|
|
||||||
|
; CHECK-LABEL: i8_loop
|
||||||
|
; CHECK; icmp eq i32 {{.*}}, 256
|
||||||
|
define i32 @i8_loop() nounwind readnone ssp uwtable {
|
||||||
|
br label %1
|
||||||
|
|
||||||
|
; <label>:1 ; preds = %1, %0
|
||||||
|
%a.0 = phi i32 [ 1, %0 ], [ %2, %1 ]
|
||||||
|
%b.0 = phi i8 [ 0, %0 ], [ %3, %1 ]
|
||||||
|
%2 = and i32 %a.0, 4
|
||||||
|
%3 = add i8 %b.0, -1
|
||||||
|
%4 = icmp eq i8 %3, 0
|
||||||
|
br i1 %4, label %5, label %1
|
||||||
|
|
||||||
|
; <label>:5 ; preds = %1
|
||||||
|
ret i32 %2
|
||||||
|
}
|
||||||
|
|
||||||
|
; CHECK-LABEL: i16_loop
|
||||||
|
; CHECK; icmp eq i32 {{.*}}, 65536
|
||||||
|
|
||||||
|
define i32 @i16_loop() nounwind readnone ssp uwtable {
|
||||||
|
br label %1
|
||||||
|
|
||||||
|
; <label>:1 ; preds = %1, %0
|
||||||
|
%a.0 = phi i32 [ 1, %0 ], [ %2, %1 ]
|
||||||
|
%b.0 = phi i16 [ 0, %0 ], [ %3, %1 ]
|
||||||
|
%2 = and i32 %a.0, 4
|
||||||
|
%3 = add i16 %b.0, -1
|
||||||
|
%4 = icmp eq i16 %3, 0
|
||||||
|
br i1 %4, label %5, label %1
|
||||||
|
|
||||||
|
; <label>:5 ; preds = %1
|
||||||
|
ret i32 %2
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user