ColossalAI

Commit Graph

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

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

3 Commits (a520610bd9eaaf6374779756fb6b2c15c390a303)