From 3da5621e9d091642e8192e35754e66ae4f5ae945 Mon Sep 17 00:00:00 2001
From: CLAY-panjw <1330286576@qq.com>
Date: Thu, 22 Sep 2022 11:01:40 +0800
Subject: [PATCH] dumpconfig

---
 tf_adapter/interface_spec/api_npu_config.pyh         |  4 ++--
 .../python/npu_bridge/estimator/npu/npu_config.py    | 12 ++++++------
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/tf_adapter/interface_spec/api_npu_config.pyh b/tf_adapter/interface_spec/api_npu_config.pyh
index 5a9978b91..521ec8da0 100644
--- a/tf_adapter/interface_spec/api_npu_config.pyh
+++ b/tf_adapter/interface_spec/api_npu_config.pyh
@@ -18,14 +18,14 @@ class NPURunConfig(run_config_lib.RunConfig):
                 distribute_config=None, modify_mixlist=None, op_precision_mode=None, device_type="default_device_type",
                 soc_config=None, hccl_timeout=None, op_wait_timeout=None, op_execute_timeout=None, HCCL_algorithm=None,
                 customize_dtypes=None, op_debug_config=None, memory_config=None, experimental_config=None,
-                jit_compile=True, topo_sorting_mode=None, dump_data="tensor", dump_layer=None):
+                jit_compile=True, topo_sorting_mode=None):
 
 class ProfilingConfig():
     def __init__(self, enable_profiling=False, profiling_options=None):
 
 class DumpConfig():
     def __init__(self, enable_dump=False, dump_path=None, dump_step=None,
-                dump_mode="output", enable_dump_debug=False, dump_debug_mode="all"):
+                dump_mode="output", enable_dump_debug=False, dump_debug_mode="all", dump_data="tensor", dump_layer=None):
 
 class DynamicInputConfig():
     def __init__(self, input_shape, dynamic_dims, dynamic_node_type):
diff --git a/tf_adapter/python/npu_bridge/estimator/npu/npu_config.py b/tf_adapter/python/npu_bridge/estimator/npu/npu_config.py
index 61b1b13a3..5781a3136 100644
--- a/tf_adapter/python/npu_bridge/estimator/npu/npu_config.py
+++ b/tf_adapter/python/npu_bridge/estimator/npu/npu_config.py
@@ -101,9 +101,7 @@ class NPURunConfig(run_config_lib.RunConfig):
                  memory_config=None,
                  experimental_config=None,
                  jit_compile=True,
-                 topo_sorting_mode=None,
-                 dump_data="tensor",
-                 dump_layer=None
+                 topo_sorting_mode=None
                  ):
         """
         Constructs a NPUConfig.
@@ -244,8 +242,6 @@ class NPURunConfig(run_config_lib.RunConfig):
         self._experimental_config = self._get_experimental_config(experimental_config)
         self._jit_compile = jit_compile
         self.topo_sorting_mode = topo_sorting_mode
-        self.dump_data = dump_data
-        self.dump_layer = dump_layer
 
 
         super(NPURunConfig, self).__init__(
@@ -344,7 +340,9 @@ class DumpConfig():
                  dump_step=None,
                  dump_mode="output",
                  enable_dump_debug=False,
-                 dump_debug_mode="all"):
+                 dump_debug_mode="all",
+                 dump_data="tensor",
+                 dump_layer=None):
         """
         Constructs a DumpConfig.
 
@@ -362,6 +360,8 @@ class DumpConfig():
         self._dump_mode = dump_mode
         self._enable_dump_debug = enable_dump_debug
         self._dump_debug_mode = dump_debug_mode
+        self.dump_data = dump_data
+        self.dump_layer = dump_layer
 
 
 class GraphMemoryOptimizeConfig():
-- 
Gitee