OLD | NEW |
1 // Copyright 2014 the V8 project authors. All rights reserved. | 1 // Copyright 2014 the V8 project authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "src/compiler/graph-reducer.h" | 5 #include "src/compiler/graph-reducer.h" |
6 | 6 |
7 #include <functional> | 7 #include <functional> |
8 | 8 |
9 #include "src/compiler/graph-inl.h" | 9 #include "src/compiler/graph-inl.h" |
10 | 10 |
11 namespace v8 { | 11 namespace v8 { |
12 namespace internal { | 12 namespace internal { |
13 namespace compiler { | 13 namespace compiler { |
14 | 14 |
15 enum class GraphReducer::State : uint8_t { | 15 GraphReducer::GraphReducer(Graph* graph) |
16 kUnvisited, | 16 : graph_(graph), reducers_(graph->zone()) {} |
17 kRevisit, | |
18 kOnStack, | |
19 kVisited | |
20 }; | |
21 | 17 |
22 | 18 |
23 GraphReducer::GraphReducer(Graph* graph, Zone* zone) | 19 static bool NodeIdIsLessThan(const Node* node, NodeId id) { |
24 : graph_(graph), | 20 return node->id() < id; |
25 reducers_(zone), | |
26 revisit_(zone), | |
27 stack_(zone), | |
28 state_(zone) {} | |
29 | |
30 | |
31 void GraphReducer::AddReducer(Reducer* reducer) { | |
32 reducers_.push_back(reducer); | |
33 } | 21 } |
34 | 22 |
35 | 23 |
36 void GraphReducer::ReduceNode(Node* node) { | 24 void GraphReducer::ReduceNode(Node* node) { |
37 DCHECK(stack_.empty()); | 25 static const unsigned kMaxAttempts = 16; |
38 DCHECK(revisit_.empty()); | 26 bool reduce = true; |
39 std::fill(state_.begin(), state_.end(), State::kUnvisited); | 27 for (unsigned attempts = 0; attempts <= kMaxAttempts; ++attempts) { |
40 Push(node); | 28 if (!reduce) return; |
41 for (;;) { | 29 reduce = false; // Assume we don't need to rerun any reducers. |
42 DCHECK(!stack_.empty() || | 30 int before = graph_->NodeCount(); |
43 std::find(state_.begin(), state_.end(), State::kOnStack) == | 31 for (ZoneVector<Reducer*>::iterator i = reducers_.begin(); |
44 state_.end()); | 32 i != reducers_.end(); ++i) { |
45 if (!stack_.empty()) { | |
46 // Process the node on the top of the stack, potentially pushing more or | |
47 // popping the node off the stack. | |
48 ReduceTop(); | |
49 } else if (!revisit_.empty()) { | |
50 // If the stack becomes empty, revisit any nodes in the revisit queue. | |
51 Node* const node = revisit_.top(); | |
52 revisit_.pop(); | |
53 if (state_[node->id()] == State::kRevisit) { | |
54 // state can change while in queue. | |
55 Push(node); | |
56 } | |
57 } else { | |
58 break; | |
59 } | |
60 } | |
61 DCHECK(std::find(state_.begin(), state_.end(), State::kOnStack) == | |
62 state_.end()); | |
63 DCHECK(revisit_.empty()); | |
64 DCHECK(stack_.empty()); | |
65 } | |
66 | |
67 | |
68 void GraphReducer::ReduceGraph() { ReduceNode(graph()->end()); } | |
69 | |
70 | |
71 Reduction GraphReducer::Reduce(Node* const node) { | |
72 auto skip = reducers_.end(); | |
73 for (auto i = reducers_.begin(); i != reducers_.end();) { | |
74 if (i != skip) { | |
75 Reduction reduction = (*i)->Reduce(node); | 33 Reduction reduction = (*i)->Reduce(node); |
76 if (!reduction.Changed()) { | 34 Node* replacement = reduction.replacement(); |
| 35 if (replacement == NULL) { |
77 // No change from this reducer. | 36 // No change from this reducer. |
78 } else if (reduction.replacement() == node) { | 37 } else if (replacement == node) { |
79 // {replacement} == {node} represents an in-place reduction. Rerun | 38 // {replacement == node} represents an in-place reduction. |
80 // all the other reducers for this node, as now there may be more | 39 // Rerun all the reducers for this node, as now there may be more |
81 // opportunities for reduction. | 40 // opportunities for reduction. |
82 skip = i; | 41 reduce = true; |
83 i = reducers_.begin(); | 42 break; |
84 continue; | |
85 } else { | 43 } else { |
86 // {node} was replaced by another node. | 44 if (node == graph_->start()) graph_->SetStart(replacement); |
87 return reduction; | 45 if (node == graph_->end()) graph_->SetEnd(replacement); |
88 } | 46 // If {node} was replaced by an old node, unlink {node} and assume that |
89 } | 47 // {replacement} was already reduced and finish. |
90 ++i; | 48 if (replacement->id() < before) { |
91 } | 49 node->ReplaceUses(replacement); |
92 if (skip == reducers_.end()) { | 50 node->Kill(); |
93 // No change from any reducer. | 51 return; |
94 return Reducer::NoChange(); | 52 } |
95 } | 53 // Otherwise, {node} was replaced by a new node. Replace all old uses of |
96 // At least one reducer did some in-place reduction. | |
97 return Reducer::Changed(node); | |
98 } | |
99 | |
100 | |
101 void GraphReducer::ReduceTop() { | |
102 Node* const node = Top(); | |
103 if (node->IsDead()) return Pop(); // Node was killed while on stack. | |
104 | |
105 // Recurse on an input if necessary. | |
106 for (auto const input : node->inputs()) { | |
107 if (input != node && Recurse(input)) return; | |
108 } | |
109 | |
110 // Remember the node count before reduction. | |
111 const int node_count = graph()->NodeCount(); | |
112 | |
113 // All inputs should be visited or on stack. Apply reductions to node. | |
114 Reduction reduction = Reduce(node); | |
115 | |
116 // After reducing the node, pop it off the stack. | |
117 Pop(); | |
118 | |
119 // If there was a reduction, revisit the uses and reduce the replacement. | |
120 if (reduction.Changed()) { | |
121 for (Node* const use : node->uses()) { | |
122 // Don't revisit this node if it refers to itself. | |
123 if (use != node) Revisit(use); | |
124 } | |
125 Node* const replacement = reduction.replacement(); | |
126 if (replacement != node) { | |
127 if (node == graph()->start()) graph()->SetStart(replacement); | |
128 if (node == graph()->end()) graph()->SetEnd(replacement); | |
129 // If {node} was replaced by an old node, unlink {node} and assume that | |
130 // {replacement} was already reduced and finish. | |
131 if (replacement->id() < node_count) { | |
132 node->ReplaceUses(replacement); | |
133 node->Kill(); | |
134 } else { | |
135 // Otherwise {node} was replaced by a new node. Replace all old uses of | |
136 // {node} with {replacement}. New nodes created by this reduction can | 54 // {node} with {replacement}. New nodes created by this reduction can |
137 // use {node}. | 55 // use {node}. |
138 node->ReplaceUsesIf([node_count](Node* const node) { | 56 node->ReplaceUsesIf( |
139 return node->id() < node_count; | 57 std::bind2nd(std::ptr_fun(&NodeIdIsLessThan), before), replacement); |
140 }, | |
141 replacement); | |
142 // Unlink {node} if it's no longer used. | 58 // Unlink {node} if it's no longer used. |
143 if (node->uses().empty()) { | 59 if (node->uses().empty()) { |
144 node->Kill(); | 60 node->Kill(); |
145 } | 61 } |
146 | 62 // Rerun all the reductions on the {replacement}. |
147 // If there was a replacement, reduce it after popping {node}. | 63 node = replacement; |
148 Recurse(replacement); | 64 reduce = true; |
| 65 break; |
149 } | 66 } |
150 } | 67 } |
151 } | 68 } |
152 } | 69 } |
153 | 70 |
154 | 71 |
155 void GraphReducer::Pop() { | 72 // A helper class to reuse the node traversal algorithm. |
156 Node* const node = Top(); | 73 struct GraphReducerVisitor FINAL : public NullNodeVisitor { |
157 state_[node->id()] = State::kVisited; | 74 explicit GraphReducerVisitor(GraphReducer* reducer) : reducer_(reducer) {} |
158 stack_.pop(); | 75 void Post(Node* node) { reducer_->ReduceNode(node); } |
| 76 GraphReducer* reducer_; |
| 77 }; |
| 78 |
| 79 |
| 80 void GraphReducer::ReduceGraph() { |
| 81 GraphReducerVisitor visitor(this); |
| 82 // Perform a post-order reduction of all nodes starting from the end. |
| 83 graph()->VisitNodeInputsFromEnd(&visitor); |
159 } | 84 } |
160 | 85 |
161 | 86 |
162 void GraphReducer::Push(Node* const node) { | 87 // TODO(titzer): partial graph reductions. |
163 size_t const id = static_cast<size_t>(node->id()); | |
164 if (id >= state_.size()) state_.resize(id + 1); | |
165 DCHECK(id < state_.size()); | |
166 DCHECK(state_[id] != State::kOnStack); | |
167 state_[id] = State::kOnStack; | |
168 stack_.push(node); | |
169 } | |
170 | |
171 | |
172 Node* GraphReducer::Top() const { | |
173 DCHECK(!stack_.empty()); | |
174 Node* const node = stack_.top(); | |
175 size_t const id = static_cast<size_t>(node->id()); | |
176 DCHECK(id < state_.size()); | |
177 DCHECK(state_[id] == State::kOnStack); | |
178 USE(id); | |
179 return node; | |
180 } | |
181 | |
182 | |
183 bool GraphReducer::Recurse(Node* const node) { | |
184 size_t const id = static_cast<size_t>(node->id()); | |
185 if (id < state_.size() && state_[id] > State::kRevisit) return false; | |
186 Push(node); | |
187 return true; | |
188 } | |
189 | |
190 | |
191 void GraphReducer::Revisit(Node* const node) { | |
192 size_t const id = static_cast<size_t>(node->id()); | |
193 if (id < state_.size() && state_[id] == State::kVisited) { | |
194 state_[id] = State::kRevisit; | |
195 revisit_.push(node); | |
196 } | |
197 } | |
198 | 88 |
199 } // namespace compiler | 89 } // namespace compiler |
200 } // namespace internal | 90 } // namespace internal |
201 } // namespace v8 | 91 } // namespace v8 |
OLD | NEW |