github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_fx/test_pipeline
History
Ziyue Jiang 632753abbc
[fx]Split partition with DAG information (#2025) 
* add DAG to split_module

* add comment

* add test case for DAG

* remove print

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
 		2022-11-25 17:42:48 +08:00
..
test_DAG [fx]Split partition with DAG information (#2025) 2022-11-25 17:42:48 +08:00
test_hf_model [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
test_timm_model [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
test_torchvision [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00