From d8a5aded19c6564500d7bbb8c4091cd06e6da003 Mon Sep 17 00:00:00 2001
From: Super Daniel <78588128+super-dainiu@users.noreply.github.com>
Date: Tue, 6 Sep 2022 11:46:12 +0800
Subject: [PATCH] [hotfix] change namespace for meta_trace. (#1541)

---
 colossalai/fx/tracer/_meta_trace.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):
         """