diff --git a/tf_adapter/interface_spec/api_npu_config.pyh b/tf_adapter/interface_spec/api_npu_config.pyh index 5a9978b91c8fc9a41e6994e7e882870e0fa54cc5..521ec8da0f0e45d05b30cdc10e1241c1e88be8af 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 61b1b13a3d29301b965c88fb22275d602328af7c..5781a313632837c55a17c2a137c054d39b67cbdf 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():