ColossalAI/colossalai/fx
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
..
codegen
passes
profiler [autoparallel] modify comm nodes' memory cost in construct chain (#2263) 2023-01-03 11:38:48 +08:00
tracer [auto-parallel] refactoring ColoTracer (#2118) 2023-01-04 14:44:22 +08:00
__init__.py
_compatibility.py
_meta_registrations.py
graph_module.py
proxy.py