github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
1,944 Commits
32 Branches
47 Tags
229 MiB
Commit Graph

3 Commits (5cd8cae0c9db7ae4bfb98e42ec9c7ab2e6ebd17f)

Author SHA1 Message Date
Super Daniel c41e59e5ad
[fx] allow native ckpt trace and codegen. (#2438) 2023-01-11 13:49:59 +08:00
Zihao 3a02b46447
[auto-parallel] refactoring ColoTracer (#2118) 
* add meta_data_computing

* add checkpoint_annotation

* rename proxy.data to proxy.meta_data and add bias addition pass

* polish code

* delete meta_prop_pass invoke and rename ori_node to orig_node

* add TracerType

* unify meta data computing

* delete TracerType

* handle setitem operation

* operator.setitem
 2023-01-04 14:44:22 +08:00
Super Daniel 2bf2d1cd3b
[fx] An experimental version of ColoTracer.' (#2002) 
* [fx] add a symbolic_trace api.

* [fx] fix import errors.

* [fx] ColoTracer experimental.
 2022-12-07 18:36:17 +08:00