1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
/*
* Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "moco/Pass/Passes/ConstantFoldAdd.h"
#include "TestHelper.h"
#include <moco/IR/TFNodes.h>
#include <loco.h>
#include <stdex/Memory.h>
#include <gtest/gtest.h>
using namespace moco::test;
namespace
{
moco::TFConst *const_vector_init(loco::Graph *graph, std::vector<int32_t> values)
{
auto const_node = graph->nodes()->create<moco::TFConst>();
auto dim = values.size();
const_node->dtype(loco::DataType::S32);
const_node->rank(1);
const_node->dim(0).set(dim);
const_node->size<loco::DataType::S32>(dim);
for (int32_t i = 0; i < dim; ++i)
const_node->at<loco::DataType::S32>(i) = values[i];
return const_node;
}
} // namespace
TEST(ConstantFoldAdd, basic_vector)
{
loco::Graph graph;
auto add_node = graph.nodes()->create<moco::TFAdd>();
{
auto const_from_ss = const_vector_init(&graph, {1, 3, 5});
add_node->x(const_from_ss);
auto const_y = const_vector_init(&graph, {2});
add_node->y(const_y);
}
setup_output_node(&graph, add_node);
auto pass = stdex::make_unique<moco::ConstantFoldAdd>();
bool cont = true;
while (cont)
{
cont = pass->run(&graph);
}
auto ssnode = find_first_node_bytype<moco::TFAdd>(&graph);
ASSERT_EQ(ssnode, nullptr);
auto ssconst = find_first_node_bytype<moco::TFConst>(&graph);
ASSERT_NE(ssconst, nullptr);
ASSERT_EQ(ssconst->size<loco::DataType::S32>(), 3);
ASSERT_EQ(ssconst->at<loco::DataType::S32>(0), 3);
ASSERT_EQ(ssconst->at<loco::DataType::S32>(1), 5);
ASSERT_EQ(ssconst->at<loco::DataType::S32>(2), 7);
}
TEST(ConstantFoldAdd, basic_refinedet_1)
{
loco::Graph graph;
auto add_node = graph.nodes()->create<moco::TFAdd>();
{
auto const_from_ss = const_vector_init(&graph, {10});
add_node->x(const_from_ss);
auto const_y = const_vector_init(&graph, {0});
add_node->y(const_y);
}
setup_output_node(&graph, add_node);
auto pass = stdex::make_unique<moco::ConstantFoldAdd>();
bool cont = true;
while (cont)
{
cont = pass->run(&graph);
}
auto ssnode = find_first_node_bytype<moco::TFAdd>(&graph);
ASSERT_EQ(ssnode, nullptr);
auto ssconst = find_first_node_bytype<moco::TFConst>(&graph);
ASSERT_NE(ssconst, nullptr);
ASSERT_EQ(ssconst->size<loco::DataType::S32>(), 1);
ASSERT_EQ(ssconst->at<loco::DataType::S32>(0), 10);
}
|