OLD | NEW |
(Empty) | |
| 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 |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "src/compiler/graph.h" |
| 6 #include "src/compiler/graph-reducer.h" |
| 7 #include "src/compiler/operator.h" |
| 8 #include "src/test/test-utils.h" |
| 9 #include "testing/gmock/include/gmock/gmock.h" |
| 10 |
| 11 using testing::_; |
| 12 using testing::DefaultValue; |
| 13 using testing::Return; |
| 14 using testing::Sequence; |
| 15 using testing::StrictMock; |
| 16 |
| 17 namespace v8 { |
| 18 namespace internal { |
| 19 namespace compiler { |
| 20 |
| 21 namespace { |
| 22 |
| 23 SimpleOperator OP0(0, Operator::kNoWrite, 0, 0, "op0"); |
| 24 |
| 25 |
| 26 struct MockReducer : public Reducer { |
| 27 MOCK_METHOD1(Reduce, Reduction(Node*)); |
| 28 }; |
| 29 |
| 30 } // namespace |
| 31 |
| 32 |
| 33 class GraphReducerTest : public TestWithZone { |
| 34 public: |
| 35 GraphReducerTest() : graph_(zone()) {} |
| 36 |
| 37 static void SetUpTestCase() { |
| 38 TestWithZone::SetUpTestCase(); |
| 39 DefaultValue<Reduction>::Set(Reducer::NoChange()); |
| 40 } |
| 41 |
| 42 static void TearDownTestCase() { |
| 43 DefaultValue<Reduction>::Clear(); |
| 44 TestWithZone::TearDownTestCase(); |
| 45 } |
| 46 |
| 47 protected: |
| 48 void ReduceNode(Node* node, Reducer* r) { |
| 49 GraphReducer reducer(graph()); |
| 50 reducer.AddReducer(r); |
| 51 reducer.ReduceNode(node); |
| 52 } |
| 53 |
| 54 void ReduceNode(Node* node, Reducer* r1, Reducer* r2) { |
| 55 GraphReducer reducer(graph()); |
| 56 reducer.AddReducer(r1); |
| 57 reducer.AddReducer(r2); |
| 58 reducer.ReduceNode(node); |
| 59 } |
| 60 |
| 61 void ReduceNode(Node* node, Reducer* r1, Reducer* r2, Reducer* r3) { |
| 62 GraphReducer reducer(graph()); |
| 63 reducer.AddReducer(r1); |
| 64 reducer.AddReducer(r2); |
| 65 reducer.AddReducer(r3); |
| 66 reducer.ReduceNode(node); |
| 67 } |
| 68 |
| 69 Graph* graph() { return &graph_; } |
| 70 |
| 71 private: |
| 72 Graph graph_; |
| 73 }; |
| 74 |
| 75 |
| 76 TEST_F(GraphReducerTest, ReduceOnceForEveryReducer) { |
| 77 StrictMock<MockReducer> r1, r2; |
| 78 Node* node0 = graph()->NewNode(&OP0); |
| 79 EXPECT_CALL(r1, Reduce(node0)); |
| 80 EXPECT_CALL(r2, Reduce(node0)); |
| 81 ReduceNode(node0, &r1, &r2); |
| 82 } |
| 83 |
| 84 |
| 85 TEST_F(GraphReducerTest, ReduceAgainAfterChanged) { |
| 86 Sequence s1, s2; |
| 87 StrictMock<MockReducer> r1, r2, r3; |
| 88 Node* node0 = graph()->NewNode(&OP0); |
| 89 EXPECT_CALL(r1, Reduce(node0)); |
| 90 EXPECT_CALL(r2, Reduce(node0)); |
| 91 EXPECT_CALL(r3, Reduce(node0)).InSequence(s1, s2).WillOnce( |
| 92 Return(Reducer::Changed(node0))); |
| 93 EXPECT_CALL(r1, Reduce(node0)).InSequence(s1); |
| 94 EXPECT_CALL(r2, Reduce(node0)).InSequence(s2); |
| 95 ReduceNode(node0, &r1, &r2, &r3); |
| 96 } |
| 97 |
| 98 |
| 99 TEST_F(GraphReducerTest, OperatorIsNullAfterReplace) { |
| 100 StrictMock<MockReducer> r; |
| 101 Node* node0 = graph()->NewNode(&OP0); |
| 102 Node* node1 = graph()->NewNode(&OP0); |
| 103 EXPECT_CALL(r, Reduce(node0)).WillOnce(Return(Reducer::Replace(node1))); |
| 104 ReduceNode(node0, &r); |
| 105 EXPECT_EQ(NULL, node0->op()); |
| 106 EXPECT_EQ(&OP0, node1->op()); |
| 107 } |
| 108 |
| 109 } // namespace compiler |
| 110 } // namespace internal |
| 111 } // namespace v8 |
OLD | NEW |