diff options
| -rw-r--r-- | lib/Analysis/PostDominators.cpp | 6 | ||||
| -rw-r--r-- | lib/VMCore/Dominators.cpp | 6 |
2 files changed, 6 insertions, 6 deletions
diff --git a/lib/Analysis/PostDominators.cpp b/lib/Analysis/PostDominators.cpp index 244f8cdff5..d2892de8ed 100644 --- a/lib/Analysis/PostDominators.cpp +++ b/lib/Analysis/PostDominators.cpp @@ -29,7 +29,7 @@ F("postdomtree", "Post-Dominator Tree Construction", true); unsigned PostDominatorTree::DFSPass(BasicBlock *V, unsigned N) { std::vector<BasicBlock *> workStack; - std::set<BasicBlock *> visited; + SmallPtrSet<BasicBlock *, 32> Visited; workStack.push_back(V); do { @@ -37,7 +37,7 @@ unsigned PostDominatorTree::DFSPass(BasicBlock *V, unsigned N) { InfoRec &CurVInfo = Info[currentBB]; // Visit each block only once. - if (visited.insert(currentBB).second) { + if (Visited.insert(currentBB)) { CurVInfo.Semi = ++N; CurVInfo.Label = currentBB; @@ -55,7 +55,7 @@ unsigned PostDominatorTree::DFSPass(BasicBlock *V, unsigned N) { InfoRec &SuccVInfo = Info[*PI]; if (SuccVInfo.Semi == 0) { SuccVInfo.Parent = currentBB; - if (!visited.count(*PI)) { + if (!Visited.count(*PI)) { workStack.push_back(*PI); visitChild = true; } diff --git a/lib/VMCore/Dominators.cpp b/lib/VMCore/Dominators.cpp index 6231eb8e46..7d0fbaa29a 100644 --- a/lib/VMCore/Dominators.cpp +++ b/lib/VMCore/Dominators.cpp @@ -211,7 +211,7 @@ unsigned DominatorTree::DFSPass(BasicBlock *V, unsigned N) { void DominatorTree::Compress(BasicBlock *VIn) { std::vector<BasicBlock *> Work; - std::set<BasicBlock *> Visited; + SmallPtrSet<BasicBlock *, 32> Visited; BasicBlock *VInAncestor = Info[VIn].Ancestor; InfoRec &VInVAInfo = Info[VInAncestor]; @@ -225,9 +225,9 @@ void DominatorTree::Compress(BasicBlock *VIn) { InfoRec &VAInfo = Info[VAncestor]; // Process Ancestor first - if (Visited.count(VAncestor) == 0 && VAInfo.Ancestor != 0) { + if (Visited.insert(VAncestor) && + VAInfo.Ancestor != 0) { Work.push_back(VAncestor); - Visited.insert(VAncestor); continue; } Work.pop_back(); |