Support Kimi Linear (#12469)

Co-authored-by: yizhang2077 <1109276519@qq.com>
2025-10-31 14:03:35 -07:00
parent 30ad107028
commit a4bf5c6ad2
18 changed files with 2847 additions and 112 deletions
@@ -151,6 +151,7 @@ suites = {
        TestFile("layers/attention/mamba/test_mamba2_mixer.py", 50),
        TestFile("lora/test_lora_tp.py", 116),
        TestFile("models/test_glm4_moe_models.py", 100),
+        TestFile("models/test_kimi_linear_models.py", 90),
        TestFile("rl/test_update_weights_from_distributed.py", 103),
        TestFile("test_data_parallelism.py", 73),
        TestFile("test_disaggregation_basic.py", 400),