Boyuan Yao 5c2ef9fc76 [autoparallel] modify comm nodes' memory cost in construct chain (#2263) ... * [autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline * [autoparallel] using fwd_time and bwd_time instead of fwd_flop and bwd_flop * [autoparallel] specifycomm nodes' memory cost in construct chain		2023-01-03 11:38:48 +08:00
..
codegen	[autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764)	2022-11-01 10:43:15 +08:00
passes	[Pipeline Middleware ] Fix deadlock when num_microbatch=num_stage (#2156)	2022-12-23 11:38:43 +08:00
profiler	[autoparallel] modify comm nodes' memory cost in construct chain (#2263)	2023-01-03 11:38:48 +08:00
tracer	[autoparallel] record parameter attribute in colotracer (#2217)	2022-12-28 19:29:08 +08:00
__init__.py	[fx] metainfo_trace as an API. (#1873)	2022-11-10 20:58:37 +08:00
_compatibility.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
_meta_registrations.py	register meta func for rnn (#2159)	2022-12-21 23:06:18 +08:00
graph_module.py	[fx] Add offload codegen (#1598)	2022-09-14 15:49:06 +08:00
proxy.py	[hotfix] fix coloproxy typos. (#1519)	2022-08-30 11:39:03 +08:00