split MergeToStreamMergePass, modify mk

4 years ago · e94a7010d9
parent 7bad9affa0
commit e94a7010d9
3 changed files with 4 additions and 0 deletions
--- a/ge/CMakeLists.txt
+++ b/ge/CMakeLists.txt
@ -237,6 +237,7 @@ set(TRAIN_SRC_LIST
    "graph/passes/switch_data_edges_bypass.cc"
    "graph/passes/switch_logic_remove_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/switch_dead_branch_elimination.cc"
@ -555,6 +556,7 @@ set(INFER_SRC_LIST
    "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"
--- a/ge/ge_inference.mk
+++ b/ge/ge_inference.mk
@ -176,6 +176,7 @@ OMG_HOST_SRC_FILES := \
    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 \
--- a/ge/ge_runner.mk
+++ b/ge/ge_runner.mk
@ -210,6 +210,7 @@ LIBGE_LOCAL_SRC_FILES := \
    graph/passes/switch_data_edges_bypass.cc \
    graph/passes/switch_logic_remove_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/switch_dead_branch_elimination.cc \