Skip to content

Commit

Permalink
Fix an issue where SROA computed different results based on the relative
Browse files Browse the repository at this point in the history
order of slices of the alloca which have exactly the same size and other
properties. This was found by a perniciously unstable sort
implementation used to flush out buggy uses of the algorithm.

The fundamental idea is that findCommonType should return the best
common type it can find across all of the slices in the range. There
were two bugs here previously:

1) We would accept an integer type smaller than a byte-width multiple,
   and if there were different bit-width integer types, we would accept
   the first one. This caused an actual failure in the testcase updated
   here when the sort order changed.
2) If we found a bad combination of types or a non-load, non-store use
   before an integer typed load or store we would bail, but if we found
   the integere typed load or store, we would use it. The correct
   behavior is to always use an integer typed operation which covers the
   partition if one exists.

While a clever debugging sort algorithm found problem brson#1 in our existing
test cases, I have no useful test case ideas for brson#2. I spotted in by
inspection when looking at this code.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@195118 91177308-0d34-0410-b5e6-96231b3b80d8
  • Loading branch information
chandlerc committed Nov 19, 2013
1 parent 1b3ab91 commit ed1951e
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 11 deletions.
29 changes: 19 additions & 10 deletions lib/Transforms/Scalar/SROA.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -938,6 +938,7 @@ static Type *findCommonType(AllocaSlices::const_iterator B,
AllocaSlices::const_iterator E,
uint64_t EndOffset) {
Type *Ty = 0;
bool IgnoreNonIntegralTypes = false;
for (AllocaSlices::const_iterator I = B; I != E; ++I) {
Use *U = I->getUse();
if (isa<IntrinsicInst>(*U->getUser()))
Expand All @@ -946,29 +947,37 @@ static Type *findCommonType(AllocaSlices::const_iterator B,
continue;

Type *UserTy = 0;
if (LoadInst *LI = dyn_cast<LoadInst>(U->getUser()))
if (LoadInst *LI = dyn_cast<LoadInst>(U->getUser())) {
UserTy = LI->getType();
else if (StoreInst *SI = dyn_cast<StoreInst>(U->getUser()))
} else if (StoreInst *SI = dyn_cast<StoreInst>(U->getUser())) {
UserTy = SI->getValueOperand()->getType();
else
return 0; // Bail if we have weird uses.
} else {
IgnoreNonIntegralTypes = true; // Give up on anything but an iN type.
continue;
}

if (IntegerType *ITy = dyn_cast<IntegerType>(UserTy)) {
// If the type is larger than the partition, skip it. We only encounter
// this for split integer operations where we want to use the type of the
// entity causing the split.
if (ITy->getBitWidth() / 8 > (EndOffset - B->beginOffset()))
// entity causing the split. Also skip if the type is not a byte width
// multiple.
if (ITy->getBitWidth() % 8 != 0 ||
ITy->getBitWidth() / 8 > (EndOffset - B->beginOffset()))
continue;

// If we have found an integer type use covering the alloca, use that
// regardless of the other types, as integers are often used for a
// "bucket
// of bits" type.
// regardless of the other types, as integers are often used for
// a "bucket of bits" type.
//
// NB: This *must* be the only return from inside the loop so that the
// order of slices doesn't impact the computed type.
return ITy;
} else if (IgnoreNonIntegralTypes) {
continue;
}

if (Ty && Ty != UserTy)
return 0;
IgnoreNonIntegralTypes = true; // Give up on anything but an iN type.

Ty = UserTy;
}
Expand Down
1 change: 0 additions & 1 deletion test/Transforms/SROA/basictest.ll
Original file line number Diff line number Diff line change
Expand Up @@ -1181,7 +1181,6 @@ entry:
store i1 %x, i1* %b.i1, align 8
%b.i8 = bitcast <{ i1 }>* %b to i8*
%foo = load i8* %b.i8, align 1
; CHECK-NEXT: {{.*}} = zext i1 %x to i8
; CHECK-NEXT: %[[ext:.*]] = zext i1 %x to i8
; CHECK-NEXT: store i8 %[[ext]], i8* %[[a]], align 8
; CHECK-NEXT: {{.*}} = load i8* %[[a]], align 8
Expand Down

0 comments on commit ed1951e

Please sign in to comment.