diff --git a/colossalai/fx/tracer/_meta_trace.py b/colossalai/fx/tracer/_meta_trace.py index 48b3e2deb..181a28fe9 100644 --- a/colossalai/fx/tracer/_meta_trace.py +++ b/colossalai/fx/tracer/_meta_trace.py @@ -20,7 +20,7 @@ def meta_trace(module: torch.nn.Module, *args, **kwargs) -> Graph: >>> graph.print_tabular() """ graph = Graph() - namespace = _Namespace() + namespace = graph._graph_namespace class MetaProxy(torch.Tensor): """