github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-31 19:35:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
3a4d6f63a8
ColossalAI/tests/test_auto_parallel/test_node_handler
History
Frank Lee 3a4d6f63a8
[autoparallel] added node handler for bmm (#1655)
2022-09-28 11:32:16 +08:00
..
test_bmm_handler.py [autoparallel] added node handler for bmm (#1655) 2022-09-28 11:32:16 +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