ColossalAI/tests/test_moe/test_kernel.py at grpo-agentic-dev - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-03-20 03:46:18 +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 Permalink Blame History

View Raw