1 /**
2 * Copyright (c) 2024 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "libabckit/include/c/metadata_core.h"
17 #include "libabckit/include/c/ir_core.h"
18 #include "libabckit/include/c/isa/isa_dynamic.h"
19 #include "libabckit/include/c/abckit.h"
20 #include "helpers/helpers.h"
21 #include "helpers/helpers_runtime.h"
22 #include <gtest/gtest.h>
23
24 // NOLINTBEGIN(readability-magic-numbers)
25
26 namespace libabckit::test {
27
28 static auto g_impl = AbckitGetApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
29 static auto g_implI = AbckitGetInspectApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
30 static auto g_implM = AbckitGetModifyApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
31 static auto g_implG = AbckitGetGraphApiImpl(ABCKIT_VERSION_RELEASE_1_0_0);
32 static auto g_dynG = AbckitGetIsaApiDynamicImpl(ABCKIT_VERSION_RELEASE_1_0_0);
33 static auto g_statG = AbckitGetIsaApiStaticImpl(ABCKIT_VERSION_RELEASE_1_0_0);
34
35 class LibAbcKitGetPhiTest : public ::testing::Test {};
36
37 namespace {
38
IRFindParameters(AbckitGraph * graph,bool isDynamic)39 std::pair<AbckitInst *, AbckitInst *> IRFindParameters(AbckitGraph *graph, bool isDynamic)
40 {
41 AbckitInst *param0 = nullptr;
42 AbckitInst *param1 = nullptr;
43 if (isDynamic) {
44 param1 = helpers::FindLastInst(graph, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER);
45 param0 = g_implG->iGetPrev(param1);
46 } else {
47 param0 = helpers::FindFirstInst(graph, ABCKIT_ISA_API_STATIC_OPCODE_PARAMETER);
48 param1 = helpers::FindFirstInst(graph, ABCKIT_ISA_API_STATIC_OPCODE_PARAMETER,
49 [counter = 0](AbckitInst * /*i*/) mutable { return counter++ > 0; });
50 }
51
52 return {param0, param1};
53 }
54
TransformIR(AbckitGraph * graph,bool isDynamic=false)55 void TransformIR(AbckitGraph *graph, bool isDynamic = false)
56 {
57 AbckitBasicBlock *startBB = g_implG->gGetStartBasicBlock(graph);
58 std::vector<AbckitBasicBlock *> succBBs = helpers::BBgetSuccBlocks(startBB);
59 AbckitBasicBlock *endBB = g_implG->gGetEndBasicBlock(graph);
60
61 auto [param0, param1] = IRFindParameters(graph, isDynamic);
62
63 ASSERT_NE(param0, nullptr);
64 ASSERT_NE(param1, nullptr);
65
66 g_implG->bbDisconnectSuccBlock(startBB, 0);
67 AbckitBasicBlock *ifBB = g_implG->bbCreateEmpty(graph);
68 g_implG->bbAppendSuccBlock(startBB, ifBB);
69 AbckitInst *ifInst = nullptr;
70 if (isDynamic) {
71 AbckitInst *intrinsicEq = g_dynG->iCreateEq(graph, param1, param0);
72 g_implG->bbAddInstBack(ifBB, intrinsicEq);
73 ifInst = g_dynG->iCreateIf(graph, intrinsicEq, ABCKIT_ISA_API_DYNAMIC_CONDITION_CODE_CC_EQ);
74 } else {
75 ifInst = g_statG->iCreateIf(graph, param0, param1, ABCKIT_ISA_API_STATIC_CONDITION_CODE_CC_NE);
76 }
77 g_implG->bbAddInstBack(ifBB, ifInst);
78
79 AbckitBasicBlock *trueBB = g_implG->bbCreateEmpty(graph);
80 g_implG->bbAppendSuccBlock(ifBB, trueBB);
81 AbckitInst *dec = nullptr;
82 if (isDynamic) {
83 dec = g_dynG->iCreateSub2(graph, param1, param0);
84 } else {
85 dec = g_statG->iCreateSub(graph, param0, param1);
86 }
87 g_implG->bbAddInstBack(trueBB, dec);
88
89 AbckitBasicBlock *falseBB = g_implG->bbCreateEmpty(graph);
90 g_implG->bbAppendSuccBlock(ifBB, falseBB);
91 AbckitInst *inc = nullptr;
92 if (isDynamic) {
93 inc = g_dynG->iCreateAdd2(graph, param1, param0);
94 } else {
95 inc = g_statG->iCreateAdd(graph, param0, param1);
96 }
97 g_implG->bbAddInstBack(falseBB, inc);
98
99 AbckitBasicBlock *retBB = g_implG->bbCreateEmpty(graph);
100 g_implG->bbAppendSuccBlock(trueBB, retBB);
101 g_implG->bbAppendSuccBlock(falseBB, retBB);
102 g_implG->bbAppendSuccBlock(retBB, endBB);
103 AbckitInst *phi = g_implG->bbCreatePhi(retBB, 2, dec, inc);
104 AbckitInst *ret = nullptr;
105 if (isDynamic) {
106 ret = g_dynG->iCreateReturn(graph, phi);
107 } else {
108 ret = g_statG->iCreateReturn(graph, phi);
109 }
110 g_implG->bbAddInstBack(retBB, ret);
111 }
112
113 } // namespace
114
115 // Test: test-kind=api, api=GraphApiImpl::bbCreatePhi, abc-kind=ArkTS2, category=positive, extension=c
TEST_F(LibAbcKitGetPhiTest,StaticBBcreatePhi)116 TEST_F(LibAbcKitGetPhiTest, StaticBBcreatePhi)
117 {
118 auto output =
119 helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/ir_core/phi/phi_static.abc", "phi_static/ETSGLOBAL", "main");
120 EXPECT_TRUE(helpers::Match(output, "2\n4\n"));
121
122 helpers::TransformMethod(
123 ABCKIT_ABC_DIR "ut/ir_core/phi/phi_static.abc", ABCKIT_ABC_DIR "ut/ir_core/phi/phi_static_modified.abc", "foo",
124 [&](AbckitFile * /*file*/, AbckitCoreFunction * /*method*/, AbckitGraph *graph) { TransformIR(graph); },
125 [&](AbckitGraph *graph) {
126 std::vector<helpers::BBSchema<AbckitIsaApiStaticOpcode>> schema = {
127 {{},
128 {1},
129 {
130 {0, ABCKIT_ISA_API_STATIC_OPCODE_PARAMETER, {}},
131 {1, ABCKIT_ISA_API_STATIC_OPCODE_PARAMETER, {}},
132 }},
133 {{0},
134 {3, 2},
135 {
136 {2, ABCKIT_ISA_API_STATIC_OPCODE_IF, {0, 1}},
137 }},
138 {{1},
139 {4},
140 {
141 {3, ABCKIT_ISA_API_STATIC_OPCODE_ADD, {0, 1}},
142 }},
143 {{1},
144 {4},
145 {
146 {4, ABCKIT_ISA_API_STATIC_OPCODE_SUB, {0, 1}},
147 }},
148 {{2, 3},
149 {5},
150 {
151 {5, ABCKIT_ISA_API_STATIC_OPCODE_PHI, {3, 4}},
152 {6, ABCKIT_ISA_API_STATIC_OPCODE_RETURN, {5}},
153 }},
154 {{4}, {}, {}}};
155 helpers::VerifyGraph(graph, schema);
156 });
157
158 output = helpers::ExecuteStaticAbc(ABCKIT_ABC_DIR "ut/ir_core/phi/phi_static_modified.abc", "phi_static/ETSGLOBAL",
159 "main");
160 EXPECT_TRUE(helpers::Match(output, "-1\n4\n"));
161 }
162
163 // Test: test-kind=api, api=GraphApiImpl::bbCreatePhi, abc-kind=ArkTS1, category=positive, extension=c
TEST_F(LibAbcKitGetPhiTest,DynamicBBcreatePhi)164 TEST_F(LibAbcKitGetPhiTest, DynamicBBcreatePhi)
165 {
166 auto output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/ir_core/phi/phi_dynamic.abc", "phi_dynamic");
167 EXPECT_TRUE(helpers::Match(output, "2\n4\n"));
168
169 helpers::TransformMethod(
170 ABCKIT_ABC_DIR "ut/ir_core/phi/phi_dynamic.abc", ABCKIT_ABC_DIR "ut/ir_core/phi/phi_dynamic_modified.abc",
171 "foo",
172 [&](AbckitFile * /*file*/, AbckitCoreFunction * /*method*/, AbckitGraph *graph) { TransformIR(graph, true); },
173 [&](AbckitGraph *graph) {
174 std::vector<helpers::BBSchema<AbckitIsaApiDynamicOpcode>> schema = {
175 {{},
176 {1},
177 {
178 {9, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
179 {10, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
180 {11, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
181 {0, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
182 {1, ABCKIT_ISA_API_DYNAMIC_OPCODE_PARAMETER, {}},
183 {2, ABCKIT_ISA_API_DYNAMIC_OPCODE_CONSTANT, {}},
184 }},
185 {{0},
186 {3, 2},
187 {
188 {3, ABCKIT_ISA_API_DYNAMIC_OPCODE_EQ, {1, 0}},
189 {4, ABCKIT_ISA_API_DYNAMIC_OPCODE_IF, {3, 2}},
190 }},
191 {{1},
192 {4},
193 {
194 {5, ABCKIT_ISA_API_DYNAMIC_OPCODE_ADD2, {1, 0}},
195 }},
196 {{1},
197 {4},
198 {
199 {6, ABCKIT_ISA_API_DYNAMIC_OPCODE_SUB2, {1, 0}},
200 }},
201 {{2, 3},
202 {5},
203 {
204 {7, ABCKIT_ISA_API_DYNAMIC_OPCODE_PHI, {5, 6}},
205 {8, ABCKIT_ISA_API_DYNAMIC_OPCODE_RETURN, {7}},
206 }},
207 {{4}, {}, {}}};
208 helpers::VerifyGraph(graph, schema);
209 });
210
211 output = helpers::ExecuteDynamicAbc(ABCKIT_ABC_DIR "ut/ir_core/phi/phi_dynamic_modified.abc", "phi_dynamic");
212 EXPECT_TRUE(helpers::Match(output, "-1\n4\n"));
213 }
214
215 } // namespace libabckit::test
216
217 // NOLINTEND(readability-magic-numbers)
218