ColossalAI/tests/test_auto_parallel/test_node_handler
YuliangLiu0306 11ec070e53
[hotfix]unit test (#1670)
2022-09-29 12:49:28 +08:00
..
test_batch_norm_handler_v2.py [autoparallel] add batch norm handler v2 (#1666) 2022-09-29 11:02:49 +08:00
test_bmm_handler.py [hotfix]unit test (#1670) 2022-09-29 12:49:28 +08:00
test_conv_handler_v2.py [autoparallel] add conv handler v2 (#1663) 2022-09-28 11:24:59 +08:00
test_linear_handler_v2.py [autoparallel] add conv handler v2 (#1663) 2022-09-28 11:24:59 +08:00