github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-04 18:40:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fx] metainfo_trace as an API. (#1873)

Browse Source 
* [fx] metainfo_trace as an API.

* [fx] add return.
This commit is contained in:
Super Daniel
2022-11-10 20:58:37 +08:00
committed by GitHub
parent 6d559ea614
commit 448248b27c
3 changed files with 39 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/fx/__init__.py
View File

@@ -1,4 +1,4 @@
from ._compatibility import compatibility, is_compatible_with_meta
from .graph_module import ColoGraphModule
from .passes import MetaInfoProp
from .passes import MetaInfoProp, metainfo_trace
from .tracer import ColoTracer, meta_trace, symbolic_trace