Optimize layernorm_gated for Qwen3-Next (#16397)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
2026-01-10 20:55:31 +08:00
parent bdb76b34db
commit d1ec93e3ac
5 changed files with 791 additions and 34 deletions
@@ -70,6 +70,7 @@ suites = {
        TestFile("models/test_ministral3_models.py"),
        TestFile("test_mistral_large3_basic.py"),
        TestFile("test_prefill_delayer.py"),
+        TestFile("test_fla_layernorm_guard.py"),
    ],
 }