Browse Source

fix(traced_module): fix trace_module function may raise error in finally scope

GitOrigin-RevId: 2689b1f6e6
release-1.10
Megvii Engine Team 3 years ago
parent
commit
5ba7663732
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      imperative/python/megengine/traced_module/traced_module.py

+ 3
- 1
imperative/python/megengine/traced_module/traced_module.py View File

@@ -34,7 +34,7 @@ from ..core._imperative_rt.core2 import (
set_module_tracing, set_module_tracing,
unset_module_tracing, unset_module_tracing,
) )
from ..core._trace_option import set_symbolic_shape
from ..core._trace_option import set_symbolic_shape, use_symbolic_shape
from ..core.ops.builtin import Copy from ..core.ops.builtin import Copy
from ..module import Module from ..module import Module
from ..module import external as MExternal from ..module import external as MExternal
@@ -2451,6 +2451,8 @@ def trace_module(
""" """
assert active_module_tracer() is None assert active_module_tracer() is None
assert isinstance(mod, Module) assert isinstance(mod, Module)
use_sym_shape = use_symbolic_shape()
inputs = []
try: try:
net_name = mod._name if mod._name else mod.__class__.__name__ net_name = mod._name if mod._name else mod.__class__.__name__
use_sym_shape = set_symbolic_shape(True) use_sym_shape = set_symbolic_shape(True)


Loading…
Cancel
Save