ColossalAI/tests/test_moe/test_kernel.py at b19355f8f046aafa2d4547de1a2bada2e7aa563f - 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

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