ColossalAI/tests/test_moe at 9ec1ce6ab1a287d898a2cb261d6ce6aa72db5512 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 01:19:15 +00:00

Files

History

Jiarui Fang 65c0f380c2 [format] polish name format for MOE (#481 )

2022-03-21 23:19:47 +08:00

..

test_grad_handler.py

[format] polish name format for MOE (#481 )

2022-03-21 23:19:47 +08:00

test_kernel.py

[MOE] add unitest for MOE experts layout, gradient handler and kernel (#469 )

2022-03-21 13:35:04 +08:00

test_moe_group.py

[format] polish name format for MOE (#481 )

2022-03-21 23:19:47 +08:00