ColossalAI/tests/test_auto_parallel/test_node_handler at 50f16a28501e98b1539e3a02e9933c1032ea7d1d - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-14 05:10:14 +00:00

Files

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