ColossalAI/colossalai/fx
Super Daniel 112a1f0a8f
[hotfix] avoid conflict of meta registry with torch 1.13.0. (#1530)
* [hotfix] avoid conflict of meta registry with torch 1.13.0.

* [hotfix] avoid conflict of meta registry with torch 1.13.0.
2022-09-01 15:31:21 +08:00
..
codegen [fx] Fix activation codegen dealing with checkpointing first op (#1510) 2022-08-27 19:39:21 +08:00
passes [fx] Fix wrong index in annotation and minimal flops in ckpt solver (#1521) 2022-08-31 18:10:48 +08:00
profiler [hotfix] avoid conflict of meta registry with torch 1.13.0. (#1530) 2022-09-01 15:31:21 +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 [fx] Fix ckpt functions' definitions in forward (#1476) 2022-08-22 16:59:54 +08:00
proxy.py [hotfix] fix coloproxy typos. (#1519) 2022-08-30 11:39:03 +08:00