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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-13 11:34:37 +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