ColossalAI/colossalai/fx
YuliangLiu0306 d87baa85d9
[autoparallel] support linear function bias addition (#2104)
2022-12-09 10:31:36 +08:00
..
codegen [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2022-11-01 10:43:15 +08:00
passes [Pipeline] Add Topo Class (#2059) 2022-12-02 18:13:20 +08:00
profiler [fx] add more meta_registry for MetaTensor execution. (#2000) 2022-11-23 10:55:46 +08:00
tracer [autoparallel] support linear function bias addition (#2104) 2022-12-09 10:31:36 +08:00
__init__.py [fx] metainfo_trace as an API. (#1873) 2022-11-10 20:58:37 +08:00
_compatibility.py
_meta_registrations.py [fx] add more meta_registry for MetaTensor execution. (#2000) 2022-11-23 10:55:46 +08:00
graph_module.py
proxy.py