Super Daniel
|
1e88811c7a
|
[autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764)
* [autoparallel] first move.
* [autoparallel] add solver rotor.
* [autoparallel] add ckpt solvers.
* [autoparallel] modify codegen.
* [fx] fix annotation in test.
* [fx] remove check.
* [autoparallel] polish docstring.
* [fx] refactor MetaTensor.
|
2022-11-01 10:43:15 +08:00 |
|
ver217
|
d068af81a3
|
[doc] update rst and docstring (#1351)
* update rst
* add zero docstr
* fix docstr
* remove fx.tracer.meta_patch
* fix docstr
* fix docstr
* update fx rst
* fix fx docstr
* remove useless rst
|
2022-07-21 15:54:53 +08:00 |
|
Frank Lee
|
05fae1fd56
|
[fx] added activation checkpointing annotation (#1349)
* [fx] added activation checkpointing annotation
* polish code
* polish code
|
2022-07-21 11:14:28 +08:00 |
|
YuliangLiu0306
|
942c8cd1fb
|
[fx] refactor tracer to trace complete graph (#1342)
* [CLI] add CLI launcher
* Revert "[CLI] add CLI launcher"
This reverts commit df7e6506d4.
* [fx] refactor tracer to trace complete graph
* add comments and solve conflicts.
|
2022-07-20 11:20:38 +08:00 |
|
Frank Lee
|
75abc75c15
|
[fx] fixed compatiblity issue with torch 1.10 (#1331)
|
2022-07-18 11:41:27 +08:00 |
|
Frank Lee
|
4a09fc0947
|
[fx] fixed tracing with apex-based T5 model (#1252)
* [fx] fixed tracing with apex-based T5 model
* polish code
* polish code
|
2022-07-12 15:19:25 +08:00 |
|
Frank Lee
|
11973d892d
|
[fx] added torchvision model tracing testing (#1216)
* [fx] added torchvision model tracing testing
* remove unused imports
|
2022-07-06 21:37:56 +08:00 |
|
Frank Lee
|
2d13a45a3b
|
[fx] added testing for all gpt variants (#1210)
* [fx] added testing for all gpt variants
* polish code
* polish code
|
2022-07-06 14:03:13 +08:00 |
|
Frank Lee
|
f7878f465c
|
[fx] supported model tracing for huggingface bert (#1201)
* [fx] supported model tracing for huggingface bert
* polish test
|
2022-07-05 13:19:57 +08:00 |
|
Frank Lee
|
6d86f1bc91
|
[fx] supported data-dependent control flow in model tracing (#1185)
* [fx] supported data-dependent control flow in model tracing
* polish code
|
2022-06-29 15:05:25 +08:00 |