github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-30 10:55:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
52fda88796
ColossalAI/tests/test_auto_parallel/test_node_handler
History
YuliangLiu0306 52fda88796
[autoparallel] add layer norm handler v2 (#1671) 
* [autoparallel] add layer norm handler v2

* polish code

* polish code
2022-10-09 14:23:22 +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 conv handler v2 (#1663) 2022-09-28 11:24:59 +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] add conv handler v2 (#1663) 2022-09-28 11:24:59 +08:00