diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt index a541adc7..f0550f76 100755 --- a/ge/CMakeLists.txt +++ b/ge/CMakeLists.txt @@ -125,7 +125,7 @@ set(EXECUTOR_SRC_LIST "graph/load/graph_loader.cc" "graph/load/model_manager/aipp_utils.cc" "graph/load/model_manager/cpu_queue_schedule.cc" - "graph/load/model_manager/data_dumper.cc" + "common/dump/data_dumper.cc" "graph/load/model_manager/data_inputer.cc" "graph/load/model_manager/davinci_model.cc" "graph/load/model_manager/model_manager.cc" diff --git a/ge/graph/load/model_manager/data_dumper.cc b/ge/common/dump/data_dumper.cc similarity index 99% rename from ge/graph/load/model_manager/data_dumper.cc rename to ge/common/dump/data_dumper.cc index 7b5d9df9..bb5b8a3f 100644 --- a/ge/graph/load/model_manager/data_dumper.cc +++ b/ge/common/dump/data_dumper.cc @@ -14,7 +14,7 @@ * limitations under the License. */ -#include "graph/load/model_manager/data_dumper.h" +#include "data_dumper.h" #include #include diff --git a/ge/graph/load/model_manager/data_dumper.h b/ge/common/dump/data_dumper.h similarity index 100% rename from ge/graph/load/model_manager/data_dumper.h rename to ge/common/dump/data_dumper.h diff --git a/ge/common/dump/opdebug_register.h b/ge/common/dump/opdebug_register.h index 5b927b67..fab1e744 100644 --- a/ge/common/dump/opdebug_register.h +++ b/ge/common/dump/opdebug_register.h @@ -20,7 +20,7 @@ #include #include "framework/common/debug/ge_log.h" #include "framework/common/debug/log.h" -#include "graph/load/model_manager/data_dumper.h" +#include "data_dumper.h" namespace ge { class OpdebugRegister { diff --git a/ge/executor/CMakeLists.txt b/ge/executor/CMakeLists.txt index f8660d43..277cd3bb 100755 --- a/ge/executor/CMakeLists.txt +++ b/ge/executor/CMakeLists.txt @@ -26,7 +26,7 @@ set(SRC_LIST "../graph/load/model_manager/model_utils.cc" "../graph/load/model_manager/aipp_utils.cc" "../graph/load/model_manager/data_inputer.cc" - "../graph/load/model_manager/data_dumper.cc" + "../common/dump/data_dumper.cc" "../graph/load/model_manager/zero_copy_task.cc" "../graph/load/model_manager/zero_copy_offset.cc" "../graph/load/model_manager/task_info/task_info.cc" diff --git a/ge/graph/load/model_manager/davinci_model.h b/ge/graph/load/model_manager/davinci_model.h index 76b0beef..5d980276 100755 --- a/ge/graph/load/model_manager/davinci_model.h +++ b/ge/graph/load/model_manager/davinci_model.h @@ -35,7 +35,7 @@ #include "framework/common/util.h" #include "graph/debug/ge_attr_define.h" #include "graph/load/model_manager/aipp_utils.h" -#include "graph/load/model_manager/data_dumper.h" +#include "common/dump/data_dumper.h" #include "graph/load/model_manager/data_inputer.h" #include "graph/load/model_manager/model_utils.h" #include "graph/load/model_manager/zero_copy_offset.h" diff --git a/ge/hybrid/executor/hybrid_model_async_executor.h b/ge/hybrid/executor/hybrid_model_async_executor.h index f94f6aa5..801679cc 100644 --- a/ge/hybrid/executor/hybrid_model_async_executor.h +++ b/ge/hybrid/executor/hybrid_model_async_executor.h @@ -23,7 +23,7 @@ #include "external/ge/ge_api_types.h" #include "common/dump/opdebug_register.h" #include "graph/load/model_manager/data_inputer.h" -#include "graph/load/model_manager/data_dumper.h" +#include "common/dump/data_dumper.h" #include "hybrid/executor/hybrid_model_executor.h" #include "hybrid/executor/hybrid_model_pipeline_executor.h" #include "runtime/stream.h" diff --git a/tests/ut/ge/CMakeLists.txt b/tests/ut/ge/CMakeLists.txt index a189763a..68395c1f 100755 --- a/tests/ut/ge/CMakeLists.txt +++ b/tests/ut/ge/CMakeLists.txt @@ -221,7 +221,7 @@ set(GRAPH_PREPARE_COMMON_SRC_FILES set(GRAPH_DAVINCI_MODEL_SRC_FILES "${GE_CODE_DIR}/ge/graph/load/model_manager/aipp_utils.cc" "${GE_CODE_DIR}/ge/graph/load/model_manager/cpu_queue_schedule.cc" - "${GE_CODE_DIR}/ge/graph/load/model_manager/data_dumper.cc" + "${GE_CODE_DIR}/ge/common/dump/data_dumper.cc" "${GE_CODE_DIR}/ge/graph/load/model_manager/data_inputer.cc" "${GE_CODE_DIR}/ge/graph/load/model_manager/davinci_model.cc" "${GE_CODE_DIR}/ge/graph/load/model_manager/model_manager.cc" diff --git a/tests/ut/ge/graph/load/data_dumper_unittest.cc b/tests/ut/ge/graph/load/data_dumper_unittest.cc index 1eb9eb49..85dc0774 100644 --- a/tests/ut/ge/graph/load/data_dumper_unittest.cc +++ b/tests/ut/ge/graph/load/data_dumper_unittest.cc @@ -18,7 +18,7 @@ #define private public #define protected public -#include "graph/load/model_manager/data_dumper.h" +#include "common/dump/data_dumper.h" #include "graph/load/model_manager/davinci_model.h" #undef private #undef protected