ColossalAI/tests/test_moe/test_kernel.py at 73bdfd88910efeecc4f09025773ecc58305aa494 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-11 06:33:30 +00:00

Files

hxwang 803878b2fd [moe] full test for deepseek and mixtral (pp + sp to fix)

2024-08-01 10:06:59 +08:00

3.2 KiB

Raw Blame History

View Raw