ColossalAI/colossalai/fx/tracer
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
__init__.py
_meta_trace.py
_symbolic_trace.py
_tracer_utils.py
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