ColossalAI/colossalai/fx
Boyuan Yao 933b6c6367
[fx] Add pofo solver (#1608)
* [fx] add pofo algorithm

* [fx] Add pofo solver

* [fx] code refactor

* [fx] fix test_linearize import
2022-09-20 11:20:48 +08:00
..
codegen
passes [fx] Add pofo solver (#1608) 2022-09-20 11:20:48 +08:00
profiler
tracer
__init__.py
graph_module.py
proxy.py