[llvm] 9d555b4 - [DFAJumpThreading] make update order deterministic
Alex Zhikhartsev via llvm-commits
llvm-commits at lists.llvm.org
Tue Feb 1 08:04:07 PST 2022
Author: Olle Fredriksson
Date: 2022-02-01T11:02:58-05:00
New Revision: 9d555b4a83bf72450d417c69aa0be250f0a490e4
URL: https://github.com/llvm/llvm-project/commit/9d555b4a83bf72450d417c69aa0be250f0a490e4
DIFF: https://github.com/llvm/llvm-project/commit/9d555b4a83bf72450d417c69aa0be250f0a490e4.diff
LOG: [DFAJumpThreading] make update order deterministic
We tracked down some non-determinism in compilation output to the
DFAJumpThreading pass. These changes fixed our issue:
* Make the DefMap type a MapVector to make its iteration order depend on
insertion order.
* Sort the values to be inserted into NewDefs by instruction order to
make the insertion order deterministic. Since these values come from
iterating over a ValueMap, which doesn't have deterministic iteration
order, I couldn't fix this at its source.
Reviewed By: alexey.zhikhar
Differential Revision: https://reviews.llvm.org/D118590
Added:
Modified:
llvm/lib/Transforms/Scalar/DFAJumpThreading.cpp
Removed:
################################################################################
diff --git a/llvm/lib/Transforms/Scalar/DFAJumpThreading.cpp b/llvm/lib/Transforms/Scalar/DFAJumpThreading.cpp
index dda1a2f080768..143a78f604fc9 100644
--- a/llvm/lib/Transforms/Scalar/DFAJumpThreading.cpp
+++ b/llvm/lib/Transforms/Scalar/DFAJumpThreading.cpp
@@ -357,7 +357,7 @@ typedef DenseMap<BasicBlock *, CloneList> DuplicateBlockMap;
// This map keeps track of all the new definitions for an instruction. This
// information is needed when restoring SSA form after cloning blocks.
-typedef DenseMap<Instruction *, std::vector<Instruction *>> DefMap;
+typedef MapVector<Instruction *, std::vector<Instruction *>> DefMap;
inline raw_ostream &operator<<(raw_ostream &OS, const PathType &Path) {
OS << "< ";
@@ -1126,6 +1126,9 @@ struct TransformDFA {
/// Add new value mappings to the DefMap to keep track of all new definitions
/// for a particular instruction. These will be used while updating SSA form.
void updateDefMap(DefMap &NewDefs, ValueToValueMapTy &VMap) {
+ SmallVector<std::pair<Instruction *, Instruction *>> NewDefsVector;
+ NewDefsVector.reserve(VMap.size());
+
for (auto Entry : VMap) {
Instruction *Inst =
dyn_cast<Instruction>(const_cast<Value *>(Entry.first));
@@ -1138,11 +1141,18 @@ struct TransformDFA {
if (!Cloned)
continue;
- if (NewDefs.find(Inst) == NewDefs.end())
- NewDefs[Inst] = {Cloned};
- else
- NewDefs[Inst].push_back(Cloned);
+ NewDefsVector.push_back({Inst, Cloned});
}
+
+ // Sort the defs to get deterministic insertion order into NewDefs.
+ sort(NewDefsVector, [](const auto &LHS, const auto &RHS) {
+ if (LHS.first == RHS.first)
+ return LHS.second->comesBefore(RHS.second);
+ return LHS.first->comesBefore(RHS.first);
+ });
+
+ for (const auto &KV : NewDefsVector)
+ NewDefs[KV.first].push_back(KV.second);
}
/// Update the last branch of a particular cloned path to point to the correct
More information about the llvm-commits
mailing list