1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-07 16:08:22 +00:00
ColossalAI/colossalai/fx
Super Daniel 2edbef13cc
[fx] add more meta_registry for MetaTensor execution. ()
* [sc] add examples for auto checkpoint.

* merge upstream

* [fx] add more meta_registry for MetaTensor execution.
2022-11-23 10:55:46 +08:00
..
codegen [autoparallel] move ckpt solvers to autoparallel folder / refactor code () 2022-11-01 10:43:15 +08:00
passes [SC] add GPT example for auto checkpoint () 2022-11-11 23:17:25 +08:00
profiler [fx] add more meta_registry for MetaTensor execution. () 2022-11-23 10:55:46 +08:00
tracer [fx] add more meta_registry for MetaTensor execution. () 2022-11-23 10:55:46 +08:00
__init__.py [fx] metainfo_trace as an API. () 2022-11-10 20:58:37 +08:00
_compatibility.py [fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug () 2022-10-18 10:44:23 +08:00
_meta_registrations.py [fx] add more meta_registry for MetaTensor execution. () 2022-11-23 10:55:46 +08:00
graph_module.py [fx] Add offload codegen () 2022-09-14 15:49:06 +08:00
proxy.py [hotfix] fix coloproxy typos. () 2022-08-30 11:39:03 +08:00