ColossalAI/colossalai/fx
Frank Lee 2cc1175c76
[fx] tested the complete workflow for auto-parallel (#1336)
* [fx] tested the complete workflow for auto-parallel

* polish code

* polish code

* polish code
2022-07-20 10:45:17 +08:00
..
passes [fx] tested the complete workflow for auto-parallel (#1336) 2022-07-20 10:45:17 +08:00
tracer [fx]refactor tracer (#1335) 2022-07-19 15:50:42 +08:00
__init__.py [fx] supported data-dependent control flow in model tracing (#1185) 2022-06-29 15:05:25 +08:00
proxy.py [fx]refactor tracer (#1335) 2022-07-19 15:50:42 +08:00