ColossalAI/colossalai/fx
YuliangLiu0306 f6032ddb17
[autoparallel] fix bias addition module (#1800)
2022-11-08 16:21:25 +08:00
..
codegen [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2022-11-01 10:43:15 +08:00
passes [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
profiler [fx] Add linear metainfo class for auto parallel (#1783) 2022-11-04 10:55:09 +08:00
tracer [autoparallel] fix bias addition module (#1800) 2022-11-08 16:21:25 +08:00
__init__.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
_compatibility.py [fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710) 2022-10-18 10:44:23 +08:00
_meta_registrations.py [fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710) 2022-10-18 10:44:23 +08:00
graph_module.py [fx] Add offload codegen (#1598) 2022-09-14 15:49:06 +08:00
proxy.py [hotfix] fix coloproxy typos. (#1519) 2022-08-30 11:39:03 +08:00