ColossalAI/colossalai/fx/passes/__init__.py

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
[fx]add autoparallel passes (#1121) * [CLI] add CLI launcher * Revert "[CLI] add CLI launcher" This reverts commit df7e6506d4500af6a9220ef7fe4d3c7b1daebd4c. * feature/add autoparallel passes 2022-06-15 08:36:46 +00:00			`from .adding_split_node_pass import balanced_split_pass, split_with_split_nodes_pass`
[fx/profiler] provide a table of summary. (#1634) * [fx/profiling] provide summary for MetaInfoProp. * [fx/profiler] provide a table of summary. * [fx] optimize table repr. 2022-09-23 10:12:43 +00:00			`from .shard_1d_pass import column_shard_linear_pass, row_shard_linear_pass`
			`from .meta_info_prop import MetaInfoProp`
[fx] Add concrete info prop (#1677) * [fx] concreteinfoprop * [fx] add concreteinfoprop * [fx] modify docstring of ConcreteInfoProp * [fx] fix device error * [fx] modify parameter calculation * [fx] modify parameters calculation 2022-10-04 08:48:24 +00:00			`from .concrete_info_prop import ConcreteInfoProp`