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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 03:38:19 +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