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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 09:52:50 +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