• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright (c) 2022 Huawei Device Co., Ltd.
2# Licensed under the Apache License, Version 2.0 (the "License");
3# you may not use this file except in compliance with the License.
4# You may obtain a copy of the License at
5#
6#     http://www.apache.org/licenses/LICENSE-2.0
7#
8# Unless required by applicable law or agreed to in writing, software
9# distributed under the License is distributed on an "AS IS" BASIS,
10# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11# See the License for the specific language governing permissions and
12# limitations under the License.
13
14import("//build/test.gni")
15
16module_output_path = "neural_network_runtime/"
17
18config("module_private_config") {
19  visibility = [ ":*" ]
20
21  include_dirs = [
22    "//third_party/googletest/googlemock/include",
23    "//foundation/ai/neural_network_runtime",
24    "//foundation/communication/ipc/interfaces/innerkits/ipc_core/include",
25    "//third_party/mindspore/mindspore/lite/mindir/include"
26  ]
27}
28
29ohos_unittest("CompilationTest") {
30  module_out_path = module_output_path
31
32  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/compilation/compilation_test.cpp" ]
33  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/compilation_mock_idevice.cpp" ]
34  configs = [ ":module_private_config" ]
35
36  deps = [
37    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
38    "//third_party/googletest:gtest_main",
39    "//third_party/googletest:gmock_main",
40  ]
41
42  external_deps = [
43    "c_utils:utils",
44    "hdf_core:libhdf_utils",
45    "hitrace_native:libhitracechain",
46    "hiviewdfx_hilog_native:libhilog",
47    "drivers_interface_nnrt:libnnrt_proxy_1.0",
48    "mindspore:mindir"
49  ]
50}
51
52ohos_unittest("ExecutorTest") {
53  module_out_path = module_output_path
54
55  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/executor/executor_test.cpp" ]
56  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/executor_mock_device.cpp" ]
57  configs = [ ":module_private_config" ]
58
59  deps = [
60    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
61    "//third_party/googletest:gtest_main",
62    "//third_party/googletest:gmock_main",
63  ]
64
65  external_deps = [
66    "c_utils:utils",
67    "hdf_core:libhdf_utils",
68    "hitrace_native:libhitracechain",
69    "hiviewdfx_hilog_native:libhilog",
70    "drivers_interface_nnrt:libnnrt_proxy_1.0",
71    "mindspore:mindir"
72  ]
73}
74
75ohos_unittest("DeviceManagerTest") {
76  module_out_path = module_output_path
77
78  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/device_manager/device_manager_test.cpp" ]
79  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/mock_idevice.cpp" ]
80  configs = [ ":module_private_config" ]
81
82  deps = [
83    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
84    "//third_party/googletest:gtest_main",
85    "//third_party/googletest:gmock_main",
86  ]
87
88  external_deps = [
89    "c_utils:utils",
90    "hitrace_native:libhitracechain",
91    "hiviewdfx_hilog_native:libhilog",
92    "drivers_interface_nnrt:libnnrt_proxy_1.0",
93    "mindspore:mindir"
94  ]
95}
96
97ohos_unittest("DeviceRegistrarTest") {
98  module_out_path = module_output_path
99
100  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/device_registrar/device_registrar_test.cpp" ]
101  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/mock_idevice.cpp" ]
102  configs = [ ":module_private_config" ]
103
104  deps = [
105    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
106    "//third_party/googletest:gtest_main",
107    "//third_party/googletest:gmock_main",
108  ]
109
110  external_deps = [
111    "c_utils:utils",
112    "hitrace_native:libhitracechain",
113    "hiviewdfx_hilog_native:libhilog",
114    "drivers_interface_nnrt:libnnrt_proxy_1.0",
115    "mindspore:mindir"
116  ]
117}
118
119ohos_unittest("HDIDeviceTest") {
120  module_out_path = module_output_path
121
122  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/hdi_device/hdi_device_test.cpp" ]
123  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/mock_idevice.cpp" ]
124  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/file_utils.cpp" ]
125  configs = [ ":module_private_config" ]
126
127  deps = [
128    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
129    "//third_party/googletest:gtest_main",
130    "//third_party/googletest:gmock_main",
131  ]
132
133  external_deps = [
134    "c_utils:utils",
135    "hitrace_native:libhitracechain",
136    "hiviewdfx_hilog_native:libhilog",
137    "drivers_interface_nnrt:libnnrt_proxy_1.0",
138    "mindspore:mindir"
139  ]
140}
141
142ohos_unittest("HDIPreparedModelTest") {
143  module_out_path = module_output_path
144
145  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/hdi_prepared_model/hdi_prepared_model_test.cpp" ]
146  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/mock_idevice.cpp" ]
147  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/file_utils.cpp" ]
148  configs = [ ":module_private_config" ]
149
150  deps = [
151    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
152    "//third_party/googletest:gtest_main",
153    "//third_party/googletest:gmock_main",
154  ]
155
156  external_deps = [
157    "c_utils:utils",
158    "hitrace_native:libhitracechain",
159    "hiviewdfx_hilog_native:libhilog",
160    "drivers_interface_nnrt:libnnrt_proxy_1.0",
161    "mindspore:mindir"
162  ]
163}
164
165ohos_unittest("MemoryManagerTest") {
166  module_out_path = module_output_path
167
168  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/memory_manager/memory_manager_test.cpp" ]
169  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/file_utils.cpp" ]
170  configs = [ ":module_private_config" ]
171
172  deps = [
173    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
174    "//third_party/googletest:gtest_main",
175    "//third_party/googletest:gmock_main",
176  ]
177
178  external_deps = [
179    "hitrace_native:libhitracechain",
180    "hiviewdfx_hilog_native:libhilog",
181    "drivers_interface_nnrt:libnnrt_proxy_1.0",
182    "mindspore:mindir"
183  ]
184}
185
186ohos_unittest("TransformTest") {
187  module_out_path = module_output_path
188
189  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/transform/transform_test.cpp" ]
190  configs = [ ":module_private_config" ]
191
192  deps = [
193    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
194    "//third_party/googletest:gtest_main",
195    "//third_party/googletest:gmock_main",
196  ]
197
198  external_deps = [
199    "hitrace_native:libhitracechain",
200    "hiviewdfx_hilog_native:libhilog",
201    "drivers_interface_nnrt:libnnrt_proxy_1.0",
202    "mindspore:mindir"
203  ]
204}
205
206ohos_unittest("InnerModelTest") {
207  module_out_path = module_output_path
208
209  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/inner_model/inner_model_test.cpp" ]
210  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/inner_model_mock_device.cpp" ]
211  configs = [ ":module_private_config" ]
212
213  deps = [
214    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
215    "//third_party/googletest:gtest_main",
216    "//third_party/googletest:gmock_main",
217  ]
218
219  external_deps = [
220    "c_utils:utils",
221    "hdf_core:libhdf_utils",
222    "hitrace_native:libhitracechain",
223    "hiviewdfx_hilog_native:libhilog",
224    "drivers_interface_nnrt:libnnrt_proxy_1.0",
225    "mindspore:mindir"
226  ]
227}
228
229ohos_unittest("NnTensorTest") {
230  module_out_path = module_output_path
231
232  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/inner_model/nn_tensor_test.cpp" ]
233  configs = [ ":module_private_config" ]
234
235  deps = [
236    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
237    "//third_party/googletest:gtest_main",
238    "//third_party/googletest:gmock_main",
239  ]
240
241  external_deps = [
242    "c_utils:utils",
243    "hdf_core:libhdf_utils",
244    "hitrace_native:libhitracechain",
245    "hiviewdfx_hilog_native:libhilog",
246    "drivers_interface_nnrt:libnnrt_proxy_1.0",
247    "mindspore:mindir"
248  ]
249}
250
251ohos_unittest("NnValidationTest") {
252  module_out_path = module_output_path
253
254  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/inner_model/nn_validation_test.cpp" ]
255  configs = [ ":module_private_config" ]
256
257  deps = [
258    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
259    "//third_party/googletest:gtest_main",
260    "//third_party/googletest:gmock_main",
261  ]
262
263  external_deps = [
264    "c_utils:utils",
265    "hdf_core:libhdf_utils",
266    "hitrace_native:libhitracechain",
267    "hiviewdfx_hilog_native:libhilog",
268    "drivers_interface_nnrt:libnnrt_proxy_1.0",
269    "mindspore:mindir"
270  ]
271}
272
273ohos_unittest("OpsRegistryTest") {
274  module_out_path = module_output_path
275
276  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/inner_model/ops_regitstry_test.cpp" ]
277  configs = [ ":module_private_config" ]
278
279  deps = [
280    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
281    "//third_party/googletest:gtest_main",
282    "//third_party/googletest:gmock_main",
283  ]
284
285  external_deps = [
286    "c_utils:utils",
287    "hdf_core:libhdf_utils",
288    "hitrace_native:libhitracechain",
289    "hiviewdfx_hilog_native:libhilog",
290    "drivers_interface_nnrt:libnnrt_proxy_1.0",
291    "mindspore:mindir"
292  ]
293}
294
295ohos_unittest("NeuralNetworkRuntimeTest") {
296  module_out_path = module_output_path
297
298  sources = [ "//foundation/ai/neural_network_runtime/test/unittest/components/neural_network_runtime_test/neural_network_runtime_test.cpp" ]
299  sources += [ "//foundation/ai/neural_network_runtime/test/unittest/common/executor_mock_device.cpp" ]
300
301  configs = [ ":module_private_config" ]
302
303  deps = [
304    "//foundation/ai/neural_network_runtime/frameworks:libneural_network_runtime",
305    "//third_party/googletest:gtest_main",
306    "//third_party/googletest:gmock_main",
307  ]
308
309  external_deps = [
310    "c_utils:utils",
311    "hdf_core:libhdf_utils",
312    "hitrace_native:libhitracechain",
313    "hiviewdfx_hilog_native:libhilog",
314    "drivers_interface_nnrt:libnnrt_proxy_1.0",
315    "mindspore:mindir"
316  ]
317}
318
319group("components_unittest") {
320  testonly = true
321  deps = [
322    ":CompilationTest",
323    ":ExecutorTest",
324    ":DeviceManagerTest",
325    ":DeviceRegistrarTest",
326    ":HDIDeviceTest",
327    ":HDIPreparedModelTest",
328    ":MemoryManagerTest",
329    ":TransformTest",
330    ":InnerModelTest",
331    ":NnTensorTest",
332    ":NnValidationTest",
333    ":OpsRegistryTest",
334    ":NeuralNetworkRuntimeTest",
335  ]
336}
337