[apply][2/2] Fused qk_norm_rope for Qwen3-MoE (#13998)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
This commit is contained in:
Yuan Luo
2025-12-07 20:25:18 +08:00
committed by GitHub
parent f2b5dcc976
commit 26d95008b6
2 changed files with 199 additions and 22 deletions

View File

@@ -542,6 +542,7 @@ class ServerArgs:
enable_attn_tp_input_scattered: bool = False
# Context parallelism used in the long sequence prefill phase of DeepSeek v3.2
enable_nsa_prefill_context_parallel: bool = False
enable_fused_qk_norm_rope: bool = False
# Dynamic batch tokenizer
enable_dynamic_batch_tokenizer: bool = False
@@ -3738,6 +3739,11 @@ class ServerArgs:
action="store_true",
help="Enable context parallelism used in the long sequence prefill phase of DeepSeek v3.2.",
)
parser.add_argument(
"--enable-fused-qk-norm-rope",
action="store_true",
help="Enable fused qk normalization and rope rotary embedding.",
)
# Dynamic batch tokenizer
parser.add_argument(