ColossalAI/tests/test_auto_parallel/test_node_handler
Frank Lee 4973157ad7
[autoparallel] added sharding spec conversion for linear handler (#1687)
2022-10-12 11:16:18 +08:00
..
test_batch_norm_handler_v2.py [autoparallel] add layer norm handler v2 (#1671) 2022-10-09 14:23:22 +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 following node generator (#1673) 2022-10-09 14:49:18 +08:00
test_getitem_handler.py [autoparallel] add following node generator (#1673) 2022-10-09 14:49:18 +08:00
test_layer_norm_handler_v2.py [autoparallel] add layer norm handler v2 (#1671) 2022-10-09 14:23:22 +08:00
test_linear_handler_v2.py [autoparallel] added sharding spec conversion for linear handler (#1687) 2022-10-12 11:16:18 +08:00
test_reshape_handler_v2.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00
test_unary_element_wise_handler_v2.py [autoparallel] add unary element wise handler v2 (#1674) 2022-10-09 17:30:42 +08:00