@@ -621,6 +683,7 @@ class TracedModuleBuilder(NodeMixin):
self._is_builtin=module_tracer.is_builtin(mod)
self._argdef_graph_map={}
self._argdef_outdef_map={}
self.nodes=set()
# The builder will be passed to self._mod.forward as 'self' argument. If the 'forward' uses super().xxx to call method of its base classes, the trace procedure will throw exceprion, because the builder doesn't inherit from self._mod.__bases__.
# modify self.__class__ and let the builder inherit from TracedModuleBuilder and mod.__class__.
...
...
@@ -631,7 +694,7 @@ class TracedModuleBuilder(NodeMixin):