From 06fb68f2c3be138e07f81e4786099a3ce5c0b9f9 Mon Sep 17 00:00:00 2001 From: Henry Date: Fri, 11 Oct 2024 11:37:43 +0800 Subject: [PATCH] s --- debug/accuracy_tools/msprobe/mindspore/dump/jit_dump.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/debug/accuracy_tools/msprobe/mindspore/dump/jit_dump.py b/debug/accuracy_tools/msprobe/mindspore/dump/jit_dump.py index ae50719ff..fd564cda7 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) -- Gitee