hummingbird
/
graphengine
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
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.
Tree: a2b6270dc4
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'a2b6270dc4'
${ noResults }
graphengine/ge/executor/CMakeLists.txt

269 lines
11 kB
Raw Blame History 

  1. set(PROTO_LIST

  2.     "${METADEF_DIR}/proto/om.proto"

  3.     "${METADEF_DIR}/proto/ge_ir.proto"

  4.     "${METADEF_DIR}/proto/insert_op.proto"

  5.     "${METADEF_DIR}/proto/task.proto"

  6.     "${METADEF_DIR}/proto/op_mapping_info.proto"

  7.     "${METADEF_DIR}/proto/dump_task.proto"

  8. )

  9. 

  10. protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})

  11. 

  12. set(SRC_LIST

  13.     "ge_executor.cc"

  14.     "../common/profiling/profiling_manager.cc"

  15.     "../common/ge/plugin_manager.cc"

  16.     "../common/ge/op_tiling_manager.cc"

  17.     "../common/dump/dump_properties.cc"

  18.     "../common/dump/dump_manager.cc"

  19.     "../common/dump/dump_op.cc"

  20.     "../common/profiling/ge_profiling.cc"

  21.     "../graph/load/graph_loader.cc"

  22.     "../graph/execute/graph_execute.cc"

  23.     "../omm/csa_interact.cc"

  24.     "../graph/manager/graph_manager_utils.cc"

  25.     "../graph/manager/graph_var_manager.cc"

  26.     "../graph/manager/graph_mem_allocator.cc"

  27.     "../graph/manager/graph_caching_allocator.cc"

  28.     "../graph/manager/trans_var_data_utils.cc"

  29.     "../graph/manager/util/debug.cc"

  30.     "../graph/manager/rdma_pool_allocator.cc"

  31.     "../hybrid/node_executor/aicpu/aicpu_ext_info.cc"

  32.     "../model/ge_model.cc"

  33.     "../model/ge_root_model.cc"

  34.     "../graph/load/new_model_manager/davinci_model.cc"

  35.     "../graph/load/new_model_manager/davinci_model_parser.cc"

  36.     "../graph/load/new_model_manager/model_manager.cc"

  37.     "../graph/load/new_model_manager/tbe_handle_store.cc"

  38.     "../graph/load/new_model_manager/cpu_queue_schedule.cc"

  39.     "../graph/load/new_model_manager/model_utils.cc"

  40.     "../graph/load/new_model_manager/aipp_utils.cc"

  41.     "../graph/load/new_model_manager/data_inputer.cc"

  42.     "../graph/load/new_model_manager/data_dumper.cc"

  43.     "../graph/load/new_model_manager/zero_copy_task.cc"

  44.     "../graph/load/new_model_manager/zero_copy_offset.cc"

  45.     "../graph/load/new_model_manager/task_info/task_info.cc"

  46.     "../graph/load/new_model_manager/task_info/event_record_task_info.cc"

  47.     "../graph/load/new_model_manager/task_info/event_wait_task_info.cc"

  48.     "../graph/load/new_model_manager/task_info/fusion_start_task_info.cc"

  49.     "../graph/load/new_model_manager/task_info/fusion_stop_task_info.cc"

  50.     "../graph/load/new_model_manager/task_info/kernel_ex_task_info.cc"

  51.     "../graph/load/new_model_manager/task_info/kernel_task_info.cc"

  52.     "../graph/load/new_model_manager/task_info/label_set_task_info.cc"

  53.     "../graph/load/new_model_manager/task_info/label_switch_by_index_task_info.cc"

  54.     "../graph/load/new_model_manager/task_info/label_goto_ex_task_info.cc"

  55.     "../graph/load/new_model_manager/task_info/memcpy_async_task_info.cc"

  56.     "../graph/load/new_model_manager/task_info/memcpy_addr_async_task_info.cc"

  57.     "../graph/load/new_model_manager/task_info/profiler_trace_task_info.cc"

  58.     "../graph/load/new_model_manager/task_info/stream_active_task_info.cc"

  59.     "../graph/load/new_model_manager/task_info/stream_switch_task_info.cc"

  60.     "../graph/load/new_model_manager/task_info/stream_switchn_task_info.cc"

  61.     "../graph/load/new_model_manager/task_info/end_graph_task_info.cc"

  62.     "../graph/load/new_model_manager/task_info/model_exit_task_info.cc"

  63.     "../graph/load/new_model_manager/task_info/super_kernel/super_kernel_factory.cc"

  64.     "../graph/load/new_model_manager/task_info/super_kernel/super_kernel.cc"

  65.     "../graph/common/local_context.cc"

  66.     "../opskernel_manager/ops_kernel_builder_manager.cc"

  67.     "../single_op/single_op_manager.cc"

  68.     "../single_op/single_op_model.cc"

  69.     "../single_op/single_op.cc"

  70.     "../single_op/stream_resource.cc"

  71.     "../single_op/task/op_task.cc"

  72.     "../single_op/task/build_task_utils.cc"

  73.     "../single_op/task/tbe_task_builder.cc"

  74.     "../single_op/task/aicpu_task_builder.cc"

  75.     "../single_op/task/aicpu_kernel_task_builder.cc"

  76.     "../hybrid/common/tensor_value.cc"

  77.     "../hybrid/common/npu_memory_allocator.cc"

  78.     "../hybrid/executor/rt_callback_manager.cc"

  79.     "../hybrid/executor/node_state.cc"

  80.     "../hybrid/executor/node_done_manager.cc"

  81.     "../hybrid/executor/hybrid_profiler.cc"

  82.     "../hybrid/executor/hybrid_model_executor.cc"

  83.     "../hybrid/executor/hybrid_model_async_executor.cc"

  84.     "../hybrid/executor/hybrid_execution_context.cc"

  85.     "../hybrid/executor/subgraph_context.cc"

  86.     "../hybrid/executor/subgraph_executor.cc"

  87.     "../hybrid/executor/worker/task_compile_engine.cc"

  88.     "../hybrid/executor/worker/shape_inference_engine.cc"

  89.     "../hybrid/executor/worker/execution_engine.cc"

  90.     "../hybrid/model/hybrid_model.cc"

  91.     "../hybrid/model/hybrid_model_builder.cc"

  92.     "../hybrid/model/node_item.cc"

  93.     "../hybrid/model/graph_item.cc"

  94.     "../hybrid/node_executor/aicore/aicore_node_executor.cc"

  95.     "../hybrid/node_executor/aicore/aicore_op_task.cc"

  96.     "../hybrid/node_executor/aicore/aicore_task_builder.cc"

  97.     "../hybrid/node_executor/aicpu/aicpu_node_executor.cc"

  98.     "../hybrid/node_executor/compiledsubgraph/known_node_executor.cc"

  99.     "../hybrid/node_executor/ge_local/ge_local_node_executor.cc"

  100.     "../hybrid/node_executor/host_cpu/host_cpu_node_executor.cc"

  101.     "../hybrid/node_executor/host_cpu/kernel_factory.cc"

  102.     "../hybrid/node_executor/host_cpu/kernel/no_op_kernel.cc"

  103.     "../hybrid/node_executor/host_cpu/kernel/variable_kernel.cc"

  104.     "../hybrid/node_executor/host_cpu/kernel/assign_kernel.cc"

  105.     "../hybrid/node_executor/host_cpu/kernel/random_uniform_kernel.cc"

  106.     "../hybrid/node_executor/controlop/control_op_executor.cc"

  107.     "../hybrid/node_executor/partitioned_call/partitioned_call_node_executor.cc"

  108.     "../hybrid/node_executor/rts/rts_node_executor.cc"

  109.     "../hybrid/node_executor/node_executor.cc"

  110.     "../hybrid/node_executor/task_context.cc"

  111.     "../hybrid/hybrid_davinci_model.cc"

  112.     "../ge_local_engine/engine/host_cpu_engine.cc"

  113.     "../graph/common/omg_util.cc"

  114.     "../graph/manager/host_mem_manager.cc"

  115.     "../graph/build/memory/var_mem_assign_util.cc"

  116.     "../host_kernels/transpose_kernel.cc"

  117.     "../host_kernels/add_kernel.cc"

  118.     "../host_kernels/broadcast_args_kernel.cc"

  119.     "../host_kernels/broadcast_gradient_args_kernel.cc"

  120.     "../host_kernels/cast_kernel.cc"

  121.     "../host_kernels/concat_offset_kernel.cc"

  122.     "../host_kernels/concat_v2_kernel.cc"

  123.     "../host_kernels/dynamic_stitch_kernel.cc"

  124.     "../host_kernels/identity_kernel.cc"

  125.     "../host_kernels/empty_kernel.cc"

  126.     "../host_kernels/expanddims_kernel.cc"

  127.     "../host_kernels/fill_kernel.cc"

  128.     "../host_kernels/floordiv_kernel.cc"

  129.     "../host_kernels/floormod_kernel.cc"

  130.     "../host_kernels/gather_v2_kernel.cc"

  131.     "../host_kernels/greater_kernel.cc"

  132.     "../host_kernels/kernel_utils.cc"

  133.     "../host_kernels/maximum_kernel.cc"

  134.     "../host_kernels/mul_kernel.cc"

  135.     "../host_kernels/pack_kernel.cc"

  136.     "../host_kernels/permute_kernel.cc"

  137.     "../host_kernels/range_kernel.cc"

  138.     "../host_kernels/rank_kernel.cc"

  139.     "../host_kernels/reduce_prod_kernel.cc"

  140.     "../host_kernels/reshape_kernel.cc"

  141.     "../host_kernels/rsqrt_kernel.cc"

  142.     "../host_kernels/shape_kernel.cc"

  143.     "../host_kernels/shape_n_kernel.cc"

  144.     "../host_kernels/size_kernel.cc"

  145.     "../host_kernels/slice_d_kernel.cc"

  146.     "../host_kernels/slice_kernel.cc"

  147.     "../host_kernels/squeeze_kernel.cc"

  148.     "../host_kernels/unsqueeze_kernel.cc"

  149.     "../host_kernels/ssd_prior_box_kernel.cc"

  150.     "../host_kernels/strided_slice_kernel.cc"

  151.     "../host_kernels/sub_kernel.cc"

  152.     "../host_kernels/transdata_kernel.cc"

  153.     "../host_kernels/unpack_kernel.cc"

  154.     "../graph/passes/pass_utils.cc"

  155.     "../graph/common/bcast.cc"

  156.     "../common/fp16_t.cc"

  157.     "../common/formats/format_transfers/format_transfer_transpose.cc"

  158.     "../common/formats/utils/formats_trans_utils.cc"

  159. )

  160. 

  161. ######## libge_executor.a ########

  162. add_library(ge_executor STATIC ${SRC_LIST} ${PROTO_HDRS})

  163. 

  164. target_compile_options(ge_executor PRIVATE

  165.     $<$<OR:$<STREQUAL:${TARGET_SYSTEM_NAME},Linux>,$<STREQUAL:${TARGET_SYSTEM_NAME},Android>>:-fvisibility=hidden -O2 -Werror -Wno-deprecated-declarations>

  166.     $<$<AND:$<STREQUAL:${TARGET_SYSTEM_NAME},Windows>,$<STREQUAL:${CMAKE_CONFIGURATION_TYPES},Debug>>:/MTd>

  167.     $<$<AND:$<STREQUAL:${TARGET_SYSTEM_NAME},Windows>,$<STREQUAL:${CMAKE_CONFIGURATION_TYPES},Release>>:/MT>

  168. )

  169. 

  170. target_compile_definitions(ge_executor PRIVATE

  171.     PROTOBUF_INLINE_NOT_IN_HEADERS=0

  172.     DAVINCI_SUPPORT_PROFILING

  173.     google=ascend_private

  174.     $<IF:$<STREQUAL:${TARGET_SYSTEM_NAME},Windows>,OS_TYPE=WIN,OS_TYPE=0>

  175.     $<$<STREQUAL:${TARGET_SYSTEM_NAME},Windows>:SECUREC_USING_STD_SECURE_LIB=0 NOMINMAX>

  176. )

  177. 

  178. target_include_directories(ge_executor PRIVATE

  179.     ${GE_CODE_DIR}/ge

  180.     ${GE_CODE_DIR}/inc

  181.     ${GE_CODE_DIR}/inc/external

  182.     ${GE_CODE_DIR}/inc/framework

  183.     ${METADEF_DIR}/inc

  184.     ${METADEF_DIR}/inc/external

  185.     ${METADEF_DIR}/inc/external/graph

  186.     ${METADEF_DIR}/inc/graph

  187.     ${CMAKE_BINARY_DIR}

  188.     ${CMAKE_BINARY_DIR}/proto/ge

  189.     #### yellow zone ####

  190.     ${GE_CODE_DIR}/../inc

  191.     ${GE_CODE_DIR}/../inc/cce

  192.     #### blue zone ####

  193.     ${GE_CODE_DIR}/third_party/fwkacllib/inc

  194. )

  195. 

  196. target_link_libraries(ge_executor PRIVATE

  197.     $<BUILD_INTERFACE:intf_pub>

  198.     json

  199.     ascend_protobuf

  200.     c_sec

  201.     $<$<NOT:$<STREQUAL:${TARGET_SYSTEM_NAME},Android>>:-lrt>

  202.     -ldl

  203. )

  204. 

  205. ######## libge_executor.so ########

  206. add_library(ge_executor_shared SHARED ${SRC_LIST} ${PROTO_HDRS})

  207. 

  208. target_compile_options(ge_executor_shared PRIVATE

  209.     -Werror

  210.     -O2

  211.     -Wno-deprecated-declarations

  212. )

  213. 

  214. target_compile_definitions(ge_executor_shared PRIVATE

  215.     PROTOBUF_INLINE_NOT_IN_HEADERS=0

  216.     DAVINCI_SUPPORT_PROFILING

  217.     google=ascend_private

  218. )

  219. 

  220. target_include_directories(ge_executor_shared PRIVATE

  221.     ${GE_CODE_DIR}/ge

  222.     ${GE_CODE_DIR}/inc

  223.     ${GE_CODE_DIR}/inc/external

  224.     ${GE_CODE_DIR}/inc/framework

  225.     ${METADEF_DIR}/inc

  226.     ${METADEF_DIR}/inc/external

  227.     ${METADEF_DIR}/inc/external/graph

  228.     ${METADEF_DIR}/inc/graph

  229.     ${CMAKE_BINARY_DIR}

  230.     ${CMAKE_BINARY_DIR}/proto/ge

  231.     #### yellow zone ####

  232.     ${GE_CODE_DIR}/../inc

  233.     ${GE_CODE_DIR}/../inc/cce

  234.     #### blue zone ####

  235.     ${GE_CODE_DIR}/third_party/fwkacllib/inc

  236. )

  237. 

  238. target_link_libraries(ge_executor_shared PRIVATE

  239.     $<BUILD_INTERFACE:intf_pub>

  240.     msprofiler

  241.     -Wl,--no-as-needed

  242.     ge_common

  243.     runtime

  244.     slog

  245.     mmpa

  246.     graph

  247.     register

  248.     error_manager

  249.     ascend_hal_stub

  250.     ascend_protobuf

  251.     c_sec

  252.     -Wl,--as-needed

  253.     json

  254.     $<$<NOT:$<STREQUAL:${TARGET_SYSTEM_NAME},Android>>:-lrt>

  255.     -ldl

  256. )

  257. 

  258. set_target_properties(ge_executor_shared PROPERTIES

  259.     OUTPUT_NAME ge_executor

  260. )

  261. 

  262. ############ install ############

  263. set(INSTALL_BASE_DIR "")

  264. set(INSTALL_LIBRARY_DIR lib)

  265. 

  266. install(TARGETS ge_executor_shared OPTIONAL

  267.     LIBRARY DESTINATION ${INSTALL_LIBRARY_DIR}

  268. )