github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 04:24:47 +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

4
colossalai/fx/passes/__init__.py
View File

@@ -1,4 +1,4 @@
from .adding_split_node_pass import balanced_split_pass, split_with_split_nodes_pass
from .shard_1d_pass import column_shard_linear_pass, row_shard_linear_pass
from .meta_info_prop import MetaInfoProp
from .concrete_info_prop import ConcreteInfoProp
from .meta_info_prop import MetaInfoProp, metainfo_trace
from .shard_1d_pass import column_shard_linear_pass, row_shard_linear_pass