Browse Source

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

pull/1546/head
Super Daniel 2 years ago committed by GitHub
parent
commit
d8a5aded19
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      colossalai/fx/tracer/_meta_trace.py

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

Loading…
Cancel
Save