github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 03:52:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] change namespace for meta_trace. (#1541)

Browse Source
This commit is contained in:
Super Daniel
2022-09-06 11:46:12 +08:00
committed by GitHub
parent a203b709d5
commit d8a5aded19
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/fx/tracer/_meta_trace.py
View File

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