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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-08 11:56:27 +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