diff --git a/debug/accuracy_tools/msprobe/mindspore/dump/jit_dump.py b/debug/accuracy_tools/msprobe/mindspore/dump/jit_dump.py
index ae50719ffe381ed79c7fb7a90c50046645df1b0b..fd564cda7a656c0eaffd28e6069a22a8f0c22034 100644
--- a/debug/accuracy_tools/msprobe/mindspore/dump/jit_dump.py
+++ b/debug/accuracy_tools/msprobe/mindspore/dump/jit_dump.py
@@ -20,7 +20,7 @@ from mindspore._c_expression import PyNativeExecutor_
 from mindspore.common.api import _MindsporeFunctionExecutor
 
 from msprobe.core.common.const import Const
-from msprobe.core.data_dump.data_processor.base import ModuleForwardInputsOutputs
+from msprobe.core.data_dump.data_processor.base import ModuleForwardInputsOutputs, ModuleBackwardInputsOutputs
 from msprobe.mindspore.dump.hook_cell.api_registry import api_register
 
 
@@ -46,6 +46,7 @@ class JitDump(_MindsporeFunctionExecutor):
     dump_config = None
     jit_enable = False
     jit_dump_switch = True
+    jit_count = 0
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)