Michelle
|
ad285e1656
|
[NFC] polish colossalai/fx/tracer/_tracer_utils.py (#3323)
* [NFC] polish colossalai/engine/schedule/_pipeline_schedule.py code style
* [NFC] polish colossalai/fx/tracer/_tracer_utils.py code style
---------
Co-authored-by: Qianran Ma <qianranm@luchentech.com>
|
2023-03-29 17:53:32 +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 |
YuliangLiu0306
|
4631fef8a0
|
[fx]refactor tracer (#1335)
|
2022-07-19 15:50:42 +08:00 |
Frank Lee
|
b2475d8c5c
|
[fx] fixed unit tests for torch 1.12 (#1327)
|
2022-07-15 18:22:15 +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
|
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 |