ColossalAI/colossalai/fx/passes
Boyuan Yao 46c6cc79a9
[fx] Add common node in model linearize (#1542)
* [fx] Add common node into linearize

* [fx] Add common node to solver
2022-09-05 18:35:05 +08:00
..
algorithms [fx] Add common node in model linearize (#1542) 2022-09-05 18:35:05 +08:00
__init__.py
adding_split_node_pass.py
meta_info_prop.py [fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515) 2022-08-31 16:30:16 +08:00
passes_for_gpt2_test.py
shard_1d_pass.py [Doc] add more doc for ColoTensor. (#1458) 2022-08-16 10:38:41 +08:00
split_module.py
utils.py [autoparallel] added liveness analysis (#1516) 2022-08-30 15:54:37 +08:00