ColossalAI/colossalai/fx
YuliangLiu0306 4b3d6caeb3
[fx]patch nn.functional convolution (#1528)
2022-09-01 19:05:07 +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 [fx]patch nn.functional convolution (#1528) 2022-09-01 19:05:07 +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