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

github/ColossalAI

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