Zihao 3a02b46447 [auto-parallel] refactoring ColoTracer (#2118) ... * add meta_data_computing * add checkpoint_annotation * rename proxy.data to proxy.meta_data and add bias addition pass * polish code * delete meta_prop_pass invoke and rename ori_node to orig_node * add TracerType * unify meta data computing * delete TracerType * handle setitem operation * operator.setitem		2023-01-04 14:44:22 +08:00
..
bias_addition_patch	[autoparallel] support linear function bias addition (#2104)	2022-12-09 10:31:36 +08:00
meta_patch	[autoparallel] support addmm in tracer and solver (#1961)	2022-11-16 14:59:18 +08:00
__init__.py	[fx] add a symbolic_trace api. (#1812)	2022-11-08 13:59:20 +08:00
_meta_trace.py	[fx] refactor memory utils and extend shard utils. (#1754)	2022-10-26 14:24:41 +08:00
_symbolic_trace.py	[fx] add more meta_registry for MetaTensor execution. (#2000)	2022-11-23 10:55:46 +08:00
_tracer_utils.py	[fx] refactor tracer to trace complete graph (#1342)	2022-07-20 11:20:38 +08:00
experimental.py	[auto-parallel] refactoring ColoTracer (#2118)	2023-01-04 14:44:22 +08:00
registry.py	[autoparallel] support addbmm computation (#2102)	2022-12-08 21:15:11 +08:00
tracer.py	[autoparallel] record parameter attribute in colotracer (#2217)	2022-12-28 19:29:08 +08:00