fix(module): fix module dumped in old version without _short_name attr

GitOrigin-RevId: a1c815f613b34ddf7e4a4a2aeec41f353bdfb679

imperative/python/megengine/module/module.py

@@ -138,7 +138,11 @@ class Module(metaclass=ABCMeta):
         return HookHandler(self._forward_hooks, hook)
 
     def __call__(self, *inputs, **kwargs):
-        AutoNaming.push_scope(self.name if self.name is not None else self._short_name)
+        AutoNaming.push_scope(
+            self.name
+            if self.name is not None
+            else (self._short_name if hasattr(self, "_short_name") else self._name)
+        )
         for hook in self._forward_pre_hooks.values():
             modified_inputs = hook(self, inputs)
             if modified_inputs is not None:

imperative/python/megengine/traced_module/utils.py

@@ -222,7 +222,7 @@ class _ModuleList(Module, MutableSequence):
     def __setattr__(self, key, value):
         # clear mod name to avoid warning in Module's setattr
         if isinstance(value, Module):
-            value._name = None
+            value._short_name = None
         super().__setattr__(key, value)
 
     def __setitem__(self, idx: int, mod: Module):
@@ -291,7 +291,7 @@ class _ModuleDict(Module, MutableMapping):
     def __setattr__(self, key, value):
         # clear mod name to avoid warning in Module's setattr
         if isinstance(value, Module):
-            value._name = None
+            value._short_name = None
         super().__setattr__(key, value)
 
     def __setitem__(self, key, value):