[PATCH] D47134: [LoopVersioning] Don't modify the list that we iterate over in addPHINodes
Bjorn Pettersson via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Mon May 21 13:27:22 PDT 2018
bjope updated this revision to Diff 147854.
bjope added a comment.
Updated according to suggestions from @mzolotukhin.
Thanks for the review (I'll push this tomorrow when I can observe the build bots).
Repository:
rL LLVM
https://reviews.llvm.org/D47134
Files:
lib/Transforms/Utils/LoopVersioning.cpp
test/Transforms/LoopVersioning/add-phi-update-users.ll
Index: test/Transforms/LoopVersioning/add-phi-update-users.ll
===================================================================
--- /dev/null
+++ test/Transforms/LoopVersioning/add-phi-update-users.ll
@@ -0,0 +1,65 @@
+; RUN: opt < %s -loop-versioning -S -o - | FileCheck %s
+
+; This test case used to end like this:
+;
+; Instruction does not dominate all uses!
+; %t2 = load i16, i16* @b, align 1, !tbaa !2, !alias.scope !6
+; %tobool = icmp eq i16 %t2, 0
+; LLVM ERROR: Broken function found, compilation aborted!
+;
+; due to a fault where we did not replace the use of %t2 in the icmp in
+; for.end, when adding a new PHI node for the versioned loops based on the
+; loop-defined values used outside of the loop.
+;
+; Verify that the code compiles, that we get a versioned loop, and that the
+; uses of %t2 in for.end and if.then are updated to use the value from the
+; added phi node.
+
+; CHECK: define void @f1
+; CHECK: for.end:
+; CHECK-NEXT: %t2.lver = phi i16 [ %t2, %for.body ], [ %t2.lver.orig, %for.body.lver.orig ]
+; CHECK-NEXT: %tobool = icmp eq i16 %t2.lver, 0
+; CHECK: if.then:
+; CHECK-NEXT: store i16 %t2.lver
+
+ at a = dso_local global i16 0, align 1
+ at b = dso_local global i16 0, align 1
+ at c = dso_local global i16* null, align 1
+
+define void @f1() {
+entry:
+ %t0 = load i16*, i16** @c, align 1
+ br label %for.cond
+
+for.cond: ; preds = %for.cond.backedge, %entry
+ br label %for.body
+
+for.body: ; preds = %for.cond, %for.body
+ %t1 = phi i64 [ 0, %for.cond ], [ %inc, %for.body ]
+ %t2 = load i16, i16* @b, align 1, !tbaa !2
+ store i16 %t2, i16* %t0, align 1, !tbaa !2
+ %inc = add nuw nsw i64 %t1, 1
+ %cmp = icmp ult i64 %inc, 3
+ br i1 %cmp, label %for.body, label %for.end
+
+for.end: ; preds = %for.body
+ %tobool = icmp eq i16 %t2, 0
+ br i1 %tobool, label %for.cond.backedge, label %if.then
+
+for.cond.backedge: ; preds = %for.end, %if.then
+ br label %for.cond
+
+if.then: ; preds = %for.end
+ store i16 %t2, i16* @a, align 1, !tbaa !2
+ br label %for.cond.backedge
+}
+
+!llvm.module.flags = !{!0}
+!llvm.ident = !{!1}
+
+!0 = !{i32 1, !"wchar_size", i32 1}
+!1 = !{!"clang version 7.0.0"}
+!2 = !{!3, !3, i64 0}
+!3 = !{!"long long", !4, i64 0}
+!4 = !{!"omnipotent char", !5, i64 0}
+!5 = !{!"Simple C/C++ TBAA"}
Index: lib/Transforms/Utils/LoopVersioning.cpp
===================================================================
--- lib/Transforms/Utils/LoopVersioning.cpp
+++ lib/Transforms/Utils/LoopVersioning.cpp
@@ -140,9 +140,12 @@
if (!PN) {
PN = PHINode::Create(Inst->getType(), 2, Inst->getName() + ".lver",
&PHIBlock->front());
- for (auto *User : Inst->users())
- if (!VersionedLoop->contains(cast<Instruction>(User)->getParent()))
- User->replaceUsesOfWith(Inst, PN);
+ SmallVector<User*, 8> UsersToUpdate;
+ for (User *U : Inst->users())
+ if (!VersionedLoop->contains(cast<Instruction>(U)->getParent()))
+ UsersToUpdate.push_back(U);
+ for (User *U : UsersToUpdate)
+ U->replaceUsesOfWith(Inst, PN);
PN->addIncoming(Inst, VersionedLoop->getExitingBlock());
}
}
-------------- next part --------------
A non-text attachment was scrubbed...
Name: D47134.147854.patch
Type: text/x-patch
Size: 3387 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20180521/53aeabd1/attachment.bin>
More information about the llvm-commits
mailing list