ColossalAI/tests/test_auto_parallel
YuliangLiu0306 0908d0fc61
[autoparallel]add backward cost info into strategies (#1524)
2022-09-07 11:19:00 +08:00
..
test_conv_handler.py [autoparallel]add backward cost info into strategies (#1524) 2022-09-07 11:19:00 +08:00
test_cost_graph.py [autoparallel] change the merge node logic (#1533) 2022-09-07 11:18:19 +08:00
test_dot_handler.py [autoparallel] add cost graph class (#1481) 2022-08-25 17:19:59 +08:00
test_liveness_analysis.py [autoparallel] added liveness analysis (#1516) 2022-08-30 15:54:37 +08:00
test_strategies_constructor.py [autoparellel]add strategies constructor (#1505) 2022-08-30 16:32:09 +08:00