You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ziyue Jiang 0f02b8c6e6 add avg partition (#2483) ... Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>		2 years ago
..
algorithms	[fx] refactor memory utils and extend shard utils. (#1754)	2 years ago
experimental	[autoparallel] refactor the runtime apply pass and add docstring to passes (#1757)	2 years ago
__init__.py	[fx] metainfo_trace as an API. (#1873)	2 years ago
adding_split_node_pass.py	add avg partition (#2483)	2 years ago
concrete_info_prop.py	[fx] refactor memory utils and extend shard utils. (#1754)	2 years ago
meta_info_prop.py	add avg partition (#2483)	2 years ago
passes_for_gpt2_test.py	[hotfix] fix some bugs during gpt2 testing (#1379)	2 years ago
shard_1d_pass.py	[Doc] add more doc for ColoTensor. (#1458)	2 years ago
split_module.py	[Pipeline] Add Topo Class (#2059)	2 years ago
utils.py	[Pipeline] Add Topo Class (#2059)	2 years ago