ColossalAI/colossalai/fx
Frank Lee a0436a62ee
[autoparallel] added liveness analysis (#1516)
* [autoparallel] added liveness analysis

* remove memory cost
2022-08-30 15:54:37 +08:00
..
codegen [fx] Fix activation codegen dealing with checkpointing first op (#1510) 2022-08-27 19:39:21 +08:00
passes [autoparallel] added liveness analysis (#1516) 2022-08-30 15:54:37 +08:00
profiler [fx] add more op patches for profiler and error message for unsupported ops. (#1495) 2022-08-25 23:11:13 +08:00
tracer [autoparallel] add cost graph class (#1481) 2022-08-25 17:19:59 +08:00
__init__.py [fx] Add activation checkpoint solver rotor (#1496) 2022-08-26 10:34:21 +08:00
graph_module.py
proxy.py [hotfix] fix coloproxy typos. (#1519) 2022-08-30 11:39:03 +08:00