graphengine/ge/ge_inference.mk

LOCAL_PATH := $(call my-dir)
include $(LOCAL_PATH)/stub/Makefile
COMMON_LOCAL_SRC_FILES := \
    proto/fusion_model.proto \
    proto/optimizer_priority.proto \
    graph/manager/trans_var_data_utils.cc \
    common/fp16_t.cc \
    common/formats/utils/formats_trans_utils.cc \
    common/formats/format_transfers/datatype_transfer.cc \
    common/formats/format_transfers/format_transfer_transpose.cc \
    common/formats/format_transfers/format_transfer_nchw_nc1hwc0.cc \
    common/formats/format_transfers/format_transfer_fractal_z.cc \
    common/formats/format_transfers/format_transfer_fractal_nz.cc \
    common/formats/format_transfers/format_transfer_fractal_zz.cc \
    common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.cc \
    common/formats/format_transfers/format_transfer_nc1hwc0_nchw.cc \
    common/formats/format_transfers/format_transfer_nc1hwc0_nhwc.cc \
    common/formats/format_transfers/format_transfer_hwcn_c1hwncoc0.cc \
    common/formats/format_transfers/format_transfer_c1hwncoc0_hwcn.cc \
    common/formats/format_transfers/format_transfer_fracz_nchw.cc \
    common/formats/format_transfers/format_transfer_fracz_nhwc.cc \
    common/formats/format_transfers/format_transfer_fracz_hwcn.cc \
    common/formats/format_transfers/format_transfer_dhwcn_fracz3D.cc \
    common/formats/format_transfers/format_transfer_dhwnc_fracz3D_transpose.cc \
    common/formats/format_transfers/format_transfer_nchw_fz_c04.cc \
    common/formats/formats.cc \
    common/profiling/profiling_manager.cc \
    common/dump/dump_properties.cc \
    common/dump/dump_manager.cc \
    common/dump/dump_op.cc \
    common/dump/dump_server.cc \
    common/helper/model_cache_helper.cc \
    ge_local_engine/engine/host_cpu_engine.cc \


GRAPH_MANAGER_LOCAL_SRC_FILES := \
    common/ge/plugin_manager.cc\
    common/ge/op_tiling_manager.cc\
    init/gelib.cc \
    session/inner_session.cc \
    session/session_manager.cc \
    engine_manager/dnnengine_manager.cc \
    opskernel_manager/ops_kernel_manager.cc \
    opskernel_manager/ops_kernel_builder_manager.cc \
    graph/manager/graph_manager.cc \
    graph/manager/graph_manager_utils.cc \
    graph/manager/graph_context.cc \
    graph/preprocess/graph_preprocess.cc \
    graph/preprocess/multi_batch_options.cc \
    graph/preprocess/multi_batch_copy_graph.cc \
    graph/execute/graph_execute.cc \
    graph/load/graph_loader.cc \
    graph/optimize/graph_optimize.cc \
    graph/optimize/mem_rw_conflict_optimize.cc \
    graph/optimize/summary_optimize.cc \
    graph/build/graph_builder.cc \
    graph/partition/engine_place.cc \
    graph/partition/graph_partition.cc \
    graph/partition/dynamic_shape_partition.cc \
    graph/partition/stage_partition.cc \
    generator/ge_generator.cc \
    generator/generator_api.cc \
    graph/manager/graph_var_manager.cc \
    graph/manager/host_mem_manager.cc \
    graph/manager/rdma_pool_allocator.cc \
    graph/manager/host_mem_allocator.cc \
    graph/manager/graph_mem_allocator.cc \
    graph/manager/graph_caching_allocator.cc \

BUILER_SRC_FILES := \
    ir_build/ge_ir_build.cc \
    ir_build/attr_options/utils.cc \
    ir_build/attr_options/keep_dtype_option.cc \
    ir_build/attr_options/weight_compress_option.cc \
    ir_build/option_utils.cc \

ANALYZER_SRC_FILES:= \
    analyzer/analyzer.cc \

OMG_HOST_SRC_FILES := \
    model/ge_model.cc \
    model/ge_root_model.cc \
    graph/common/transop_util.cc \
    graph/passes/pass_manager.cc \
    graph/passes/resource_pair_add_control_pass.cc \
    graph/passes/resource_pair_remove_control_pass.cc \
    graph/passes/pass_utils.cc \
    graph/passes/base_pass.cc \
    graph/passes/bitcast_pass.cc \
    graph/passes/constant_folding_pass.cc \
    graph/passes/aicpu_constant_folding_pass.cc \
    graph/passes/reshape_remove_pass.cc \
    graph/passes/reshape_recovery_pass.cc \
    graph/passes/transop_breadth_fusion_pass.cc \
    graph/passes/transop_depth_fusion_pass.cc \
    graph/passes/transop_nearby_allreduce_fusion_pass.cc \
    graph/passes/same_transdata_breadth_fusion_pass.cc \
    graph/passes/transop_without_reshape_fusion_pass.cc \
    graph/passes/compile_nodes_pass.cc \
    graph/passes/variable_prepare_op_pass.cc \
    graph/passes/variable_ref_delete_op_pass.cc \
    graph/passes/variable_ref_useless_control_out_delete_pass.cc \
    graph/passes/subgraph_pass.cc \
    graph/passes/data_pass.cc \
    graph/passes/net_output_pass.cc \
    graph/passes/replace_transshape_pass.cc \
    graph/passes/constant_fuse_same_pass.cc \
    graph/passes/fuse_data_nodes_with_common_input_pass.cc \
    graph/passes/print_op_pass.cc \
    graph/passes/no_use_reshape_remove_pass.cc \
    graph/passes/iterator_op_pass.cc \
    graph/passes/input_output_connection_identify_pass.cc \
    graph/passes/atomic_addr_clean_pass.cc \
    graph/passes/mark_same_addr_pass.cc \
    graph/passes/mark_graph_unknown_status_pass.cc \
    graph/passes/mark_node_unknown_shape_pass.cc \
    graph/passes/mark_agnostic_pass.cc \
    graph/common/omg_util.cc \
    graph/common/bcast.cc \
    graph/common/local_context.cc \
    graph/passes/dimension_compute_pass.cc \
    graph/passes/dimension_adjust_pass.cc \
    graph/passes/get_original_format_pass.cc \
    graph/passes/shape_operate_op_remove_pass.cc \
    graph/passes/assert_pass.cc \
    graph/passes/dropout_pass.cc \
    graph/passes/infershape_pass.cc \
    graph/passes/unused_const_pass.cc \
    graph/passes/permute_pass.cc \
    graph/passes/ctrl_edge_transfer_pass.cc \
    graph/passes/end_of_sequence_add_control_pass.cc \
    host_kernels/broadcast_gradient_args_kernel.cc \
    host_kernels/greater_kernel.cc \
    host_kernels/gather_v2_kernel.cc  \
    host_kernels/maximum_kernel.cc \
    host_kernels/floormod_kernel.cc \
    host_kernels/floordiv_kernel.cc \
    host_kernels/range_kernel.cc \
    host_kernels/shape_kernel.cc \
    host_kernels/size_kernel.cc \
    host_kernels/shape_n_kernel.cc \
    host_kernels/rank_kernel.cc \
    host_kernels/broadcast_args_kernel.cc \
    host_kernels/fill_kernel.cc \
    host_kernels/empty_kernel.cc \
    host_kernels/expanddims_kernel.cc \
    host_kernels/reshape_kernel.cc \
    host_kernels/squeeze_kernel.cc \
    host_kernels/unsqueeze_kernel.cc \
    host_kernels/kernel_utils.cc \
    host_kernels/cast_kernel.cc \
    host_kernels/transdata_kernel.cc \
    host_kernels/unpack_kernel.cc \
    host_kernels/transpose_kernel.cc \
    host_kernels/permute_kernel.cc \
    host_kernels/pack_kernel.cc \
    host_kernels/concat_v2_kernel.cc \
    host_kernels/concat_offset_kernel.cc \
    host_kernels/strided_slice_kernel.cc \
    host_kernels/ssd_prior_box_kernel.cc \
    host_kernels/add_kernel.cc \
    host_kernels/sub_kernel.cc \
    host_kernels/mul_kernel.cc \
    host_kernels/reduce_prod_kernel.cc \
    host_kernels/rsqrt_kernel.cc \
    host_kernels/slice_kernel.cc \
    host_kernels/slice_d_kernel.cc \
    host_kernels/dynamic_stitch_kernel.cc \
    host_kernels/identity_kernel.cc \
    host_kernels/reformat_kernel.cc \
    graph/passes/stop_gradient_pass.cc \
    graph/passes/prevent_gradient_pass.cc \
    graph/passes/identity_pass.cc \
    graph/passes/ref_identity_delete_op_pass.cc \
    graph/passes/placeholder_with_default_pass.cc \
    graph/passes/snapshot_pass.cc \
    graph/passes/guarantee_const_pass.cc \
    graph/passes/var_is_initialized_op_pass.cc \
    graph/passes/parallel_concat_start_op_pass.cc \
    graph/passes/folding_pass.cc \
    graph/passes/cast_translate_pass.cc \
    graph/passes/prune_pass.cc \
    graph/passes/merge_to_stream_merge_pass.cc \
    graph/passes/merge_input_memcpy_pass.cc \
    graph/passes/switch_to_stream_switch_pass.cc \
    graph/passes/attach_stream_label_pass.cc \
    graph/passes/multi_batch_pass.cc \
    graph/passes/multi_batch_clone_pass.cc \
    graph/passes/subexpression_migration_pass.cc \
    graph/passes/subgraph_const_migration_pass.cc \
    graph/passes/unused_args_clean_pass.cc \
    graph/passes/next_iteration_pass.cc \
    graph/passes/control_trigger_pass.cc \
    graph/passes/cond_pass.cc \
    graph/passes/cond_remove_pass.cc \
    graph/passes/remove_same_const_pass.cc \
    graph/passes/useless_control_out_remove_pass.cc \
    graph/passes/for_pass.cc \
    graph/passes/enter_pass.cc \
    graph/passes/assign_remove_pass.cc \
    graph/passes/inplace_support_check_pass.cc \
    graph/passes/addn_pass.cc \
    graph/passes/common_subexpression_elimination_pass.cc \
    graph/passes/transop_symmetry_elimination_pass.cc \
    graph/passes/save_pass.cc \
    graph/passes/switch_dead_branch_elimination.cc \
    graph/passes/switch_logic_remove_pass.cc \
    graph/passes/switch_data_edges_bypass.cc \
    graph/passes/merge_pass.cc \
    graph/passes/variable_op_pass.cc \
    graph/passes/cast_remove_pass.cc \
    graph/passes/transpose_transdata_pass.cc \
    graph/passes/hccl_memcpy_pass.cc \
    graph/passes/hccl_continuous_memcpy_pass.cc \
    graph/passes/flow_ctrl_pass.cc \
    graph/passes/global_step_insert_pass.cc \
    graph/passes/link_gen_mask_nodes_pass.cc \
    graph/passes/replace_with_empty_const_pass.cc \
    graph/passes/hccl_group_pass.cc \
    graph/passes/memcpy_addr_async_pass.cc \
    graph/passes/set_input_output_offset_pass.cc \
    graph/passes/buffer_pool_memory_pass.cc \

OMG_DEVICE_SRC_FILES := $(OMG_HOST_SRC_FILES)


OME_HOST_SRC_FILES := \
    graph/manager/model_manager/event_manager.cc        \
    graph/manager/util/rt_context_util.cc               \
    graph/manager/util/variable_accelerate_ctrl.cc       \
    graph/manager/util/debug.cc  \
    graph/load/model_manager/model_manager.cc                        \
    graph/load/model_manager/data_inputer.cc                         \
    graph/load/model_manager/davinci_model.cc                        \
    graph/load/model_manager/davinci_model_parser.cc                 \
    graph/load/model_manager/model_utils.cc                          \
    graph/load/model_manager/aipp_utils.cc                           \
    graph/load/model_manager/tbe_handle_store.cc                     \
    graph/load/model_manager/cpu_queue_schedule.cc                   \
    graph/load/model_manager/zero_copy_task.cc                       \
    graph/load/model_manager/zero_copy_offset.cc                     \
    graph/load/model_manager/data_dumper.cc                          \
    graph/load/model_manager/task_info/task_info.cc                  \
    graph/load/model_manager/task_info/event_record_task_info.cc     \
    graph/load/model_manager/task_info/event_wait_task_info.cc       \
    graph/load/model_manager/task_info/fusion_start_task_info.cc     \
    graph/load/model_manager/task_info/fusion_stop_task_info.cc      \
    graph/load/model_manager/task_info/kernel_ex_task_info.cc        \
    graph/load/model_manager/task_info/kernel_task_info.cc           \
    graph/load/model_manager/task_info/label_set_task_info.cc        \
    graph/load/model_manager/task_info/label_switch_by_index_task_info.cc \
    graph/load/model_manager/task_info/label_goto_ex_task_info.cc    \
    graph/load/model_manager/task_info/memcpy_async_task_info.cc     \
    graph/load/model_manager/task_info/memcpy_addr_async_task_info.cc \
    graph/load/model_manager/task_info/profiler_trace_task_info.cc   \
    graph/load/model_manager/task_info/stream_active_task_info.cc    \
    graph/load/model_manager/task_info/stream_switch_task_info.cc    \
    graph/load/model_manager/task_info/stream_switchn_task_info.cc   \
    graph/load/model_manager/task_info/end_graph_task_info.cc        \
    graph/load/model_manager/task_info/model_exit_task_info.cc       \
    graph/load/model_manager/task_info/super_kernel/super_kernel_factory.cc   \
    graph/load/model_manager/task_info/super_kernel/super_kernel.cc  \
    single_op/task/op_task.cc                                            \
    single_op/task/build_task_utils.cc                                   \
    single_op/task/tbe_task_builder.cc                                   \
    single_op/task/aicpu_task_builder.cc                                 \
    single_op/task/aicpu_kernel_task_builder.cc                          \
    single_op/single_op.cc                                               \
    single_op/single_op_model.cc                                         \
    single_op/stream_resource.cc                                         \
    single_op/single_op_manager.cc                                       \
    hybrid/hybrid_davinci_model_stub.cc                                  \
    hybrid/node_executor/aicpu/aicpu_ext_info.cc                         \
    # graph/load/model_manager/task_info/hccl_task_info.cc

OME_DEVICE_SRC_FILES := $(OME_HOST_SRC_FILES)

COMMON_LOCAL_C_INCLUDES := \
    proto/om.proto \
    proto/task.proto \
    proto/insert_op.proto \
    proto/ge_ir.proto \
    proto/fwk_adapter.proto    \
    proto/op_mapping_info.proto \
    proto/dump_task.proto       \
    proto/tensorflow/attr_value.proto \
    proto/tensorflow/function.proto \
    proto/tensorflow/graph.proto \
    proto/tensorflow/node_def.proto \
    proto/tensorflow/op_def.proto \
    proto/tensorflow/resource_handle.proto \
    proto/tensorflow/tensor.proto \
    proto/tensorflow/tensor_shape.proto \
    proto/tensorflow/types.proto \
    proto/tensorflow/versions.proto \
    $(LOCAL_PATH) ./ \
    $(TOPDIR)inc \
    $(TOPDIR)metadef/inc \
    $(TOPDIR)graphengine/inc \
    $(TOPDIR)inc/external \
    $(TOPDIR)metadef/inc/external \
    $(TOPDIR)graphengine/inc/external \
    $(TOPDIR)metadef/inc/external/graph \
    $(TOPDIR)graphengine/inc/framework \
    $(TOPDIR)graphengine/inc/framework/common \
    $(TOPDIR)metadef/inc/common \
    $(TOPDIR)inc/runtime \
    $(TOPDIR)libc_sec/include \
    $(TOPDIR)ops/built-in/op_proto/inc \
    $(TOPDIR)toolchain/ide/ide-daemon/external \
    third_party/json/include \
    third_party/protobuf/include \
    third_party/opencv/include \

ANALYZER_LOCAL_INCLUDES := \
    $(TOPDIR)graphengine/ge/analyzer \

NEW_OMG_HOST_SRC_FILES := \
    graph/preprocess/insert_op/util_insert_aipp_op.cc \
    graph/preprocess/insert_op/ge_aipp_op.cc \
    graph/build/model_builder.cc \
    graph/build/task_generator.cc \
    graph/build/stream_allocator.cc \
    graph/build/logical_stream_allocator.cc \
    graph/build/stream_graph_optimizer.cc \
    graph/build/run_context.cc \
    graph/build/label_allocator.cc \
    graph/label/label_maker.cc \
    graph/label/if_label_maker.cc \
    graph/label/case_label_maker.cc \
    graph/label/while_label_maker.cc \
    graph/label/partitioned_call_label_maker.cc \

OME_HOST_SRC_FILES += $(NEW_OMG_HOST_SRC_FILES)
OMG_DEVICE_SRC_FILES += $(NEW_OMG_HOST_SRC_FILES)

DEVICE_LOCAL_C_INCLUDES := \
    proto/om.proto \
    proto/task.proto \
    proto/insert_op.proto \
    proto/ge_ir.proto \
    proto/fwk_adapter.proto    \
    proto/op_mapping_info.proto \
    proto/tensorflow/attr_value.proto \
    proto/tensorflow/function.proto \
    proto/tensorflow/graph.proto \
    proto/tensorflow/node_def.proto \
    proto/tensorflow/op_def.proto \
    proto/tensorflow/resource_handle.proto \
    proto/tensorflow/tensor.proto \
    proto/tensorflow/tensor_shape.proto \
    proto/tensorflow/types.proto \
    proto/tensorflow/versions.proto \
    $(LOCAL_PATH) ./ \
    $(TOPDIR)inc \
    $(TOPDIR)metadef/inc \
    $(TOPDIR)graphengine/inc \
    $(TOPDIR)libc_sec/include \
    $(TOPDIR)inc/external \
    $(TOPDIR)metadef/inc/external \
    $(TOPDIR)graphengine/inc/external \
    $(TOPDIR)metadef/inc/external/graph \
    $(TOPDIR)metadef/inc/common/util \
    $(TOPDIR)graphengine/inc/framework \
    $(TOPDIR)graphengine/inc/framework/common \
    $(TOPDIR)inc/runtime \
    $(TOPDIR)ops/built-in/op_proto/inc \
    $(TOPDIR)graphengine/ge \
    $(TOPDIR)toolchain/ide/ide-daemon/external \
    third_party/json/include \
    third_party/protobuf/include \
    third_party/opencv/include \

#compiler for host infer
include $(CLEAR_VARS)

LOCAL_MODULE := libge_compiler

LOCAL_CFLAGS += -DPROTOBUF_INLINE_NOT_IN_HEADERS=0 -DREUSE_MEMORY=1 -O2
# from ome_inference.mk
LOCAL_CFLAGS += -DFMK_HOST_INFER -DFMK_SUPPORT_DUMP -DCOMPILE_OMG_PACKAGE -Dgoogle=ascend_private
ifeq ($(DEBUG), 1)
LOCAL_CFLAGS += -g -O0
endif

LOCAL_C_INCLUDES := $(COMMON_LOCAL_C_INCLUDES)
LOCAL_C_INCLUDES += $(ANALYZER_LOCAL_INCLUDES)

LOCAL_SRC_FILES := $(COMMON_LOCAL_SRC_FILES)
LOCAL_SRC_FILES += $(GRAPH_MANAGER_LOCAL_SRC_FILES)
LOCAL_SRC_FILES += $(OMG_HOST_SRC_FILES)
LOCAL_SRC_FILES += $(OME_HOST_SRC_FILES)
LOCAL_SRC_FILES += $(NEW_OME_DEVICE_SRC_FILES)
LOCAL_SRC_FILES += $(BUILER_SRC_FILES)
LOCAL_SRC_FILES += $(ANALYZER_SRC_FILES)

LOCAL_STATIC_LIBRARIES := libge_memory \
                          libmmpa \

LOCAL_SHARED_LIBRARIES := \
    libc_sec \
    libascend_protobuf \
    libslog \
    libgraph \
    libregister \
    libge_common \
    libruntime_compile \
    libresource \
    liberror_manager \

LOCAL_LDFLAGS := -lrt -ldl


include $(BUILD_HOST_SHARED_LIBRARY)

#compiler for host infer
include $(CLEAR_VARS)

LOCAL_MODULE := stub/libge_compiler

LOCAL_CFLAGS += -DPROTOBUF_INLINE_NOT_IN_HEADERS=0 -DREUSE_MEMORY=1 -O2
LOCAL_CFLAGS += -DFMK_HOST_INFER -DFMK_SUPPORT_DUMP
ifeq ($(DEBUG), 1)
LOCAL_CFLAGS += -g -O0
endif

LOCAL_C_INCLUDES := $(COMMON_LOCAL_C_INCLUDES)

LOCAL_SRC_FILES := ../../out/ge/lib64/stub/ge_ir_build.cc


LOCAL_SHARED_LIBRARIES :=

LOCAL_LDFLAGS := -lrt -ldl

include $(BUILD_HOST_SHARED_LIBRARY)

#compiler for device
include $(CLEAR_VARS)

LOCAL_MODULE := libge_compiler
LOCAL_CFLAGS += -DGOOGLE_PROTOBUF_NO_RTTI -DDEV_VISIBILITY -DNONSUPPORT_SAVE_TO_FILE
LOCAL_CFLAGS += -DPROTOBUF_INLINE_NOT_IN_HEADERS=0
LOCAL_CFLAGS += -DREUSE_MEMORY=1 -DFMK_SUPPORT_DUMP -DCOMPILE_OMG_PACKAGE
LOCAL_CFLAGS += -DOMG_DEVICE_VERSION
LOCAL_CFLAGS += -O2 -Dgoogle=ascend_private
LOCAL_MODULE_CLASS := SHARED_LIBRARIES


LOCAL_SRC_FILES := $(COMMON_LOCAL_SRC_FILES)
LOCAL_SRC_FILES += $(GRAPH_MANAGER_LOCAL_SRC_FILES)
LOCAL_SRC_FILES += $(OMG_DEVICE_SRC_FILES)
LOCAL_SRC_FILES += $(OME_DEVICE_SRC_FILES)
LOCAL_SRC_FILES += $(BUILER_SRC_FILES)
LOCAL_SRC_FILES += $(ANALYZER_SRC_FILES)


LOCAL_C_INCLUDES := $(DEVICE_LOCAL_C_INCLUDES)
LOCAL_C_INCLUDES += $(ANALYZER_LOCAL_INCLUDES)

LOCAL_STATIC_LIBRARIES := libge_memory \
                          libmmpa \

LOCAL_SHARED_LIBRARIES := \
    libc_sec \
    libascend_protobuf \
    libslog \
    libgraph \
    libregister \
    libresource \
    libruntime_compile  \
    libge_common \




ifeq ($(device_os),android)
LOCAL_LDFLAGS := -ldl
else
LOCAL_LDFLAGS := -lrt -ldl
endif

LOCAL_CFLAGS += \
    -Wall

ifeq ($(device_os),android)
LOCAL_LDLIBS += -L$(PWD)/prebuilts/clang/linux-x86/aarch64/android-ndk-r21/sysroot/usr/lib/aarch64-linux-android/29 -llog
endif
include $(BUILD_SHARED_LIBRARY)