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 36 kB

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
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
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928
  1. if (NOT ENABLE_D AND NOT ENABLE_ACL)
  2. add_subdirectory(common)
  3. add_subdirectory(plugin/engine)
  4. add_subdirectory(graph/build/memory)
  5. add_subdirectory(ge_local_engine)
  6. add_subdirectory(host_cpu_engine)
  7. add_subdirectory(executor)
  8. add_subdirectory(offline)
  9. elseif (ENABLE_D)
  10. add_subdirectory(common)
  11. add_subdirectory(ge_runtime)
  12. endif ()
  13. set(PROTO_LIST
  14. "${METADEF_DIR}/proto/fusion_model.proto"
  15. "${GE_CODE_DIR}/ge/proto/optimizer_priority.proto"
  16. )
  17. set(PROTO_CLIENT_LIST
  18. "${METADEF_DIR}/proto/ge_api.proto"
  19. )
  20. set(PROTO_HEADER_LIST
  21. "${METADEF_DIR}/proto/om.proto"
  22. "${METADEF_DIR}/proto/task.proto"
  23. "${METADEF_DIR}/proto/insert_op.proto"
  24. "${METADEF_DIR}/proto/ge_ir.proto"
  25. "${METADEF_DIR}/proto/fwk_adapter.proto"
  26. "${METADEF_DIR}/proto/op_mapping_info.proto"
  27. )
  28. protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})
  29. protobuf_generate(ge PROTO_CLIENT_SRCS PROTO_CLIENT_HDRS ${PROTO_CLIENT_LIST})
  30. protobuf_generate(ge PROTO_HEADER_SRCS PROTO_HEADER_HDRS ${PROTO_HEADER_LIST})
  31. set(TRAIN_SRC_LIST
  32. "common/formats/format_transfers/datatype_transfer.cc"
  33. "common/formats/format_transfers/format_transfer_c1hwncoc0_hwcn.cc"
  34. "common/formats/format_transfers/format_transfer_dhwcn_fracz3D.cc"
  35. "common/formats/format_transfers/format_transfer_dhwnc_fracz3D_transpose.cc"
  36. "common/formats/format_transfers/format_transfer_fractal_nz.cc"
  37. "common/formats/format_transfers/format_transfer_fractal_z.cc"
  38. "common/formats/format_transfers/format_transfer_fractal_zz.cc"
  39. "common/formats/format_transfers/format_transfer_fracz_hwcn.cc"
  40. "common/formats/format_transfers/format_transfer_fracz_nchw.cc"
  41. "common/formats/format_transfers/format_transfer_fracz_nhwc.cc"
  42. "common/formats/format_transfers/format_transfer_hwcn_c1hwncoc0.cc"
  43. "common/formats/format_transfers/format_transfer_nc1hwc0_nchw.cc"
  44. "common/formats/format_transfers/format_transfer_nc1hwc0_nhwc.cc"
  45. "common/formats/format_transfers/format_transfer_nchw_nc1hwc0.cc"
  46. "common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.cc"
  47. "common/formats/format_transfers/format_transfer_transpose.cc"
  48. "common/formats/formats.cc"
  49. "common/formats/utils/formats_trans_utils.cc"
  50. "common/fp16_t.cc"
  51. "common/ge/plugin_manager.cc"
  52. "common/ge/op_tiling_manager.cc"
  53. "common/helper/model_cache_helper.cc"
  54. "common/profiling/profiling_manager.cc"
  55. "common/dump/dump_manager.cc"
  56. "common/dump/dump_properties.cc"
  57. "common/dump/dump_op.cc"
  58. "engine_manager/dnnengine_manager.cc"
  59. "ge_local_engine/engine/host_cpu_engine.cc"
  60. "generator/ge_generator.cc"
  61. "generator/generator_api.cc"
  62. "graph/build/graph_builder.cc"
  63. "graph/build/label_allocator.cc"
  64. "graph/build/logical_stream_allocator.cc"
  65. "graph/build/model_builder.cc"
  66. "graph/build/run_context.cc"
  67. "graph/build/stream_allocator.cc"
  68. "graph/build/stream_graph_optimizer.cc"
  69. "graph/build/task_generator.cc"
  70. "graph/common/bcast.cc"
  71. "graph/common/local_context.cc"
  72. "graph/common/omg_util.cc"
  73. "graph/common/transop_util.cc"
  74. "graph/execute/graph_execute.cc"
  75. "graph/label/case_label_maker.cc"
  76. "graph/label/if_label_maker.cc"
  77. "graph/label/label_maker.cc"
  78. "graph/label/partitioned_call_label_maker.cc"
  79. "graph/label/while_label_maker.cc"
  80. "graph/load/graph_loader.cc"
  81. "graph/load/new_model_manager/cpu_queue_schedule.cc"
  82. "graph/load/new_model_manager/data_dumper.cc"
  83. "graph/load/new_model_manager/data_inputer.cc"
  84. "graph/load/new_model_manager/davinci_model.cc"
  85. "graph/load/new_model_manager/davinci_model_parser.cc"
  86. "graph/load/new_model_manager/model_manager.cc"
  87. "graph/load/new_model_manager/model_utils.cc"
  88. "graph/load/new_model_manager/aipp_utils.cc"
  89. "graph/load/new_model_manager/task_info/end_graph_task_info.cc"
  90. "graph/load/new_model_manager/task_info/model_exit_task_info.cc"
  91. "graph/load/new_model_manager/task_info/event_record_task_info.cc"
  92. "graph/load/new_model_manager/task_info/event_wait_task_info.cc"
  93. "graph/load/new_model_manager/task_info/fusion_start_task_info.cc"
  94. "graph/load/new_model_manager/task_info/fusion_stop_task_info.cc"
  95. "graph/load/new_model_manager/task_info/hccl_task_info.cc"
  96. "graph/load/new_model_manager/task_info/kernel_ex_task_info.cc"
  97. "graph/load/new_model_manager/task_info/kernel_task_info.cc"
  98. "graph/load/new_model_manager/task_info/label_set_task_info.cc"
  99. "graph/load/new_model_manager/task_info/label_switch_by_index_task_info.cc"
  100. "graph/load/new_model_manager/task_info/label_goto_ex_task_info.cc"
  101. "graph/load/new_model_manager/task_info/memcpy_addr_async_task_info.cc"
  102. "graph/load/new_model_manager/task_info/memcpy_async_task_info.cc"
  103. "graph/load/new_model_manager/task_info/profiler_trace_task_info.cc"
  104. "graph/load/new_model_manager/task_info/stream_active_task_info.cc"
  105. "graph/load/new_model_manager/task_info/stream_switch_task_info.cc"
  106. "graph/load/new_model_manager/task_info/stream_switchn_task_info.cc"
  107. "graph/load/new_model_manager/task_info/super_kernel/super_kernel.cc"
  108. "graph/load/new_model_manager/task_info/super_kernel/super_kernel_factory.cc"
  109. "graph/load/new_model_manager/task_info/task_info.cc"
  110. "graph/load/new_model_manager/tbe_handle_store.cc"
  111. "graph/load/new_model_manager/zero_copy_task.cc"
  112. "graph/load/new_model_manager/zero_copy_offset.cc"
  113. "graph/manager/graph_context.cc"
  114. "graph/manager/graph_manager.cc"
  115. "graph/manager/graph_manager_utils.cc"
  116. "graph/manager/graph_mem_allocator.cc"
  117. "graph/manager/graph_caching_allocator.cc"
  118. "graph/manager/graph_var_manager.cc"
  119. "graph/manager/host_mem_manager.cc"
  120. "graph/manager/rdma_pool_allocator.cc"
  121. "graph/manager/memory_api.cc"
  122. "graph/manager/model_manager/event_manager.cc"
  123. "graph/manager/trans_var_data_utils.cc"
  124. "graph/manager/util/debug.cc"
  125. "graph/manager/util/hcom_util.cc"
  126. "graph/manager/util/rt_context_util.cc"
  127. "graph/manager/util/variable_accelerate_ctrl.cc"
  128. "graph/optimize/graph_optimize.cc"
  129. "graph/optimize/mem_rw_conflict_optimize.cc"
  130. "graph/optimize/summary_optimize.cc"
  131. "graph/partition/engine_place.cc"
  132. "graph/partition/graph_partition.cc"
  133. "graph/passes/addn_pass.cc"
  134. "graph/passes/aicpu_constant_folding_pass.cc"
  135. "graph/passes/assert_pass.cc"
  136. "graph/passes/input_output_connection_identify_pass.cc"
  137. "graph/passes/atomic_addr_clean_pass.cc"
  138. "graph/passes/mark_same_addr_pass.cc"
  139. "graph/passes/mark_graph_unknown_status_pass.cc"
  140. "graph/passes/mark_agnostic_pass.cc"
  141. "graph/partition/dynamic_shape_partition.cc"
  142. "graph/partition/stage_partition.cc"
  143. "graph/passes/base_pass.cc"
  144. "graph/passes/bitcast_pass.cc"
  145. "graph/passes/cast_remove_pass.cc"
  146. "graph/passes/cast_translate_pass.cc"
  147. "graph/passes/common_subexpression_elimination_pass.cc"
  148. "graph/passes/transop_symmetry_elimination_pass.cc"
  149. "graph/passes/compile_nodes_pass.cc"
  150. "graph/passes/constant_folding_pass.cc"
  151. "graph/passes/constant_fuse_same_pass.cc"
  152. "graph/passes/control_trigger_pass.cc"
  153. "graph/passes/dimension_adjust_pass.cc"
  154. "graph/passes/dimension_compute_pass.cc"
  155. "graph/passes/dropout_pass.cc"
  156. "graph/passes/hccl_group_pass.cc"
  157. "graph/passes/enter_pass.cc"
  158. "graph/passes/assign_pass.cc"
  159. "graph/passes/flow_ctrl_pass.cc"
  160. "graph/passes/global_step_insert_pass.cc"
  161. "host_kernels/transpose_kernel.cc"
  162. "host_kernels/add_kernel.cc"
  163. "host_kernels/broadcast_args_kernel.cc"
  164. "host_kernels/broadcast_gradient_args_kernel.cc"
  165. "host_kernels/cast_kernel.cc"
  166. "host_kernels/concat_offset_kernel.cc"
  167. "host_kernels/concat_v2_kernel.cc"
  168. "host_kernels/dynamic_stitch_kernel.cc"
  169. "host_kernels/identity_kernel.cc"
  170. "host_kernels/empty_kernel.cc"
  171. "host_kernels/expanddims_kernel.cc"
  172. "host_kernels/fill_kernel.cc"
  173. "host_kernels/floordiv_kernel.cc"
  174. "host_kernels/floormod_kernel.cc"
  175. "host_kernels/gather_v2_kernel.cc"
  176. "host_kernels/greater_kernel.cc"
  177. "host_kernels/kernel_utils.cc"
  178. "host_kernels/maximum_kernel.cc"
  179. "host_kernels/mul_kernel.cc"
  180. "host_kernels/pack_kernel.cc"
  181. "host_kernels/permute_kernel.cc"
  182. "host_kernels/range_kernel.cc"
  183. "host_kernels/rank_kernel.cc"
  184. "host_kernels/reduce_prod_kernel.cc"
  185. "host_kernels/reshape_kernel.cc"
  186. "host_kernels/rsqrt_kernel.cc"
  187. "host_kernels/shape_kernel.cc"
  188. "host_kernels/shape_n_kernel.cc"
  189. "host_kernels/size_kernel.cc"
  190. "host_kernels/slice_d_kernel.cc"
  191. "host_kernels/slice_kernel.cc"
  192. "host_kernels/squeeze_kernel.cc"
  193. "host_kernels/unsqueeze_kernel.cc"
  194. "host_kernels/ssd_prior_box_kernel.cc"
  195. "host_kernels/strided_slice_kernel.cc"
  196. "host_kernels/sub_kernel.cc"
  197. "host_kernels/transdata_kernel.cc"
  198. "host_kernels/unpack_kernel.cc"
  199. "graph/passes/folding_pass.cc"
  200. "graph/passes/get_original_format_pass.cc"
  201. "graph/passes/guarantee_const_pass.cc"
  202. "graph/passes/hccl_memcpy_pass.cc"
  203. "graph/passes/identity_pass.cc"
  204. "graph/passes/ref_identity_delete_op_pass.cc"
  205. "graph/passes/infershape_pass.cc"
  206. "graph/passes/isolated_op_remove_pass.cc"
  207. "graph/passes/iterator_op_pass.cc"
  208. "graph/passes/link_gen_mask_nodes_pass.cc"
  209. "graph/passes/merge_pass.cc"
  210. "graph/passes/multi_batch_pass.cc"
  211. "graph/passes/multi_batch_clone_pass.cc"
  212. "graph/passes/subexpression_migration_pass.cc"
  213. "graph/passes/subgraph_const_migration_pass.cc"
  214. "graph/passes/unused_args_clean_pass.cc"
  215. "graph/passes/net_output_pass.cc"
  216. "graph/passes/next_iteration_pass.cc"
  217. "graph/passes/no_use_reshape_remove_pass.cc"
  218. "graph/passes/pass_manager.cc"
  219. "graph/passes/pass_utils.cc"
  220. "graph/passes/permute_pass.cc"
  221. "graph/passes/placeholder_with_default_pass.cc"
  222. "graph/passes/prevent_gradient_pass.cc"
  223. "graph/passes/print_op_pass.cc"
  224. "graph/passes/prune_pass.cc"
  225. "graph/passes/ctrl_edge_transfer_pass.cc"
  226. "graph/passes/replace_with_empty_const_pass.cc"
  227. "graph/passes/reshape_remove_pass.cc"
  228. "graph/passes/reshape_recovery_pass.cc"
  229. "graph/passes/resource_pair_add_control_pass.cc"
  230. "graph/passes/resource_pair_remove_control_pass.cc"
  231. "graph/passes/same_transdata_breadth_fusion_pass.cc"
  232. "graph/passes/save_pass.cc"
  233. "graph/passes/shape_operate_op_remove_pass.cc"
  234. "graph/passes/snapshot_pass.cc"
  235. "graph/passes/stop_gradient_pass.cc"
  236. "graph/passes/subgraph_pass.cc"
  237. "graph/passes/data_pass.cc"
  238. "graph/passes/switch_data_edges_bypass.cc"
  239. "graph/passes/switch_logic_remove_pass.cc"
  240. "graph/passes/merge_to_stream_merge_pass.cc"
  241. "graph/passes/merge_input_memcpy_pass.cc"
  242. "graph/passes/switch_to_stream_switch_pass.cc"
  243. "graph/passes/attach_stream_label_pass.cc"
  244. "graph/passes/switch_dead_branch_elimination.cc"
  245. "graph/passes/replace_transshape_pass.cc"
  246. "graph/passes/transop_breadth_fusion_pass.cc"
  247. "graph/passes/transop_depth_fusion_pass.cc"
  248. "graph/passes/transop_nearby_allreduce_fusion_pass.cc"
  249. "graph/passes/transop_without_reshape_fusion_pass.cc"
  250. "graph/passes/transpose_transdata_pass.cc"
  251. "graph/passes/unused_const_pass.cc"
  252. "graph/passes/unused_op_remove_pass.cc"
  253. "graph/passes/var_is_initialized_op_pass.cc"
  254. "graph/passes/parallel_concat_start_op_pass.cc"
  255. "graph/passes/cond_pass.cc"
  256. "graph/passes/cond_remove_pass.cc"
  257. "graph/passes/for_pass.cc"
  258. "graph/passes/variable_format_pass.cc"
  259. "graph/passes/variable_op_pass.cc"
  260. "graph/passes/variable_prepare_op_pass.cc"
  261. "graph/passes/variable_ref_delete_op_pass.cc"
  262. "graph/passes/variable_ref_useless_control_out_delete_pass.cc"
  263. "graph/passes/end_of_sequence_add_control_pass.cc"
  264. "graph/passes/memcpy_addr_async_pass.cc"
  265. "graph/passes/set_input_output_offset_pass.cc"
  266. "graph/preprocess/graph_preprocess.cc"
  267. "graph/preprocess/insert_op/ge_aipp_op.cc"
  268. "graph/preprocess/insert_op/util_insert_aipp_op.cc"
  269. "graph/preprocess/multi_batch_options.cc"
  270. "graph/preprocess/multi_batch_copy_graph.cc"
  271. "init/gelib.cc"
  272. "model/ge_model.cc"
  273. "model/ge_root_model.cc"
  274. "omm/csa_interact.cc"
  275. "opskernel_manager/ops_kernel_manager.cc"
  276. "opskernel_manager/ops_kernel_builder_manager.cc"
  277. "session/inner_session.cc"
  278. "session/session_manager.cc"
  279. "single_op/single_op.cc"
  280. "single_op/single_op_manager.cc"
  281. "single_op/single_op_model.cc"
  282. "single_op/stream_resource.cc"
  283. "single_op/task/build_task_utils.cc"
  284. "single_op/task/op_task.cc"
  285. "single_op/task/tbe_task_builder.cc"
  286. "single_op/task/aicpu_task_builder.cc"
  287. "single_op/task/aicpu_kernel_task_builder.cc"
  288. "hybrid/common/tensor_value.cc"
  289. "hybrid/common/npu_memory_allocator.cc"
  290. "hybrid/executor/rt_callback_manager.cc"
  291. "hybrid/executor/node_state.cc"
  292. "hybrid/executor/node_done_manager.cc"
  293. "hybrid/executor/hybrid_profiler.cc"
  294. "hybrid/executor/hybrid_model_executor.cc"
  295. "hybrid/executor/hybrid_model_async_executor.cc"
  296. "hybrid/executor/hybrid_execution_context.cc"
  297. "hybrid/executor/subgraph_context.cc"
  298. "hybrid/executor/subgraph_executor.cc"
  299. "hybrid/executor/worker/task_compile_engine.cc"
  300. "hybrid/executor/worker/shape_inference_engine.cc"
  301. "hybrid/executor/worker/execution_engine.cc"
  302. "hybrid/model/hybrid_model.cc"
  303. "hybrid/model/hybrid_model_builder.cc"
  304. "hybrid/model/node_item.cc"
  305. "hybrid/model/graph_item.cc"
  306. "hybrid/node_executor/aicore/aicore_node_executor.cc"
  307. "hybrid/node_executor/aicore/aicore_op_task.cc"
  308. "hybrid/node_executor/aicore/aicore_task_builder.cc"
  309. "hybrid/node_executor/aicore/aicore_task_compiler.cc"
  310. "hybrid/node_executor/aicpu/aicpu_ext_info.cc"
  311. "hybrid/node_executor/aicpu/aicpu_node_executor.cc"
  312. "hybrid/node_executor/compiledsubgraph/known_node_executor.cc"
  313. "hybrid/node_executor/ge_local/ge_local_node_executor.cc"
  314. "hybrid/node_executor/host_cpu/host_cpu_node_executor.cc"
  315. "hybrid/node_executor/host_cpu/kernel_factory.cc"
  316. "hybrid/node_executor/host_cpu/kernel/no_op_kernel.cc"
  317. "hybrid/node_executor/host_cpu/kernel/variable_kernel.cc"
  318. "hybrid/node_executor/host_cpu/kernel/assign_kernel.cc"
  319. "hybrid/node_executor/host_cpu/kernel/random_uniform_kernel.cc"
  320. "hybrid/node_executor/controlop/control_op_executor.cc"
  321. "hybrid/node_executor/partitioned_call/partitioned_call_node_executor.cc"
  322. "hybrid/node_executor/hccl/hccl_node_executor.cc"
  323. "hybrid/node_executor/rts/rts_node_executor.cc"
  324. "hybrid/node_executor/node_executor.cc"
  325. "hybrid/node_executor/task_context.cc"
  326. "hybrid/hybrid_davinci_model.cc"
  327. "executor/ge_executor.cc"
  328. "client/ge_api.cc"
  329. "client/ge_prof.cc"
  330. "analyzer/analyzer.cc"
  331. "ir_build/ge_ir_build.cc"
  332. "ir_build/atc_ir_common.cc"
  333. )
  334. set(INFER_SRC_LIST
  335. "graph/manager/trans_var_data_utils.cc"
  336. "omm/csa_interact.cc"
  337. "common/fp16_t.cc"
  338. "common/formats/utils/formats_trans_utils.cc"
  339. "common/formats/format_transfers/datatype_transfer.cc"
  340. "common/formats/format_transfers/format_transfer_transpose.cc"
  341. "common/formats/format_transfers/format_transfer_nchw_nc1hwc0.cc"
  342. "common/formats/format_transfers/format_transfer_fractal_z.cc"
  343. "common/formats/format_transfers/format_transfer_fractal_nz.cc"
  344. "common/formats/format_transfers/format_transfer_fractal_zz.cc"
  345. "common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.cc"
  346. "common/formats/format_transfers/format_transfer_nc1hwc0_nchw.cc"
  347. "common/formats/format_transfers/format_transfer_nc1hwc0_nhwc.cc"
  348. "common/formats/format_transfers/format_transfer_hwcn_c1hwncoc0.cc"
  349. "common/formats/format_transfers/format_transfer_c1hwncoc0_hwcn.cc"
  350. "common/formats/format_transfers/format_transfer_fracz_nchw.cc"
  351. "common/formats/format_transfers/format_transfer_fracz_nhwc.cc"
  352. "common/formats/format_transfers/format_transfer_fracz_hwcn.cc"
  353. "common/formats/format_transfers/format_transfer_dhwcn_fracz3D.cc"
  354. "common/formats/format_transfers/format_transfer_dhwnc_fracz3D_transpose.cc"
  355. "common/formats/format_transfers/format_transfer_nchw_fz_c04.cc"
  356. "common/formats/formats.cc"
  357. "common/profiling/profiling_manager.cc"
  358. "common/dump/dump_properties.cc"
  359. "common/dump/dump_manager.cc"
  360. "common/dump/dump_op.cc"
  361. "common/dump/dump_server.cc"
  362. "common/helper/model_cache_helper.cc"
  363. "ge_local_engine/engine/host_cpu_engine.cc"
  364. "common/ge/plugin_manager.cc"
  365. "common/ge/op_tiling_manager.cc"
  366. "init/gelib.cc"
  367. "session/inner_session.cc"
  368. "session/session_manager.cc"
  369. "engine_manager/dnnengine_manager.cc"
  370. "opskernel_manager/ops_kernel_manager.cc"
  371. "opskernel_manager/ops_kernel_builder_manager.cc"
  372. "graph/manager/graph_manager.cc"
  373. "graph/manager/graph_manager_utils.cc"
  374. "graph/manager/graph_context.cc"
  375. "graph/preprocess/graph_preprocess.cc"
  376. "graph/preprocess/multi_batch_options.cc"
  377. "graph/preprocess/multi_batch_copy_graph.cc"
  378. "graph/execute/graph_execute.cc"
  379. "graph/load/graph_loader.cc"
  380. "graph/optimize/graph_optimize.cc"
  381. "graph/optimize/mem_rw_conflict_optimize.cc"
  382. "graph/optimize/summary_optimize.cc"
  383. "graph/build/graph_builder.cc"
  384. "graph/partition/engine_place.cc"
  385. "graph/partition/graph_partition.cc"
  386. "graph/partition/dynamic_shape_partition.cc"
  387. "graph/partition/stage_partition.cc"
  388. "generator/ge_generator.cc"
  389. "generator/generator_api.cc"
  390. "graph/manager/graph_var_manager.cc"
  391. "graph/manager/host_mem_manager.cc"
  392. "graph/manager/rdma_pool_allocator.cc"
  393. "graph/manager/graph_mem_allocator.cc"
  394. "graph/manager/graph_caching_allocator.cc"
  395. "model/ge_model.cc"
  396. "model/ge_root_model.cc"
  397. "graph/common/transop_util.cc"
  398. "graph/passes/pass_manager.cc"
  399. "graph/passes/resource_pair_add_control_pass.cc"
  400. "graph/passes/resource_pair_remove_control_pass.cc"
  401. "graph/passes/pass_utils.cc"
  402. "graph/passes/base_pass.cc"
  403. "graph/passes/bitcast_pass.cc"
  404. "graph/passes/constant_folding_pass.cc"
  405. "graph/passes/aicpu_constant_folding_pass.cc"
  406. "graph/passes/reshape_remove_pass.cc"
  407. "graph/passes/reshape_recovery_pass.cc"
  408. "graph/passes/transop_breadth_fusion_pass.cc"
  409. "graph/passes/transop_depth_fusion_pass.cc"
  410. "graph/passes/transop_nearby_allreduce_fusion_pass.cc"
  411. "graph/passes/same_transdata_breadth_fusion_pass.cc"
  412. "graph/passes/transop_without_reshape_fusion_pass.cc"
  413. "graph/passes/compile_nodes_pass.cc"
  414. "graph/passes/variable_prepare_op_pass.cc"
  415. "graph/passes/variable_ref_delete_op_pass.cc"
  416. "graph/passes/variable_ref_useless_control_out_delete_pass.cc"
  417. "graph/passes/subgraph_pass.cc"
  418. "graph/passes/data_pass.cc"
  419. "graph/passes/net_output_pass.cc"
  420. "graph/passes/replace_transshape_pass.cc"
  421. "graph/passes/constant_fuse_same_pass.cc"
  422. "graph/passes/print_op_pass.cc"
  423. "graph/passes/no_use_reshape_remove_pass.cc"
  424. "graph/passes/iterator_op_pass.cc"
  425. "graph/passes/input_output_connection_identify_pass.cc"
  426. "graph/passes/atomic_addr_clean_pass.cc"
  427. "graph/passes/mark_same_addr_pass.cc"
  428. "graph/passes/mark_graph_unknown_status_pass.cc"
  429. "graph/passes/mark_agnostic_pass.cc"
  430. "graph/common/omg_util.cc"
  431. "graph/common/bcast.cc"
  432. "graph/common/local_context.cc"
  433. "graph/passes/dimension_compute_pass.cc"
  434. "graph/passes/dimension_adjust_pass.cc"
  435. "graph/passes/get_original_format_pass.cc"
  436. "graph/passes/shape_operate_op_remove_pass.cc"
  437. "graph/passes/unused_op_remove_pass.cc"
  438. "graph/passes/assert_pass.cc"
  439. "graph/passes/dropout_pass.cc"
  440. "graph/passes/infershape_pass.cc"
  441. "graph/passes/unused_const_pass.cc"
  442. "graph/passes/isolated_op_remove_pass.cc"
  443. "graph/passes/permute_pass.cc"
  444. "graph/passes/ctrl_edge_transfer_pass.cc"
  445. "graph/passes/end_of_sequence_add_control_pass.cc"
  446. "host_kernels/broadcast_gradient_args_kernel.cc"
  447. "host_kernels/greater_kernel.cc"
  448. "host_kernels/gather_v2_kernel.cc"
  449. "host_kernels/maximum_kernel.cc"
  450. "host_kernels/floormod_kernel.cc"
  451. "host_kernels/floordiv_kernel.cc"
  452. "host_kernels/range_kernel.cc"
  453. "host_kernels/shape_kernel.cc"
  454. "host_kernels/size_kernel.cc"
  455. "host_kernels/shape_n_kernel.cc"
  456. "host_kernels/rank_kernel.cc"
  457. "host_kernels/broadcast_args_kernel.cc"
  458. "host_kernels/fill_kernel.cc"
  459. "host_kernels/empty_kernel.cc"
  460. "host_kernels/expanddims_kernel.cc"
  461. "host_kernels/reshape_kernel.cc"
  462. "host_kernels/squeeze_kernel.cc"
  463. "host_kernels/unsqueeze_kernel.cc"
  464. "host_kernels/kernel_utils.cc"
  465. "host_kernels/cast_kernel.cc"
  466. "host_kernels/transdata_kernel.cc"
  467. "host_kernels/unpack_kernel.cc"
  468. "host_kernels/transpose_kernel.cc"
  469. "host_kernels/permute_kernel.cc"
  470. "host_kernels/pack_kernel.cc"
  471. "host_kernels/concat_v2_kernel.cc"
  472. "host_kernels/concat_offset_kernel.cc"
  473. "host_kernels/strided_slice_kernel.cc"
  474. "host_kernels/ssd_prior_box_kernel.cc"
  475. "host_kernels/add_kernel.cc"
  476. "host_kernels/sub_kernel.cc"
  477. "host_kernels/mul_kernel.cc"
  478. "host_kernels/reduce_prod_kernel.cc"
  479. "host_kernels/rsqrt_kernel.cc"
  480. "host_kernels/slice_kernel.cc"
  481. "host_kernels/slice_d_kernel.cc"
  482. "host_kernels/dynamic_stitch_kernel.cc"
  483. "host_kernels/identity_kernel.cc"
  484. "graph/passes/stop_gradient_pass.cc"
  485. "graph/passes/prevent_gradient_pass.cc"
  486. "graph/passes/identity_pass.cc"
  487. "graph/passes/ref_identity_delete_op_pass.cc"
  488. "graph/passes/placeholder_with_default_pass.cc"
  489. "graph/passes/snapshot_pass.cc"
  490. "graph/passes/guarantee_const_pass.cc"
  491. "graph/passes/var_is_initialized_op_pass.cc"
  492. "graph/passes/parallel_concat_start_op_pass.cc"
  493. "graph/passes/folding_pass.cc"
  494. "graph/passes/cast_translate_pass.cc"
  495. "graph/passes/prune_pass.cc"
  496. "graph/passes/merge_to_stream_merge_pass.cc"
  497. "graph/passes/merge_input_memcpy_pass.cc"
  498. "graph/passes/switch_to_stream_switch_pass.cc"
  499. "graph/passes/attach_stream_label_pass.cc"
  500. "graph/passes/multi_batch_pass.cc"
  501. "graph/passes/multi_batch_clone_pass.cc"
  502. "graph/passes/subexpression_migration_pass.cc"
  503. "graph/passes/subgraph_const_migration_pass.cc"
  504. "graph/passes/unused_args_clean_pass.cc"
  505. "graph/passes/next_iteration_pass.cc"
  506. "graph/passes/control_trigger_pass.cc"
  507. "graph/passes/cond_pass.cc"
  508. "graph/passes/cond_remove_pass.cc"
  509. "graph/passes/for_pass.cc"
  510. "graph/passes/enter_pass.cc"
  511. "graph/passes/assign_pass.cc"
  512. "graph/passes/addn_pass.cc"
  513. "graph/passes/common_subexpression_elimination_pass.cc"
  514. "graph/passes/transop_symmetry_elimination_pass.cc"
  515. "graph/passes/save_pass.cc"
  516. "graph/passes/switch_dead_branch_elimination.cc"
  517. "graph/passes/switch_logic_remove_pass.cc"
  518. "graph/passes/switch_data_edges_bypass.cc"
  519. "graph/passes/merge_pass.cc"
  520. "graph/passes/variable_format_pass.cc"
  521. "graph/passes/variable_op_pass.cc"
  522. "graph/passes/cast_remove_pass.cc"
  523. "graph/passes/transpose_transdata_pass.cc"
  524. "graph/passes/hccl_memcpy_pass.cc"
  525. "graph/passes/flow_ctrl_pass.cc"
  526. "graph/passes/global_step_insert_pass.cc"
  527. "graph/passes/link_gen_mask_nodes_pass.cc"
  528. "graph/passes/replace_with_empty_const_pass.cc"
  529. "graph/passes/hccl_group_pass.cc"
  530. "graph/passes/memcpy_addr_async_pass.cc"
  531. "graph/passes/set_input_output_offset_pass.cc"
  532. "graph/manager/model_manager/event_manager.cc"
  533. "graph/manager/util/rt_context_util.cc"
  534. "graph/manager/util/variable_accelerate_ctrl.cc"
  535. "graph/manager/util/debug.cc"
  536. "graph/load/new_model_manager/model_manager.cc"
  537. "graph/load/new_model_manager/data_inputer.cc"
  538. "graph/load/new_model_manager/davinci_model.cc"
  539. "graph/load/new_model_manager/davinci_model_parser.cc"
  540. "graph/load/new_model_manager/model_utils.cc"
  541. "graph/load/new_model_manager/aipp_utils.cc"
  542. "graph/load/new_model_manager/tbe_handle_store.cc"
  543. "graph/load/new_model_manager/cpu_queue_schedule.cc"
  544. "graph/load/new_model_manager/zero_copy_task.cc"
  545. "graph/load/new_model_manager/zero_copy_offset.cc"
  546. "graph/load/new_model_manager/data_dumper.cc"
  547. "graph/load/new_model_manager/task_info/task_info.cc"
  548. "graph/load/new_model_manager/task_info/event_record_task_info.cc"
  549. "graph/load/new_model_manager/task_info/event_wait_task_info.cc"
  550. "graph/load/new_model_manager/task_info/fusion_start_task_info.cc"
  551. "graph/load/new_model_manager/task_info/fusion_stop_task_info.cc"
  552. "graph/load/new_model_manager/task_info/kernel_ex_task_info.cc"
  553. "graph/load/new_model_manager/task_info/kernel_task_info.cc"
  554. "graph/load/new_model_manager/task_info/label_set_task_info.cc"
  555. "graph/load/new_model_manager/task_info/label_switch_by_index_task_info.cc"
  556. "graph/load/new_model_manager/task_info/label_goto_ex_task_info.cc"
  557. "graph/load/new_model_manager/task_info/memcpy_async_task_info.cc"
  558. "graph/load/new_model_manager/task_info/memcpy_addr_async_task_info.cc"
  559. "graph/load/new_model_manager/task_info/profiler_trace_task_info.cc"
  560. "graph/load/new_model_manager/task_info/stream_active_task_info.cc"
  561. "graph/load/new_model_manager/task_info/stream_switch_task_info.cc"
  562. "graph/load/new_model_manager/task_info/stream_switchn_task_info.cc"
  563. "graph/load/new_model_manager/task_info/end_graph_task_info.cc"
  564. "graph/load/new_model_manager/task_info/model_exit_task_info.cc"
  565. "graph/load/new_model_manager/task_info/super_kernel/super_kernel_factory.cc"
  566. "graph/load/new_model_manager/task_info/super_kernel/super_kernel.cc"
  567. "single_op/task/op_task.cc"
  568. "single_op/task/build_task_utils.cc"
  569. "single_op/task/tbe_task_builder.cc"
  570. "single_op/task/aicpu_task_builder.cc"
  571. "single_op/task/aicpu_kernel_task_builder.cc"
  572. "single_op/single_op.cc"
  573. "single_op/single_op_model.cc"
  574. "single_op/stream_resource.cc"
  575. "single_op/single_op_manager.cc"
  576. "hybrid/hybrid_davinci_model_stub.cc"
  577. "ir_build/ge_ir_build.cc"
  578. "ir_build/atc_ir_common.cc"
  579. "graph/preprocess/insert_op/ge_aipp_op.cc"
  580. "graph/preprocess/insert_op/util_insert_aipp_op.cc"
  581. "hybrid/node_executor/aicpu/aicpu_ext_info.cc"
  582. "graph/build/model_builder.cc"
  583. "graph/build/task_generator.cc"
  584. "graph/build/stream_allocator.cc"
  585. "graph/build/logical_stream_allocator.cc"
  586. "graph/build/stream_graph_optimizer.cc"
  587. "graph/build/run_context.cc"
  588. "graph/build/label_allocator.cc"
  589. "graph/label/label_maker.cc"
  590. "graph/label/if_label_maker.cc"
  591. "graph/label/case_label_maker.cc"
  592. "graph/label/while_label_maker.cc"
  593. "graph/label/partitioned_call_label_maker.cc"
  594. "analyzer/analyzer.cc"
  595. )
  596. if (NOT ENABLE_D AND NOT ENABLE_ACL)
  597. ############ libge_runner.so ############
  598. add_library(ge_runner SHARED ${TRAIN_SRC_LIST} ${PROTO_SRCS} ${PROTO_CLIENT_SRCS})
  599. target_compile_definitions(ge_runner PRIVATE
  600. PROTOBUF_INLINE_NOT_IN_HEADERS=0
  601. DAVINCI_SUPPORT_PROFILING
  602. REUSE_MEMORY=1
  603. FMK_SUPPORT_DUMP
  604. DAVINCI_CLOUD
  605. google=ascend_private
  606. )
  607. target_compile_options(ge_runner PRIVATE
  608. -O2
  609. )
  610. target_include_directories(ge_runner PRIVATE
  611. ${GE_CODE_DIR}/ge
  612. ${GE_CODE_DIR}/ge/analyzer
  613. ${GE_CODE_DIR}/inc
  614. ${GE_CODE_DIR}/inc/external
  615. ${GE_CODE_DIR}/inc/framework
  616. ${GE_CODE_DIR}/inc/framework/common
  617. ${METADEF_DIR}
  618. ${METADEF_DIR}/inc
  619. ${METADEF_DIR}/inc/external/graph
  620. ${METADEF_DIR}/inc/external
  621. ${METADEF_DIR}/inc/graph
  622. ${CMAKE_BINARY_DIR}
  623. ${CMAKE_BINARY_DIR}/proto/ge
  624. #### yellow zone ####
  625. ${GE_CODE_DIR}/../inc
  626. ${GE_CODE_DIR}/../inc/external
  627. ${GE_CODE_DIR}/../inc/cce
  628. ${GE_CODE_DIR}/../toolchain/ide/ide-daemon/external
  629. #### blue zone
  630. ${ASCEND_DIR}/driver/include
  631. ${ASCEND_DIR}/fwkacllib/include
  632. ${GE_CODE_DIR}/third_party/fwkacllib/inc
  633. ${GE_CODE_DIR}/third_party/fwkacllib/inc/toolchain
  634. )
  635. target_link_libraries(ge_runner
  636. $<BUILD_INTERFACE:intf_pub>
  637. ge_memory
  638. adump_server
  639. msprofiler
  640. -Wl,--no-as-needed
  641. graph
  642. ge_common
  643. ascend_protobuf
  644. register
  645. c_sec
  646. slog
  647. mmpa
  648. msprof
  649. runtime
  650. resource
  651. error_manager
  652. ascend_hal_stub
  653. -Wl,--as-needed
  654. json
  655. -lrt
  656. -ldl
  657. )
  658. ############ libge_compiler.so ############
  659. add_library(ge_compiler SHARED ${INFER_SRC_LIST} ${PROTO_SRCS})
  660. target_compile_definitions(ge_compiler PRIVATE
  661. PROTOBUF_INLINE_NOT_IN_HEADERS=0
  662. REUSE_MEMORY=1
  663. FMK_SUPPORT_DUMP
  664. FMK_HOST_INFER
  665. COMPILE_OMG_PACKAGE
  666. google=ascend_private
  667. )
  668. target_compile_options(ge_compiler PRIVATE
  669. -O2
  670. )
  671. target_include_directories(ge_compiler PRIVATE
  672. ${GE_CODE_DIR}/ge
  673. ${GE_CODE_DIR}/ge/analyzer
  674. ${GE_CODE_DIR}/inc
  675. ${GE_CODE_DIR}/inc/external
  676. ${GE_CODE_DIR}/inc/framework
  677. ${GE_CODE_DIR}/inc/framework/common
  678. ${METADEF_DIR}
  679. ${METADEF_DIR}/inc
  680. ${METADEF_DIR}/inc/external/graph
  681. ${METADEF_DIR}/inc/external
  682. ${METADEF_DIR}/inc/graph
  683. ${CMAKE_BINARY_DIR}
  684. ${CMAKE_BINARY_DIR}/proto/ge
  685. #### yellow zone ####
  686. ${GE_CODE_DIR}/../inc
  687. ${GE_CODE_DIR}/../inc/external
  688. ${GE_CODE_DIR}/../inc/cce
  689. ${GE_CODE_DIR}/../toolchain/ide/ide-daemon/external
  690. #### blue zone ####
  691. ${ASCEND_DIR}/driver/include
  692. ${ASCEND_DIR}/fwkacllib/include
  693. ${GE_CODE_DIR}/third_party/fwkacllib/inc
  694. ${GE_CODE_DIR}/third_party/fwkacllib/inc/toolchain
  695. )
  696. target_link_libraries(ge_compiler
  697. $<BUILD_INTERFACE:intf_pub>
  698. ge_memory
  699. -Wl,--no-as-needed
  700. graph
  701. ge_common
  702. ascend_protobuf
  703. register
  704. c_sec
  705. error_manager
  706. slog
  707. mmpa
  708. runtime_compile
  709. resource
  710. -Wl,--as-needed
  711. json
  712. -lrt
  713. -ldl
  714. )
  715. ############ libascendcl.so ############
  716. file(GENERATE OUTPUT ${CMAKE_BINARY_DIR}/dummy.c CONTENT "")
  717. #add_library(dummy_obj OBJECT ${CMAKE_BINARY_DIR}/dummy.c)
  718. #set(DUMMY_OBJ $<TARGET_OBJECTS:dummy_obj>)
  719. file(MAKE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/ascendcl_object)
  720. if(EXISTS ${STATIC_ACL_LIB}/libascendcl.a)
  721. execute_process(
  722. COMMAND ar x ${STATIC_ACL_LIB}/libascendcl.a
  723. WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/ascendcl_object
  724. )
  725. file(GLOB OBJECT_LIST ${CMAKE_CURRENT_BINARY_DIR}/ascendcl_object/*.o)
  726. else()
  727. set(OBJECT_LIST ${CMAKE_BINARY_DIR}/dummy.c)
  728. endif()
  729. add_library(opensrc_ascendcl SHARED
  730. ${OBJECT_LIST}
  731. )
  732. target_compile_definitions(opensrc_ascendcl PRIVATE
  733. google=ascend_private
  734. )
  735. target_compile_options(opensrc_ascendcl PRIVATE
  736. -O2
  737. -fvisibility=hidden
  738. )
  739. target_link_options(opensrc_ascendcl PRIVATE
  740. -rdynamic
  741. -Wl,--allow-multiple-definition
  742. -Wl,-z,muldefs
  743. -Wl,-Bsymbolic
  744. -Wl,--exclude-libs,ALL
  745. )
  746. target_link_libraries(opensrc_ascendcl PRIVATE
  747. -Wl,--whole-archive
  748. ge_executor
  749. ge_common_static
  750. graph_static
  751. ascend_protobuf_static
  752. register_static
  753. error_manager_static
  754. adump_server
  755. msprofiler
  756. -Wl,--no-whole-archive
  757. -Wl,--no-as-needed
  758. c_sec
  759. runtime
  760. mmpa
  761. slog
  762. msprof
  763. ascend_hal_stub
  764. -Wl,--as-needed
  765. -ldl
  766. json
  767. )
  768. set_target_properties(opensrc_ascendcl PROPERTIES
  769. OUTPUT_NAME ascendcl
  770. )
  771. ##################################################################
  772. add_custom_command(
  773. OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/stub_ge_ir_build.cc
  774. ${CMAKE_CURRENT_BINARY_DIR}/stub_ge_api.cc
  775. ${CMAKE_CURRENT_BINARY_DIR}/stub_ge_prof.cc
  776. COMMAND echo "Generating stub files."
  777. && ${HI_PYTHON} ${CMAKE_CURRENT_LIST_DIR}/stub/gen_stubapi.py ${GE_CODE_DIR}/inc/external ${CMAKE_CURRENT_BINARY_DIR}
  778. && mv ge_ir_build.cc stub_ge_ir_build.cc
  779. && mv ge_api.cc stub_ge_api.cc
  780. && mv ge_prof.cc stub_ge_prof.cc
  781. && echo "Generating stub files end."
  782. #WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
  783. #DEPENDS stub/gen_stubapi.py ${TOP_DIR}/inc/external ${CMAKE_CURRENT_BINARY_DIR}
  784. )
  785. add_custom_target(ge_stub
  786. DEPENDS ${CMAKE_CURRENT_BINARY_DIR}/stub_ge_ir_build.cc
  787. ${CMAKE_CURRENT_BINARY_DIR}/stub_ge_api.cc
  788. ${CMAKE_CURRENT_BINARY_DIR}/stub_ge_prof.cc
  789. )
  790. ##################################################################
  791. ############ stub/libge_compiler.so ############
  792. add_library(atc_stub_ge_compiler SHARED
  793. stub_ge_ir_build.cc
  794. )
  795. add_dependencies(atc_stub_ge_compiler ge_stub)
  796. target_link_libraries(atc_stub_ge_compiler PRIVATE
  797. $<BUILD_INTERFACE:intf_pub>
  798. )
  799. set_target_properties(atc_stub_ge_compiler PROPERTIES
  800. OUTPUT_NAME ge_compiler
  801. LIBRARY_OUTPUT_DIRECTORY atc_stub
  802. )
  803. target_include_directories(atc_stub_ge_compiler PRIVATE
  804. ${GE_CODE_DIR}
  805. ${GE_CODE_DIR}/ge
  806. ${GE_CODE_DIR}/ge/analyzer
  807. ${GE_CODE_DIR}/inc
  808. ${GE_CODE_DIR}/inc/framework
  809. ${GE_CODE_DIR}/inc/framework/common
  810. ${GE_CODE_DIR}/inc/external
  811. ${METADEF_DIR}/inc/external
  812. ${METADEF_DIR}/inc/external/graph
  813. ${METADEF_DIR}/inc/graph
  814. #### yellow zone ####
  815. ${GE_CODE_DIR}/../inc/cce
  816. ${GE_CODE_DIR}/../toolchain/ide/ide-daemon/external
  817. #### blue zone ####
  818. ${ASCEND_DIR}/driver/include
  819. ${ASCEND_DIR}/fwkacllib/include
  820. )
  821. ############ stub/libge_runner.so ############
  822. add_library(fwk_stub_ge_runner SHARED
  823. stub_ge_api.cc
  824. stub_ge_prof.cc
  825. stub_ge_ir_build.cc
  826. )
  827. add_dependencies(fwk_stub_ge_runner ge_stub)
  828. target_link_libraries(fwk_stub_ge_runner PRIVATE
  829. $<BUILD_INTERFACE:intf_pub>
  830. )
  831. set_target_properties(fwk_stub_ge_runner PROPERTIES
  832. OUTPUT_NAME ge_runner
  833. LIBRARY_OUTPUT_DIRECTORY fwk_stub
  834. )
  835. target_include_directories(fwk_stub_ge_runner PRIVATE
  836. ${GE_CODE_DIR}
  837. ${GE_CODE_DIR}/ge
  838. ${GE_CODE_DIR}/ge/analyzer
  839. ${GE_CODE_DIR}/inc
  840. ${GE_CODE_DIR}/inc/external
  841. ${GE_CODE_DIR}/inc/framework
  842. ${GE_CODE_DIR}/inc/framework/common
  843. ${METADEF_DIR}/inc/external
  844. ${METADEF_DIR}/inc/external/graph
  845. ${METADEF_DIR}/inc/graph
  846. #### yellow zone ####
  847. ${GE_CODE_DIR}/../inc/cce
  848. ${GE_CODE_DIR}/../toolchain/ide/ide-daemon/external
  849. #### blue zone ####
  850. ${ASCEND_DIR}/driver/include
  851. ${ASCEND_DIR}/fwkacllib/include
  852. )
  853. ###############################################################
  854. add_custom_target(
  855. engine_conf.json ALL
  856. DEPENDS ${CMAKE_CURRENT_BINARY_DIR}/engine_conf.json
  857. )
  858. add_custom_command(
  859. OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/engine_conf.json
  860. COMMAND cp ${CMAKE_CURRENT_LIST_DIR}/engine_manager/engine_conf.json ${CMAKE_CURRENT_BINARY_DIR}/
  861. )
  862. ###############################################################
  863. add_custom_target(
  864. optimizer_priority.pbtxt ALL
  865. DEPENDS ${CMAKE_CURRENT_BINARY_DIR}/optimizer_priority.pbtxt
  866. )
  867. add_custom_command(
  868. OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/optimizer_priority.pbtxt
  869. COMMAND cp ${CMAKE_CURRENT_LIST_DIR}/opskernel_manager/optimizer_priority.pbtxt ${CMAKE_CURRENT_BINARY_DIR}/
  870. )
  871. ###############################################################
  872. ############ install ############
  873. set(INSTALL_BASE_DIR "")
  874. set(INSTALL_LIBRARY_DIR lib)
  875. install(TARGETS ge_runner ge_compiler opensrc_ascendcl OPTIONAL
  876. LIBRARY DESTINATION ${INSTALL_LIBRARY_DIR}
  877. )
  878. install(TARGETS atc_stub_ge_compiler fwk_stub_ge_runner OPTIONAL
  879. LIBRARY DESTINATION ${INSTALL_LIBRARY_DIR}/stub
  880. )
  881. install(FILES
  882. ${CMAKE_CURRENT_BINARY_DIR}/engine_conf.json
  883. ${CMAKE_CURRENT_BINARY_DIR}/optimizer_priority.pbtxt OPTIONAL
  884. DESTINATION ${INSTALL_LIBRARY_DIR}
  885. )
  886. endif()

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