ColossalAI/colossalai/fx/passes/__init__.py

from .adding_split_node_pass import balanced_split_pass, split_with_split_nodes_pass
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
[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] 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`
[fx] metainfo_trace as an API. (#1873) * [fx] metainfo_trace as an API. * [fx] add return. 2022-11-10 12:58:37 +00:00			`from .meta_info_prop import MetaInfoProp, metainfo_trace`
			`from .shard_1d_pass import column_shard_linear_pass, row_shard_linear_pass`