Super Daniel 0584654c79 [fx] refactor memory utils and extend shard utils. (#1754) ... * [fx] change memory.py to memory_utils.py. * [fx] add shard utils. * [fx] fix import. * [fx] check code style. * [fx] add comment. * [autoparallel] first move. * [fx] add time computations.		2022-10-26 14:24:41 +08:00
..
test_ckpt_solvers	[fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710)	2022-10-18 10:44:23 +08:00
test_codegen	[fx] fix offload codegen test (#1648)	2022-09-27 10:25:27 +08:00
test_meta	[fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710)	2022-10-18 10:44:23 +08:00
test_pipeline	[fx] refactor tracer to trace complete graph (#1342)	2022-07-20 11:20:38 +08:00
test_profiler	[fx] refactor memory utils and extend shard utils. (#1754)	2022-10-26 14:24:41 +08:00
test_tracer	[fx] test tracer on diffuser modules. (#1750)	2022-10-20 18:25:05 +08:00
test_coloproxy.py	[fx] refactor tracer to trace complete graph (#1342)	2022-07-20 11:20:38 +08:00
test_comm_size_compute.py	[fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710)	2022-10-18 10:44:23 +08:00
test_complete_workflow.py	[fx] tested the complete workflow for auto-parallel (#1336)	2022-07-20 10:45:17 +08:00
test_graph_manipulation.py	[fx] methods to get fx graph property. (#1246)	2022-07-12 14:10:37 +08:00
test_meta_info_prop.py	[fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710)	2022-10-18 10:44:23 +08:00
test_parallel_1d.py	[fx]add autoparallel passes (#1121)	2022-06-15 16:36:46 +08:00
test_pipeline_passes.py	[fx] recovered skipped pipeline tests (#1338)	2022-07-19 09:49:50 +08:00