You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

CMakeLists.txt 42 kB

5 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
4 years ago
5 years ago
4 years ago
4 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900
  1. # Copyright 2019-2020 Huawei Technologies Co., Ltd
  2. #
  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. project(ut_ge)
  16. set(CMAKE_CXX_STANDARD 11)
  17. set(PROTO_LIST
  18. "${GE_CODE_DIR}/metadef/proto/om.proto"
  19. "${GE_CODE_DIR}/metadef/proto/ge_ir.proto"
  20. "${GE_CODE_DIR}/metadef/proto/ge_api.proto"
  21. "${GE_CODE_DIR}/metadef/proto/insert_op.proto"
  22. "${GE_CODE_DIR}/metadef/proto/dump_task.proto"
  23. "${GE_CODE_DIR}/metadef/proto/fwk_adapter.proto"
  24. "${GE_CODE_DIR}/metadef/proto/op_mapping_info.proto"
  25. "${GE_CODE_DIR}/metadef/proto/optimizer_priority.proto"
  26. "${GE_CODE_DIR}/metadef/proto/ge_api.proto"
  27. "${GE_CODE_DIR}/metadef/proto/tensorflow/attr_value.proto"
  28. "${GE_CODE_DIR}/metadef/proto/tensorflow/tensor.proto"
  29. "${GE_CODE_DIR}/metadef/proto/tensorflow/resource_handle.proto"
  30. "${GE_CODE_DIR}/metadef/proto/tensorflow/tensor_shape.proto"
  31. "${GE_CODE_DIR}/metadef/proto/tensorflow/types.proto"
  32. "${GE_CODE_DIR}/metadef/proto/tensorflow/node_def.proto"
  33. "${GE_CODE_DIR}/metadef/proto/proto_inner/ge_onnx.proto"
  34. )
  35. protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})
  36. # include directories
  37. include_directories(${CMAKE_CURRENT_LIST_DIR})
  38. include_directories(${GE_CODE_DIR})
  39. include_directories(${GE_CODE_DIR}/inc)
  40. include_directories(${GE_CODE_DIR}/metadef/inc)
  41. include_directories(${GE_CODE_DIR}/ge)
  42. include_directories(${GE_CODE_DIR}/ge/inc)
  43. include_directories(${GE_CODE_DIR}/metadef)
  44. include_directories(${GE_CODE_DIR}/metadef/graph)
  45. include_directories(${GE_CODE_DIR}/inc/external)
  46. include_directories(${GE_CODE_DIR}/metadef/inc/external)
  47. include_directories(${GE_CODE_DIR}/parser/parser)
  48. include_directories(${GE_CODE_DIR}/metadef/inc/external/graph)
  49. include_directories(${GE_CODE_DIR}/metadef/inc/graph)
  50. include_directories(${GE_CODE_DIR}/inc/framework)
  51. include_directories(${GE_CODE_DIR}/metadef/inc/common)
  52. include_directories(${GE_CODE_DIR}/metadef/third_party)
  53. include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc)
  54. include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce)
  55. include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/ops)
  56. include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/toolchain)
  57. include_directories(${GE_CODE_DIR}/tests/ut/ge)
  58. include_directories(${CMAKE_BINARY_DIR})
  59. include_directories(${CMAKE_BINARY_DIR}/proto/ge)
  60. include_directories(${CMAKE_BINARY_DIR}/proto/ge/proto)
  61. set(COMMON_SRC_FILES
  62. "${GE_CODE_DIR}/ge/common/properties_manager.cc"
  63. "${GE_CODE_DIR}/ge/common/ge/plugin_manager.cc"
  64. "${GE_CODE_DIR}/ge/common/ge/tbe_plugin_manager.cc"
  65. "${GE_CODE_DIR}/metadef/graph/option/ge_local_context.cc"
  66. "${GE_CODE_DIR}/metadef/graph/option/ge_context.cc"
  67. "${GE_CODE_DIR}/ge/common/types.cc"
  68. "${GE_CODE_DIR}/parser/parser/common/op_map.cc"
  69. "${GE_CODE_DIR}/ge/common/fmk_error_codes.cc"
  70. "${GE_CODE_DIR}/ge/common/op/ge_op_utils.cc"
  71. "${GE_CODE_DIR}/ge/graph/manager/util/variable_accelerate_ctrl.cc"
  72. "${GE_CODE_DIR}/ge/opskernel_manager/ops_kernel_manager.cc"
  73. "${GE_CODE_DIR}/ge/generator/ge_generator.cc"
  74. "${GE_CODE_DIR}/ge/generator/generator_api.cc"
  75. "${GE_CODE_DIR}/ge/graph/common/omg_util.cc"
  76. "${GE_CODE_DIR}/ge/graph/common/bcast.cc"
  77. "${GE_CODE_DIR}/ge/common/util.cc"
  78. "${GE_CODE_DIR}/ge/common/ge/op_tiling_manager.cc"
  79. "${GE_CODE_DIR}/ge/init/gelib.cc"
  80. "${GE_CODE_DIR}/metadef/graph/ge_attr_define.cc"
  81. "${GE_CODE_DIR}/metadef/graph/anchor.cc"
  82. "${GE_CODE_DIR}/metadef/graph/ge_attr_value.cc"
  83. "${GE_CODE_DIR}/metadef/graph/attr_value.cc"
  84. "${GE_CODE_DIR}/metadef/graph/buffer.cc"
  85. "${GE_CODE_DIR}/metadef/graph/compute_graph.cc"
  86. "${GE_CODE_DIR}/metadef/graph/graph.cc"
  87. "${GE_CODE_DIR}/metadef/graph/gnode.cc"
  88. "${GE_CODE_DIR}/metadef/graph/ascend_string.cc"
  89. "${GE_CODE_DIR}/metadef/graph/inference_context.cc"
  90. "${GE_CODE_DIR}/metadef/graph/shape_refiner.cc"
  91. "${GE_CODE_DIR}/metadef/graph/model.cc"
  92. "${GE_CODE_DIR}/metadef/graph/model_serialize.cc"
  93. "${GE_CODE_DIR}/metadef/graph/node.cc"
  94. "${GE_CODE_DIR}/metadef/graph/runtime_inference_context.cc"
  95. "${GE_CODE_DIR}/metadef/graph/op_desc.cc"
  96. "${GE_CODE_DIR}/metadef/third_party/transformer/src/transfer_shape_according_to_format.cpp"
  97. "${GE_CODE_DIR}/metadef/third_party/transformer/src/axis_util.cpp"
  98. "${GE_CODE_DIR}/metadef/graph/operator.cc"
  99. "${GE_CODE_DIR}/metadef/graph/operator_factory.cc"
  100. "${GE_CODE_DIR}/metadef/graph/operator_factory_impl.cc"
  101. "${GE_CODE_DIR}/metadef/graph/ge_tensor.cc"
  102. "${GE_CODE_DIR}/metadef/graph/ref_relation.cc"
  103. "${GE_CODE_DIR}/metadef/graph/tensor.cc"
  104. "${GE_CODE_DIR}/metadef/graph/detail/attributes_holder.cc"
  105. "${GE_CODE_DIR}/metadef/graph/utils/anchor_utils.cc"
  106. "${GE_CODE_DIR}/metadef/graph/utils/graph_utils.cc"
  107. "${GE_CODE_DIR}/metadef/graph/utils/ge_ir_utils.cc"
  108. "${GE_CODE_DIR}/metadef/graph/utils/node_utils.cc"
  109. "${GE_CODE_DIR}/metadef/graph/utils/op_desc_utils.cc"
  110. "${GE_CODE_DIR}/metadef/graph/utils/tensor_utils.cc"
  111. "${GE_CODE_DIR}/metadef/graph/utils/type_utils.cc"
  112. "${GE_CODE_DIR}/metadef/graph/utils/transformer_utils.cc"
  113. "${GE_CODE_DIR}/metadef/graph/debug/graph_debug.cc"
  114. "${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc"
  115. "${GE_CODE_DIR}/metadef/ops/op_imp.cpp"
  116. "${GE_CODE_DIR}/metadef/register/register.cpp"
  117. "${GE_CODE_DIR}/metadef/register/op_kernel_registry.cpp"
  118. "${GE_CODE_DIR}/metadef/register/auto_mapping_util.cpp"
  119. "${GE_CODE_DIR}/metadef/register/tensor_assign.cpp"
  120. "${GE_CODE_DIR}/metadef/register/register_format_transfer.cc"
  121. "${GE_CODE_DIR}/metadef/graph/format_refiner.cc"
  122. "${GE_CODE_DIR}/ge/engine_manager/dnnengine_manager.cc"
  123. "${GE_CODE_DIR}/ge/opskernel_manager/ops_kernel_manager.cc"
  124. "${GE_CODE_DIR}/ge/session/session_manager.cc"
  125. "${GE_CODE_DIR}/ge/opskernel_manager/ops_kernel_builder_manager.cc"
  126. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/model_manager.cc"
  127. "${GE_CODE_DIR}/ge/common/profiling/profiling_manager.cc"
  128. "${GE_CODE_DIR}/ge/graph/manager/host_mem_manager.cc"
  129. "${GE_CODE_DIR}/ge/session/inner_session.cc"
  130. "${GE_CODE_DIR}/ge/graph/manager/util/rt_context_util.cc"
  131. "${GE_CODE_DIR}/ge/graph/execute/graph_execute.cc"
  132. "${GE_CODE_DIR}/ge/graph/preprocess/graph_preprocess.cc"
  133. "${GE_CODE_DIR}/ge/hybrid/hybrid_davinci_model_stub.cc"
  134. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/davinci_model.cc"
  135. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/data_inputer.cc"
  136. "${GE_CODE_DIR}/ge/common/dump/dump_properties.cc"
  137. "${GE_CODE_DIR}/ge/common/helper/model_helper.cc"
  138. "${GE_CODE_DIR}/ge/common/dump/dump_manager.cc"
  139. "${GE_CODE_DIR}/ge/common/helper/om_file_helper.cc"
  140. "${GE_CODE_DIR}/ge/model/ge_root_model.cc"
  141. "${GE_CODE_DIR}/ge/common/model_parser/base.cc"
  142. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/data_dumper.cc"
  143. "${GE_CODE_DIR}/ge/graph/manager/graph_manager.cc"
  144. "${GE_CODE_DIR}/ge/common/dump/dump_server.cc"
  145. "${GE_CODE_DIR}/ge/graph/preprocess/insert_op/util_insert_aipp_op.cc"
  146. "${GE_CODE_DIR}/ge/graph/preprocess/multi_batch_copy_graph.cc"
  147. "${GE_CODE_DIR}/ge/graph/optimize/mem_rw_conflict_optimize.cc"
  148. "${GE_CODE_DIR}/ge/graph/passes/pass_manager.cc"
  149. "${GE_CODE_DIR}/ge/graph/passes/resource_pair_add_control_pass.cc"
  150. "${GE_CODE_DIR}/ge/graph/passes/resource_pair_remove_control_pass.cc"
  151. "${GE_CODE_DIR}/ge/graph/passes/pass_utils.cc"
  152. "${GE_CODE_DIR}/ge/graph/passes/base_pass.cc"
  153. "${GE_CODE_DIR}/ge/graph/passes/bitcast_pass.cc"
  154. "${GE_CODE_DIR}/ge/graph/passes/constant_folding_pass.cc"
  155. "${GE_CODE_DIR}/ge/graph/passes/aicpu_constant_folding_pass.cc"
  156. "${GE_CODE_DIR}/ge/graph/passes/reshape_remove_pass.cc"
  157. "${GE_CODE_DIR}/ge/graph/passes/reshape_recovery_pass.cc"
  158. "${GE_CODE_DIR}/ge/graph/passes/transop_breadth_fusion_pass.cc"
  159. "${GE_CODE_DIR}/ge/graph/passes/transop_depth_fusion_pass.cc"
  160. "${GE_CODE_DIR}/ge/graph/passes/transop_nearby_allreduce_fusion_pass.cc"
  161. "${GE_CODE_DIR}/ge/graph/passes/same_transdata_breadth_fusion_pass.cc"
  162. "${GE_CODE_DIR}/ge/graph/passes/transop_without_reshape_fusion_pass.cc"
  163. "${GE_CODE_DIR}/ge/graph/passes/compile_nodes_pass.cc"
  164. "${GE_CODE_DIR}/ge/graph/passes/variable_prepare_op_pass.cc"
  165. "${GE_CODE_DIR}/ge/graph/passes/variable_ref_delete_op_pass.cc"
  166. "${GE_CODE_DIR}/ge/graph/passes/variable_ref_useless_control_out_delete_pass.cc"
  167. "${GE_CODE_DIR}/ge/graph/passes/subgraph_pass.cc"
  168. "${GE_CODE_DIR}/ge/graph/passes/data_pass.cc"
  169. "${GE_CODE_DIR}/ge/graph/passes/net_output_pass.cc"
  170. "${GE_CODE_DIR}/ge/graph/passes/replace_transshape_pass.cc"
  171. "${GE_CODE_DIR}/ge/graph/passes/constant_fuse_same_pass.cc"
  172. "${GE_CODE_DIR}/ge/graph/passes/print_op_pass.cc"
  173. "${GE_CODE_DIR}/ge/graph/passes/no_use_reshape_remove_pass.cc"
  174. "${GE_CODE_DIR}/ge/graph/passes/iterator_op_pass.cc"
  175. "${GE_CODE_DIR}/ge/graph/passes/input_output_connection_identify_pass.cc"
  176. "${GE_CODE_DIR}/ge/graph/passes/atomic_addr_clean_pass.cc"
  177. "${GE_CODE_DIR}/ge/graph/passes/mark_same_addr_pass.cc"
  178. "${GE_CODE_DIR}/ge/graph/passes/mark_graph_unknown_status_pass.cc"
  179. "${GE_CODE_DIR}/ge/graph/passes/mark_agnostic_pass.cc"
  180. "${GE_CODE_DIR}/ge/graph/passes/dimension_compute_pass.cc"
  181. "${GE_CODE_DIR}/ge/graph/passes/dimension_adjust_pass.cc"
  182. "${GE_CODE_DIR}/ge/graph/passes/get_original_format_pass.cc"
  183. "${GE_CODE_DIR}/ge/graph/passes/shape_operate_op_remove_pass.cc"
  184. "${GE_CODE_DIR}/ge/graph/passes/unused_op_remove_pass.cc"
  185. "${GE_CODE_DIR}/ge/graph/passes/assert_pass.cc"
  186. "${GE_CODE_DIR}/ge/graph/passes/dropout_pass.cc"
  187. "${GE_CODE_DIR}/ge/graph/passes/infershape_pass.cc"
  188. "${GE_CODE_DIR}/ge/graph/passes/unused_const_pass.cc"
  189. "${GE_CODE_DIR}/ge/graph/passes/isolated_op_remove_pass.cc"
  190. "${GE_CODE_DIR}/ge/graph/passes/permute_pass.cc"
  191. "${GE_CODE_DIR}/ge/graph/passes/ctrl_edge_transfer_pass.cc"
  192. "${GE_CODE_DIR}/ge/graph/passes/end_of_sequence_add_control_pass.cc"
  193. "${GE_CODE_DIR}/ge/graph/passes/stop_gradient_pass.cc"
  194. "${GE_CODE_DIR}/ge/graph/passes/prevent_gradient_pass.cc"
  195. "${GE_CODE_DIR}/ge/graph/passes/identity_pass.cc"
  196. "${GE_CODE_DIR}/ge/graph/passes/ref_identity_delete_op_pass.cc"
  197. "${GE_CODE_DIR}/ge/graph/passes/placeholder_with_default_pass.cc"
  198. "${GE_CODE_DIR}/ge/graph/passes/snapshot_pass.cc"
  199. "${GE_CODE_DIR}/ge/graph/passes/guarantee_const_pass.cc"
  200. "${GE_CODE_DIR}/ge/graph/passes/var_is_initialized_op_pass.cc"
  201. "${GE_CODE_DIR}/ge/graph/passes/parallel_concat_start_op_pass.cc"
  202. "${GE_CODE_DIR}/ge/graph/passes/folding_pass.cc"
  203. "${GE_CODE_DIR}/ge/graph/passes/cast_translate_pass.cc"
  204. "${GE_CODE_DIR}/ge/graph/passes/prune_pass.cc"
  205. "${GE_CODE_DIR}/ge/graph/passes/merge_to_stream_merge_pass.cc"
  206. "${GE_CODE_DIR}/ge/graph/passes/merge_input_memcpy_pass.cc"
  207. "${GE_CODE_DIR}/ge/graph/passes/switch_to_stream_switch_pass.cc"
  208. "${GE_CODE_DIR}/ge/graph/passes/attach_stream_label_pass.cc"
  209. "${GE_CODE_DIR}/ge/graph/passes/multi_batch_pass.cc"
  210. "${GE_CODE_DIR}/ge/graph/passes/multi_batch_clone_pass.cc"
  211. "${GE_CODE_DIR}/ge/graph/passes/subexpression_migration_pass.cc"
  212. "${GE_CODE_DIR}/ge/graph/passes/subgraph_const_migration_pass.cc"
  213. "${GE_CODE_DIR}/ge/graph/passes/unused_args_clean_pass.cc"
  214. "${GE_CODE_DIR}/ge/graph/passes/next_iteration_pass.cc"
  215. "${GE_CODE_DIR}/ge/graph/passes/control_trigger_pass.cc"
  216. "${GE_CODE_DIR}/ge/graph/passes/cond_pass.cc"
  217. "${GE_CODE_DIR}/ge/graph/passes/cond_remove_pass.cc"
  218. "${GE_CODE_DIR}/ge/graph/passes/for_pass.cc"
  219. "${GE_CODE_DIR}/ge/graph/passes/enter_pass.cc"
  220. "${GE_CODE_DIR}/ge/graph/passes/assign_pass.cc"
  221. "${GE_CODE_DIR}/ge/graph/passes/addn_pass.cc"
  222. "${GE_CODE_DIR}/ge/graph/passes/common_subexpression_elimination_pass.cc"
  223. "${GE_CODE_DIR}/ge/graph/passes/transop_symmetry_elimination_pass.cc"
  224. "${GE_CODE_DIR}/ge/graph/passes/save_pass.cc"
  225. "${GE_CODE_DIR}/ge/graph/passes/switch_dead_branch_elimination.cc"
  226. "${GE_CODE_DIR}/ge/graph/passes/switch_logic_remove_pass.cc"
  227. "${GE_CODE_DIR}/ge/graph/passes/switch_data_edges_bypass.cc"
  228. "${GE_CODE_DIR}/ge/graph/passes/merge_pass.cc"
  229. "${GE_CODE_DIR}/ge/graph/passes/variable_format_pass.cc"
  230. "${GE_CODE_DIR}/ge/graph/passes/variable_op_pass.cc"
  231. "${GE_CODE_DIR}/ge/graph/passes/cast_remove_pass.cc"
  232. "${GE_CODE_DIR}/ge/graph/passes/transpose_transdata_pass.cc"
  233. "${GE_CODE_DIR}/ge/graph/passes/hccl_memcpy_pass.cc"
  234. "${GE_CODE_DIR}/ge/graph/passes/flow_ctrl_pass.cc"
  235. "${GE_CODE_DIR}/ge/graph/passes/global_step_insert_pass.cc"
  236. "${GE_CODE_DIR}/ge/graph/passes/link_gen_mask_nodes_pass.cc"
  237. "${GE_CODE_DIR}/ge/graph/passes/replace_with_empty_const_pass.cc"
  238. "${GE_CODE_DIR}/ge/graph/passes/hccl_group_pass.cc"
  239. "${GE_CODE_DIR}/ge/graph/passes/memcpy_addr_async_pass.cc"
  240. "${GE_CODE_DIR}/ge/graph/passes/set_input_output_offset_pass.cc"
  241. "${GE_CODE_DIR}/ge/model/ge_model.cc"
  242. "${GE_CODE_DIR}/ge/common/cust_aicpu_kernel_store.cc"
  243. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/model_utils.cc"
  244. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/zero_copy_offset.cc"
  245. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/zero_copy_task.cc"
  246. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/cpu_queue_schedule.cc"
  247. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/aipp_utils.cc"
  248. "${GE_CODE_DIR}/ge/omm/csa_interact.cc"
  249. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/tbe_handle_store.cc"
  250. "${GE_CODE_DIR}/ge/common/kernel_store.cc"
  251. "${GE_CODE_DIR}/ge/common/tbe_kernel_store.cc"
  252. "${GE_CODE_DIR}/ge/common/auth/file_saver.cc"
  253. "${GE_CODE_DIR}/ge/graph/manager/util/debug.cc"
  254. "${GE_CODE_DIR}/ge/common/debug/memory_dumper.cc"
  255. "${GE_CODE_DIR}/ge/graph/manager/graph_context.cc"
  256. "${GE_CODE_DIR}/ge/graph/load/graph_loader.cc"
  257. "${GE_CODE_DIR}/ge/graph/optimize/graph_optimize.cc"
  258. "${GE_CODE_DIR}/ge/graph/build/graph_builder.cc"
  259. "${GE_CODE_DIR}/ge/graph/partition/graph_partition.cc"
  260. "${GE_CODE_DIR}/ge/common/helper/model_cache_helper.cc"
  261. "${GE_CODE_DIR}/ge/ir_build/ge_ir_build.cc"
  262. "${GE_CODE_DIR}/ge/graph/build/label_allocator.cc"
  263. "${GE_CODE_DIR}/ge/graph/passes/memcpy_addr_async_pass.cc"
  264. "${GE_CODE_DIR}/ge/graph/partition/stage_partition.cc"
  265. "${GE_CODE_DIR}/ge/graph/partition/dynamic_shape_partition.cc"
  266. "${GE_CODE_DIR}/ge/graph/optimize/summary_optimize.cc"
  267. "${GE_CODE_DIR}/ge/ir_build/atc_ir_common.cc"
  268. "${GE_CODE_DIR}/ge/graph/preprocess/insert_op/ge_aipp_op.cc"
  269. "${GE_CODE_DIR}/ge/graph/preprocess/multi_batch_options.cc"
  270. "${GE_CODE_DIR}/ge/graph/build/model_builder.cc"
  271. "${GE_CODE_DIR}/ge/graph/build/run_context.cc"
  272. "${GE_CODE_DIR}/ge/graph/build/stream_graph_optimizer.cc"
  273. "${GE_CODE_DIR}/ge/graph/build/task_generator.cc"
  274. "${GE_CODE_DIR}/ge/graph/partition/graph_partition.cc"
  275. "${GE_CODE_DIR}/ge/graph/partition/engine_place.cc"
  276. "${GE_CODE_DIR}/ge/graph/build/stream_allocator.cc"
  277. "${GE_CODE_DIR}/ge/graph/build/memory/memory_assigner.cc"
  278. "${GE_CODE_DIR}/ge/graph/build/logical_stream_allocator.cc"
  279. "${GE_CODE_DIR}/ge/graph/build/memory/graph_mem_assigner.cc"
  280. "${GE_CODE_DIR}/ge/graph/build/memory/var_mem_assign_util.cc"
  281. "${GE_CODE_DIR}/ge/graph/build/memory/hybrid_mem_assigner.cc"
  282. "${GE_CODE_DIR}/ge/graph/build/memory/block_mem_assigner.cc"
  283. "${GE_CODE_DIR}/ge/graph/build/memory/binary_block_mem_assigner.cc"
  284. "${GE_CODE_DIR}/ge/graph/build/memory/max_block_mem_assigner.cc"
  285. "${GE_CODE_DIR}/ge/graph/manager/graph_mem_allocator.cc"
  286. "${GE_CODE_DIR}/ge/graph/manager/graph_var_manager.cc"
  287. "${GE_CODE_DIR}/ge/analyzer/analyzer.cc"
  288. "${GE_CODE_DIR}/ge/common/thread_pool.cc"
  289. "${GE_CODE_DIR}/ge/graph/common/transop_util.cc"
  290. "${GE_CODE_DIR}/ge/graph/manager/graph_manager_utils.cc"
  291. "${GE_CODE_DIR}/ge/graph/manager/trans_var_data_utils.cc"
  292. "${GE_CODE_DIR}/ge/graph/common/local_context.cc"
  293. "${GE_CODE_DIR}/ge/graph/manager/graph_caching_allocator.cc"
  294. "${GE_CODE_DIR}/ge/graph/manager/rdma_pool_allocator.cc"
  295. "${GE_CODE_DIR}/ge/common/dump/dump_op.cc"
  296. "${GE_CODE_DIR}/ge/hybrid/node_executor/aicpu/aicpu_ext_info.cc"
  297. "${GE_CODE_DIR}/ge/common/ge/datatype_util.cc"
  298. "${GE_CODE_DIR}/metadef/register/ops_kernel_builder_registry.cc"
  299. "${GE_CODE_DIR}/metadef/register/op_tiling.cpp"
  300. "${GE_CODE_DIR}/metadef/graph/utils/tuning_utils.cc"
  301. "${GE_CODE_DIR}/metadef/register/op_tiling_registry.cpp"
  302. "${GE_CODE_DIR}/ge/ge_local_engine/engine/host_cpu_engine.cc"
  303. )
  304. set(COMMON_FORMAT_SRC_FILES
  305. "${GE_CODE_DIR}/ge/common/fp16_t.cc"
  306. "${GE_CODE_DIR}/ge/common/ge_format_util.cc"
  307. "${GE_CODE_DIR}/ge/common/formats/formats.cc"
  308. "${GE_CODE_DIR}/ge/common/formats/format_transfers/datatype_transfer.cc"
  309. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_transpose.cc"
  310. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_nchw_nc1hwc0.cc"
  311. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fractal_z.cc"
  312. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fractal_nz.cc"
  313. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fractal_zz.cc"
  314. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.cc"
  315. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_nc1hwc0_nchw.cc"
  316. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_nc1hwc0_nhwc.cc"
  317. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_hwcn_c1hwncoc0.cc"
  318. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_c1hwncoc0_hwcn.cc"
  319. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_nchw.cc"
  320. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_nhwc.cc"
  321. "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_hwcn.cc"
  322. "${GE_CODE_DIR}/ge/common/formats/utils/formats_trans_utils.cc"
  323. )
  324. set(GRAPH_OPTIMIZE_COMMON_SRC_FILES
  325. "${GE_CODE_DIR}/ge/graph/optimize/graph_optimize.cc"
  326. "${GE_CODE_DIR}/ge/graph/optimize/summary_optimize.cc"
  327. )
  328. set(GRAPH_PREPARE_COMMON_SRC_FILES
  329. "${GE_CODE_DIR}/ge/graph/preprocess/graph_preprocess.cc"
  330. "${GE_CODE_DIR}/ge/graph/preprocess/insert_op/util_insert_aipp_op.cc"
  331. "${GE_CODE_DIR}/ge/graph/preprocess/insert_op/ge_aipp_op.cc"
  332. #"${GE_CODE_DIR}/ge/graph/preprocess/insert_op/base_insert_op.cc"
  333. )
  334. set(GRAPH_PARTITION_COMMON_SRC_FILES
  335. "${GE_CODE_DIR}/ge/graph/partition/graph_partition.cc"
  336. "${GE_CODE_DIR}/ge/plugin/engine/dnnengines.cc"
  337. "${GE_CODE_DIR}/ge/graph/partition/engine_place.cc"
  338. )
  339. set(GRAPH_LOAD_COMMON_SRC_FILES
  340. "${GE_CODE_DIR}/ge/graph/load/graph_loader.cc"
  341. "${GE_CODE_DIR}/ge/graph/manager/graph_manager_utils.cc"
  342. "${GE_CODE_DIR}/ge/omm/csa_interact.cc"
  343. "${GE_CODE_DIR}/ge/graph/manager/graph_mem_allocator.cc"
  344. "${GE_CODE_DIR}/ge/graph/manager/graph_var_manager.cc"
  345. "${GE_CODE_DIR}/ge/graph/manager/trans_var_data_utils.cc"
  346. "${GE_CODE_DIR}/ge/graph/manager/graph_caching_allocator.cc"
  347. "${GE_CODE_DIR}/ge/graph/manager/rdma_pool_allocator.cc"
  348. "${GE_CODE_DIR}/ge/common/thread_pool.cc"
  349. )
  350. set(DISTINCT_GRAPH_LOAD_SRC_FILES
  351. "${GE_CODE_DIR}/ge/graph/manager/util/hcom_util.cc"
  352. "${GE_CODE_DIR}/ge/graph/manager/util/debug.cc"
  353. "${GE_CODE_DIR}/ge/common/properties_manager.cc"
  354. "${GE_CODE_DIR}/ge/common/profiling/profiling_manager.cc"
  355. "${GE_CODE_DIR}/ge/common/model_parser/base.cc"
  356. "${GE_CODE_DIR}/ge/common/tbe_kernel_store.cc"
  357. "${GE_CODE_DIR}/ge/common/util.cc"
  358. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/cpu_queue_schedule.cc"
  359. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/data_dumper.cc"
  360. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/data_inputer.cc"
  361. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/davinci_model.cc"
  362. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/davinci_model_parser.cc"
  363. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/model_manager.cc"
  364. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/model_utils.cc"
  365. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/tbe_handle_store.cc"
  366. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/task_info.cc"
  367. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/event_record_task_info.cc"
  368. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/event_wait_task_info.cc"
  369. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/fusion_start_task_info.cc"
  370. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/fusion_stop_task_info.cc"
  371. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/hccl_task_info.cc"
  372. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc"
  373. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/kernel_task_info.cc"
  374. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/label_set_task_info.cc"
  375. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc"
  376. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/profiler_trace_task_info.cc"
  377. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/stream_active_task_info.cc"
  378. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/stream_switch_task_info.cc"
  379. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/end_graph_task_info.cc"
  380. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/model_exit_task_info.cc"
  381. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/super_kernel/super_kernel.cc"
  382. "${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/super_kernel/super_kernel_factory.cc"
  383. "${GE_CODE_DIR}/ge/model/ge_model.cc"
  384. "${GE_CODE_DIR}/ge/common/helper/model_helper.cc"
  385. "${GE_CODE_DIR}/ge/common/helper/om_file_helper.cc"
  386. "${GE_CODE_DIR}/ge/common/debug/memory_dumper.cc"
  387. "${GE_CODE_DIR}/ge/executor/ge_executor.cc"
  388. "${GE_CODE_DIR}/ge/common/auth/file_saver.cc"
  389. "${GE_CODE_DIR}/ge/graph/manager/model_manager/event_manager.cc"
  390. )
  391. set(GRAPH_EXECUTE_COMMON_SRC_FILES
  392. "${GE_CODE_DIR}/ge/graph/execute/graph_execute.cc"
  393. "${GE_CODE_DIR}/ge/graph/manager/graph_manager.cc"
  394. "${GE_CODE_DIR}/ge/graph/manager/graph_context.cc"
  395. "${GE_CODE_DIR}/ge/graph/manager/util/rt_context_util.cc"
  396. "${GE_CODE_DIR}/ge/graph/manager/graph_context.h"
  397. )
  398. set(GRAPH_BUILD_COMMON_SRC_FILES
  399. "${GE_CODE_DIR}/ge/graph/build/graph_builder.cc"
  400. "${GE_CODE_DIR}/ge/graph/build/task_generator.cc"
  401. "${GE_CODE_DIR}/ge/client/ge_api.cc"
  402. "${GE_CODE_DIR}/ge/session/inner_session.cc"
  403. "${GE_CODE_DIR}/ge/session/session_manager.cc"
  404. "${GE_CODE_DIR}/ge/engine_manager/dnnengine_manager.cc"
  405. "${GE_CODE_DIR}/ge/plugin/engine/engine_manage.cc"
  406. "${GE_CODE_DIR}/ge/graph/build/logical_stream_allocator.cc"
  407. "${GE_CODE_DIR}/ge/graph/build/stream_allocator.cc"
  408. "${GE_CODE_DIR}/ge/graph/build/memory/block_mem_assigner.cc"
  409. "${GE_CODE_DIR}/ge/graph/build/memory/binary_block_mem_assigner.cc"
  410. "${GE_CODE_DIR}/ge/graph/build/memory/hybrid_mem_assigner.cc"
  411. "${GE_CODE_DIR}/ge/graph/build/memory/max_block_mem_assigner.cc"
  412. "${GE_CODE_DIR}/ge/model/ge_model.cc"
  413. "${GE_CODE_DIR}/ge/common/helper/model_helper.cc"
  414. "${GE_CODE_DIR}/ge/common/helper/om_file_helper.cc"
  415. "${GE_CODE_DIR}/ge/common/tbe_kernel_store.cc"
  416. "${GE_CODE_DIR}/ge/common/thread_pool.cc"
  417. "${GE_CODE_DIR}/ge/common/model_parser/base.cc"
  418. "${GE_CODE_DIR}/ge/graph/build/run_context.cc"
  419. "${GE_CODE_DIR}/ge/graph/common/local_context.cc"
  420. )
  421. set(GRAPH_PASS_COMMON_SRC_FILES
  422. "${GE_CODE_DIR}/ge/graph/passes/pass_manager.cc"
  423. "${GE_CODE_DIR}/ge/graph/passes/base_pass.cc"
  424. "${GE_CODE_DIR}/ge/graph/passes/variable_prepare_op_pass.cc"
  425. "${GE_CODE_DIR}/ge/graph/passes/variable_ref_delete_op_pass.cc"
  426. "${GE_CODE_DIR}/ge/graph/passes/atomic_addr_clean_pass.cc"
  427. "${GE_CODE_DIR}/ge/graph/passes/constant_folding_pass.cc"
  428. "${GE_CODE_DIR}/parser/parser/tensorflow/iterator_fusion_pass.cc"
  429. "${GE_CODE_DIR}/ge/graph/passes/iterator_op_pass.cc"
  430. "${GE_CODE_DIR}/ge/graph/passes/net_output_pass.cc"
  431. "${GE_CODE_DIR}/ge/graph/passes/print_op_pass.cc"
  432. "${GE_CODE_DIR}/ge/graph/passes/dimension_compute_pass.cc"
  433. "${GE_CODE_DIR}/ge/graph/passes/dimension_adjust_pass.cc"
  434. "${GE_CODE_DIR}/ge/graph/passes/get_original_format_pass.cc"
  435. "${GE_CODE_DIR}/ge/graph/passes/stop_gradient_pass.cc"
  436. "${GE_CODE_DIR}/ge/graph/passes/guarantee_const_pass.cc"
  437. "${GE_CODE_DIR}/ge/graph/passes/prevent_gradient_pass.cc"
  438. "${GE_CODE_DIR}/ge/graph/passes/identity_pass.cc"
  439. "${GE_CODE_DIR}/ge/graph/passes/placeholder_with_default_pass.cc"
  440. "${GE_CODE_DIR}/ge/graph/passes/snapshot_pass.cc"
  441. "${GE_CODE_DIR}/ge/graph/passes/shape_operate_op_remove_pass.cc"
  442. "${GE_CODE_DIR}/ge/graph/passes/unused_op_remove_pass.cc"
  443. "${GE_CODE_DIR}/ge/graph/passes/isolated_op_remove_pass.cc"
  444. "${GE_CODE_DIR}/ge/graph/passes/permute_pass.cc"
  445. "${GE_CODE_DIR}/ge/graph/passes/var_is_initialized_op_pass.cc"
  446. "${GE_CODE_DIR}/ge/graph/passes/cast_translate_pass.cc"
  447. "${GE_CODE_DIR}/ge/graph/passes/prune_pass.cc"
  448. "${GE_CODE_DIR}/ge/graph/passes/enter_pass.cc"
  449. "${GE_CODE_DIR}/ge/graph/passes/next_iteration_pass.cc"
  450. "${GE_CODE_DIR}/ge/graph/passes/pass_utils.cc"
  451. "${GE_CODE_DIR}/ge/graph/passes/addn_pass.cc"
  452. "${GE_CODE_DIR}/ge/graph/passes/save_pass.cc"
  453. "${GE_CODE_DIR}/ge/graph/passes/merge_pass.cc"
  454. "${GE_CODE_DIR}/ge/graph/passes/switch_logic_remove_pass.cc"
  455. "${GE_CODE_DIR}/ge/graph/passes/assert_pass.cc"
  456. "${GE_CODE_DIR}/ge/graph/passes/dropout_pass.cc"
  457. "${GE_CODE_DIR}/ge/graph/passes/unused_const_pass.cc"
  458. "${GE_CODE_DIR}/ge/graph/passes/reshape_remove_pass.cc"
  459. "${GE_CODE_DIR}/ge/graph/passes/resource_pair_add_control_pass.cc"
  460. "${GE_CODE_DIR}/ge/graph/passes/resource_pair_remove_control_pass.cc"
  461. "${GE_CODE_DIR}/ge/graph/passes/transop_breadth_fusion_pass.cc"
  462. "${GE_CODE_DIR}/ge/graph/passes/transop_without_reshape_fusion_pass.cc"
  463. "${GE_CODE_DIR}/ge/graph/passes/transop_depth_fusion_pass.cc"
  464. "${GE_CODE_DIR}/ge/graph/passes/transop_nearby_allreduce_fusion_pass.cc"
  465. "${GE_CODE_DIR}/ge/graph/passes/same_transdata_breadth_fusion_pass.cc"
  466. "${GE_CODE_DIR}/ge/graph/passes/compile_nodes_pass.cc"
  467. "${GE_CODE_DIR}/ge/graph/common/transop_util.cc"
  468. "${GE_CODE_DIR}/ge/graph/passes/flow_ctrl_pass.cc"
  469. "${GE_CODE_DIR}/ge/graph/optimize/optimizer/allreduce_fusion_pass.cc"
  470. "${GE_CODE_DIR}/ge/graph/passes/folding_pass.cc"
  471. "${GE_CODE_DIR}/ge/graph/passes/variable_op_pass.cc"
  472. "${GE_CODE_DIR}/ge/graph/passes/transpose_transdata_pass.cc"
  473. "${GE_CODE_DIR}/ge/graph/passes/hccl_memcpy_pass.cc"
  474. "${GE_CODE_DIR}/ge/graph/passes/no_use_reshape_remove_pass.cc"
  475. "${GE_CODE_DIR}/ge/graph/passes/infershape_pass.cc"
  476. "${GE_CODE_DIR}/ge/ge_local_engine/engine/host_cpu_engine.cc"
  477. "${GE_CODE_DIR}/ge/analyzer/analyzer.cc"
  478. "${GE_CODE_DIR}/ge/graph/passes/net_output_pass.cc"
  479. "${GE_CODE_DIR}/ge/graph/common/local_context.cc"
  480. )
  481. set(KERNEL_SRC_FILES
  482. "${GE_CODE_DIR}/ge/host_kernels/broadcast_gradient_args_kernel.cc"
  483. "${GE_CODE_DIR}/ge/host_kernels/broadcast_args_kernel.cc"
  484. "${GE_CODE_DIR}/ge/host_kernels/greater_kernel.cc"
  485. "${GE_CODE_DIR}/ge/host_kernels/gather_v2_kernel.cc"
  486. "${GE_CODE_DIR}/ge/host_kernels/maximum_kernel.cc"
  487. "${GE_CODE_DIR}/ge/host_kernels/floormod_kernel.cc"
  488. "${GE_CODE_DIR}/ge/host_kernels/floordiv_kernel.cc"
  489. "${GE_CODE_DIR}/ge/host_kernels/range_kernel.cc"
  490. "${GE_CODE_DIR}/ge/host_kernels/mul_kernel.cc"
  491. "${GE_CODE_DIR}/ge/host_kernels/shape_kernel.cc"
  492. "${GE_CODE_DIR}/ge/host_kernels/shape_n_kernel.cc"
  493. "${GE_CODE_DIR}/ge/host_kernels/size_kernel.cc"
  494. "${GE_CODE_DIR}/ge/host_kernels/rank_kernel.cc"
  495. "${GE_CODE_DIR}/ge/host_kernels/fill_kernel.cc"
  496. "${GE_CODE_DIR}/ge/host_kernels/empty_kernel.cc"
  497. "${GE_CODE_DIR}/ge/host_kernels/cast_kernel.cc"
  498. "${GE_CODE_DIR}/ge/host_kernels/transdata_kernel.cc"
  499. "${GE_CODE_DIR}/ge/host_kernels/permute_kernel.cc"
  500. "${GE_CODE_DIR}/ge/host_kernels/squeeze_kernel.cc"
  501. "${GE_CODE_DIR}/ge/host_kernels/reshape_kernel.cc"
  502. "${GE_CODE_DIR}/ge/host_kernels/reformat_kernel.cc"
  503. "${GE_CODE_DIR}/ge/host_kernels/kernel_utils.cc"
  504. "${GE_CODE_DIR}/ge/host_kernels/expanddims_kernel.cc"
  505. "${GE_CODE_DIR}/ge/host_kernels/ssd_prior_box_kernel.cc"
  506. "${GE_CODE_DIR}/ge/host_kernels/pack_kernel.cc"
  507. "${GE_CODE_DIR}/ge/host_kernels/strided_slice_kernel.cc"
  508. "${GE_CODE_DIR}/ge/host_kernels/concat_v2_kernel.cc"
  509. "${GE_CODE_DIR}/ge/host_kernels/add_kernel.cc"
  510. "${GE_CODE_DIR}/ge/host_kernels/sub_kernel.cc"
  511. "${GE_CODE_DIR}/ge/host_kernels/reduce_prod_kernel.cc"
  512. "${GE_CODE_DIR}/ge/host_kernels/rsqrt_kernel.cc"
  513. "${GE_CODE_DIR}/ge/host_kernels/concat_offset_kernel.cc"
  514. "${GE_CODE_DIR}/ge/host_kernels/slice_kernel.cc"
  515. "${GE_CODE_DIR}/ge/host_kernels/dynamic_stitch_kernel.cc"
  516. )
  517. set(SINGLE_OP_SRC_FILES
  518. "${GE_CODE_DIR}/ge/single_op/task/build_task_utils.cc"
  519. "${GE_CODE_DIR}/ge/single_op/task/op_task.cc"
  520. "${GE_CODE_DIR}/ge/single_op/task/tbe_task_builder.cc"
  521. "${GE_CODE_DIR}/ge/single_op/single_op.cc"
  522. "${GE_CODE_DIR}/ge/single_op/single_op_model.cc"
  523. "${GE_CODE_DIR}/ge/single_op/stream_resource.cc"
  524. "${GE_CODE_DIR}/ge/single_op/single_op_manager.cc"
  525. "${GE_CODE_DIR}/ge/single_op/task/aicpu_task_builder.cc"
  526. "${GE_CODE_DIR}/ge/single_op/task/aicpu_kernel_task_builder.cc"
  527. )
  528. # test files
  529. set(COMMON_TEST_FILES
  530. "graph/passes/graph_builder_utils.cc"
  531. "test.cc"
  532. )
  533. set(DISTINCT_GRAPH_LOAD_TEST_FILES
  534. "graph/load/data_dumper_unittest.cc"
  535. #"graph/load/new_model_manager_data_inputer_unittest.cc"
  536. #"graph/load/new_model_manager_davinci_model_unittest.cc"
  537. #"graph/load/new_model_manager_model_manager_unittest.cc"
  538. #"graph/load/new_model_manager_task_build_unittest.cc"
  539. "graph/load/end_graph_task_unittest.cc"
  540. "graph/load/new_model_manager_event_manager_unittest.cc"
  541. #"graph/load/output_net_output_unittest.cc"
  542. "graph/load/tbe_handle_store_unittest.cc"
  543. #"graph/graph_load_unittest.cc"
  544. "graph/ge_executor_unittest.cc"
  545. )
  546. set(PASS_TEST_FILES
  547. "graph/passes/prune_pass_unittest.cc"
  548. "graph/passes/enter_pass_unittest.cc"
  549. "graph/passes/switch_op_pass_unittest.cc"
  550. "graph/passes/get_original_format_pass_unittest.cc"
  551. "graph/passes/pass_manager_unittest.cc"
  552. "graph/passes/permute_pass_unittest.cc"
  553. "graph/passes/print_op_pass_unittest.cc"
  554. "graph/passes/shape_operate_op_remove_pass_unittest.cc"
  555. "graph/passes/unused_and_isolated_op_remove_pass_unittest.cc"
  556. "graph/passes/variable_op_pass_unittest.cc"
  557. "graph/passes/base_pass_unittest.cc"
  558. "graph/passes/addn_pass_unittest.cc"
  559. "graph/passes/save_pass_unittest.cc"
  560. "graph/passes/merge_pass_unittest.cc"
  561. #"graph/passes/switch_pass_unittest.cc"
  562. "graph/passes/switch_logic_remove_pass_unittest.cc"
  563. "graph/passes/assert_pass_unittest.cc"
  564. "graph/passes/dropout_pass_unittest.cc"
  565. "graph/passes/unused_const_pass_unittest.cc"
  566. "graph/passes/reshape_remove_pass_unittest.cc"
  567. "graph/passes/resource_pair_control_pass_unittest.cc"
  568. "graph/passes/trans_op_breadth_fusion_pass_unittest.cc"
  569. "graph/passes/trans_op_depth_fusion_pass_unittest.cc"
  570. "graph/passes/transop_nearby_allreduce_fusion_pass_unittest.cc"
  571. "graph/passes/constant_folding_pass_unittest.cc"
  572. "graph/passes/stop_gradient_pass_unittest.cc"
  573. "graph/passes/prevent_gradient_pass_unittest.cc"
  574. "graph/passes/identity_pass_unittest.cc"
  575. "graph/passes/placeholder_with_default_pass_unittest.cc"
  576. "graph/passes/snapshot_pass_unittest.cc"
  577. "graph/passes/guarantee_const_pass_unittest.cc"
  578. "graph/passes/flow_ctrl_pass_unittest.cc"
  579. "graph/passes/dimension_compute_pass_unittest.cc"
  580. "graph/passes/variable_prepare_pass_unittest.cc"
  581. "graph/passes/variable_ref_delete_pass_unittest.cc"
  582. "graph/passes/dimension_adjust_pass_unittest.cc"
  583. "graph/passes/pass_utils_unittest.cc"
  584. "graph/passes/net_output_pass_unittest.cc"
  585. "graph/passes/no_use_reshape_remove_pass_unittest.cc"
  586. "graph/passes/infershape_pass_unittest.cc"
  587. )
  588. set(KERNEL_TEST_FILES
  589. "graph/passes/folding_kernel/greater_kernel_unittest.cc"
  590. "graph/passes/folding_kernel/maximum_kernel_unittest.cc"
  591. "graph/passes/folding_kernel/floormod_kernel_unittest.cc"
  592. "graph/passes/folding_kernel/floordiv_kernel_unittest.cc"
  593. "graph/passes/folding_kernel/range_kernel_unittest.cc"
  594. "graph/passes/folding_kernel/mul_kernel_unittest.cc"
  595. "graph/passes/folding_kernel/shape_kernel_unittest.cc"
  596. "graph/passes/folding_kernel/shape_n_kernel_unittest.cc"
  597. "graph/passes/folding_kernel/size_kernel_unittest.cc"
  598. "graph/passes/folding_kernel/broadcast_args_kernel_unittest.cc"
  599. "graph/passes/folding_kernel/broadcast_gradient_args_kernel_unittest.cc"
  600. "graph/passes/folding_kernel/fill_kernel_unittest.cc"
  601. "graph/passes/folding_kernel/empty_kernel_unittest.cc"
  602. "graph/passes/folding_kernel/rank_kernel_unittest.cc"
  603. "graph/passes/folding_kernel/expanddims_kernel_unittest.cc"
  604. "graph/passes/folding_kernel/reshape_kernel_unittest.cc"
  605. "graph/passes/folding_kernel/reformat_kernel_unittest.cc"
  606. "graph/passes/folding_kernel/squeeze_kernel_unittest.cc"
  607. "graph/passes/folding_kernel/transdata_kernel_unittest.cc"
  608. "graph/passes/folding_kernel/permute_kernel_unittest.cc"
  609. "graph/passes/folding_kernel/cast_kernel_unittest.cc"
  610. "graph/passes/folding_kernel/ssd_prior_box_kernel_unittest.cc"
  611. "graph/passes/folding_kernel/strided_slice_kernel_unittest.cc"
  612. "graph/passes/folding_kernel/pack_kernel_unittest.cc"
  613. "graph/passes/folding_kernel/concat_v2_kernel_unittest.cc"
  614. "graph/passes/folding_kernel/add_kernel_unittest.cc"
  615. "graph/passes/folding_kernel/sub_kernel_unittest.cc"
  616. "graph/passes/folding_kernel/reduce_prod_kernel_unittest.cc"
  617. "graph/passes/folding_kernel/rsqrt_kernel_unittest.cc"
  618. "graph/passes/folding_kernel/concat_offset_kernel_unittest.cc"
  619. "graph/passes/folding_kernel/gather_v2_kernel_unittest.cc"
  620. "graph/passes/folding_kernel/slice_kernel_unittest.cc"
  621. "graph/passes/folding_kernel/dynamic_stitch_kernel_unittest.cc"
  622. )
  623. set(MULTI_PARTS_TEST_FILES
  624. "graph_ir/ge_operator_factory_unittest.cc"
  625. "graph/transop_util_unittest.cc"
  626. "common/datatype_transfer_unittest.cc"
  627. "common/format_transfer_unittest.cc"
  628. "common/format_transfer_transpose_unittest.cc"
  629. "common/format_transfer_nchw_5d_unittest.cc"
  630. "common/format_transfer_nchw_fractalz_unittest.cc"
  631. "common/format_transfer_hwcn_fractalz_unittest.cc"
  632. "common/format_transfer_nhwc_fractalz_unittest.cc"
  633. #"common/format_transfer_fractal_nz_unittest.cc"
  634. "common/format_transfer_fractal_zz_unittest.cc"
  635. "common/format_transfer_nhwc_5d_unittest.cc"
  636. "common/format_transfer_5d_nchw_unittest.cc"
  637. "common/format_transfer_5d_nhwc_unittest.cc"
  638. "common/format_transfer_hwcn_c1hwncoc0_unittest.cc"
  639. "common/format_transfer_c1hwncoc0_hwcn_unittest.cc"
  640. "common/format_transfer_fracz_nchw_unittest.cc"
  641. "common/format_transfer_fracz_nhwc_unittest.cc"
  642. "common/format_transfer_fracz_hwcn_unittest.cc"
  643. "common/ge_format_util_unittest.cc"
  644. "graph/variable_accelerate_ctrl_unittest.cc"
  645. "graph/build/logical_stream_allocator_unittest.cc"
  646. "graph/build/mem_assigner_unittest.cc"
  647. )
  648. set(SINGLE_OP_TEST_FILES
  649. "single_op/single_op_model_unittest.cc"
  650. "single_op/single_op_manager_unittest.cc"
  651. "single_op/stream_resource_unittest.cc"
  652. )
  653. set(PROFILING_MNG_TEST_FILES
  654. "profiling/ge_profiling_manager_unittest.cc"
  655. )
  656. set(OTHERS_TEST_FILES
  657. "plugin_manager/ge_util_unittest.cc"
  658. )
  659. list(APPEND COMMON_SHARED_LIBRARIES
  660. omg_stub
  661. c_sec
  662. slog_stub
  663. cce_ge_stub
  664. runtime_stub
  665. profiler_stub
  666. mmpa_stub
  667. hccl_stub
  668. error_manager_stub
  669. )
  670. # build common
  671. add_library(ge_ut_common STATIC ${COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  672. target_compile_definitions(ge_ut_common PRIVATE
  673. google=ascend_private
  674. )
  675. target_link_libraries(ge_ut_common PRIVATE
  676. $<BUILD_INTERFACE:intf_pub>
  677. c_sec
  678. ascend_protobuf
  679. json
  680. )
  681. # build common format
  682. add_library(ge_ut_common_format STATIC ${COMMON_SRC_FILES} ${COMMON_FORMAT_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  683. target_compile_definitions(ge_ut_common_format PRIVATE
  684. google=ascend_private
  685. )
  686. target_link_libraries(ge_ut_common_format PRIVATE
  687. $<BUILD_INTERFACE:intf_pub>
  688. c_sec
  689. ascend_protobuf
  690. json
  691. )
  692. # build graph prepare common
  693. add_library(ge_prepare_common STATIC ${GRAPH_PREPARE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  694. target_compile_definitions(ge_prepare_common PRIVATE
  695. google=ascend_private
  696. )
  697. target_link_libraries(ge_prepare_common PRIVATE
  698. $<BUILD_INTERFACE:intf_pub>
  699. c_sec
  700. ascend_protobuf
  701. json
  702. )
  703. # build graph optimize common
  704. add_library(ge_optimize_common STATIC ${GRAPH_OPTIMIZE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  705. target_compile_definitions(ge_optimize_common PRIVATE
  706. google=ascend_private
  707. )
  708. target_link_libraries(ge_optimize_common PRIVATE
  709. $<BUILD_INTERFACE:intf_pub>
  710. ascend_protobuf
  711. c_sec
  712. json
  713. )
  714. # build graph partition common
  715. add_library(ge_partition_common STATIC ${GRAPH_PARTITION_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  716. target_compile_definitions(ge_partition_common PRIVATE
  717. google=ascend_private
  718. )
  719. target_link_libraries(ge_partition_common PRIVATE
  720. $<BUILD_INTERFACE:intf_pub>
  721. ascend_protobuf
  722. c_sec
  723. json
  724. )
  725. # build build graph load common
  726. add_library(ge_load_common STATIC ${GRAPH_LOAD_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  727. target_compile_definitions(ge_load_common PRIVATE
  728. google=ascend_private
  729. )
  730. target_link_libraries(ge_load_common PRIVATE
  731. $<BUILD_INTERFACE:intf_pub>
  732. c_sec
  733. ascend_protobuf
  734. json
  735. )
  736. # build graph execute common
  737. add_library(ge_execute_common STATIC ${GRAPH_EXECUTE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  738. target_compile_definitions(ge_execute_common PRIVATE
  739. google=ascend_private
  740. )
  741. target_link_libraries(ge_execute_common PRIVATE
  742. $<BUILD_INTERFACE:intf_pub>
  743. c_sec
  744. json
  745. ascend_protobuf
  746. )
  747. # build graph build common
  748. add_library(ge_build_common STATIC ${GRAPH_BUILD_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  749. target_compile_definitions(ge_build_common PRIVATE
  750. google=ascend_private
  751. )
  752. target_link_libraries(ge_build_common PRIVATE
  753. $<BUILD_INTERFACE:intf_pub>
  754. c_sec
  755. ascend_protobuf
  756. json
  757. )
  758. # build graph pass common
  759. add_library(ge_pass_common STATIC ${GRAPH_PASS_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  760. target_compile_definitions(ge_pass_common PRIVATE
  761. google=ascend_private
  762. )
  763. target_link_libraries(ge_pass_common PRIVATE
  764. $<BUILD_INTERFACE:intf_pub>
  765. ascend_protobuf
  766. c_sec
  767. json
  768. )
  769. # build single_op common
  770. add_library(ge_single_op STATIC ${SINGLE_OP_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
  771. target_compile_definitions(ge_single_op PRIVATE
  772. google=ascend_private
  773. )
  774. target_link_libraries(ge_single_op PRIVATE
  775. $<BUILD_INTERFACE:intf_pub>
  776. ascend_protobuf
  777. json
  778. c_sec
  779. )
  780. # ut binary
  781. # libge_mutiparts_utest
  782. add_executable(ut_libge_multiparts_utest
  783. ${COMMON_TEST_FILES}
  784. ${COMMON_FORMAT_SRC_FILES}
  785. ${MULTI_PARTS_TEST_FILES}
  786. )
  787. target_compile_definitions(ut_libge_multiparts_utest PRIVATE
  788. google=ascend_private
  789. )
  790. target_link_libraries(ut_libge_multiparts_utest
  791. $<BUILD_INTERFACE:intf_pub>
  792. ge_build_common ge_load_common ge_execute_common ge_optimize_common ge_partition_common ge_prepare_common ge_single_op ge_ut_common gtest gtest_main ascend_protobuf ${COMMON_SHARED_LIBRARIES} json -lrt -ldl
  793. )
  794. # libge_others_utest
  795. add_executable(ut_libge_others_utest
  796. ${COMMON_TEST_FILES}
  797. ${COMMON_FORMAT_SRC_FILES}
  798. ${PASS_TEST_FILES}
  799. ${EXECUTE_TEST_FILES}
  800. ${OTHERS_TEST_FILES}
  801. )
  802. target_link_libraries(ut_libge_others_utest
  803. $<BUILD_INTERFACE:intf_pub>
  804. ge_load_common ge_execute_common ge_ut_common gtest gtest_main ascend_protobuf ${COMMON_SHARED_LIBRARIES} json -lrt -ldl
  805. )
  806. # libge_kernel_utest
  807. add_executable(ut_libge_kernel_utest
  808. ${COMMON_TEST_FILES}
  809. ${COMMON_FORMAT_SRC_FILES}
  810. ${KERNEL_TEST_FILES}
  811. ${KERNEL_SRC_FILES}
  812. )
  813. target_link_libraries(ut_libge_kernel_utest
  814. $<BUILD_INTERFACE:intf_pub>
  815. ge_load_common ge_ut_common gtest gtest_main ascend_protobuf ${COMMON_SHARED_LIBRARIES} json -lrt -ldl
  816. )
  817. # libge_distinct_load_utest
  818. add_executable(ut_libge_distinct_load_utest
  819. ${COMMON_TEST_FILES}
  820. ${DISTINCT_GRAPH_LOAD_TEST_FILES}
  821. ${DISTINCT_GRAPH_LOAD_SRC_FILES}
  822. ${SINGLE_OP_TEST_FILES}
  823. ${PROFILING_MNG_TEST_FILES}
  824. )
  825. target_compile_definitions(ut_libge_distinct_load_utest PRIVATE
  826. google=ascend_private
  827. )
  828. target_link_libraries(ut_libge_distinct_load_utest
  829. ${COMMON_SHARED_LIBRARIES}
  830. $<BUILD_INTERFACE:intf_pub>
  831. ge_execute_common ge_ut_common_format ge_load_common
  832. ge_single_op ge_prepare_common
  833. ge_optimize_common ge_build_common ge_partition_common ge_ut_common
  834. gtest gtest_main ascend_protobuf json c_sec -lrt -ldl -lpthread
  835. )

图引擎模块(GE)是MindSpore的一个子模块,其代码由C++实现,位于前端模块ME和底层硬件之间,起到承接作用。图引擎模块以ME下发的图作为输入,然后进行一系列的深度图优化操作,最后输出一张可以在底层硬件上高效运行的图。GE针对昇腾AI处理器的硬件结构特点,做了特定的优化工作,以此来充分发挥出昇腾AI处理器的强大算力。在进行模型训练/推理时,GE会被自动调用而用户并不感知。GE主要由GE API和GE Core两部分组成,详细的架构图如下所示