diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt index fa315516..594dd648 100755 --- 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" diff --git a/ge/ge_inference.mk b/ge/ge_inference.mk index ac106346..24bfd865 100755 --- 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 \ diff --git a/ge/ge_runner.mk b/ge/ge_runner.mk index 6c448a46..50faa2d8 100644 --- 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 \