Add back missing DeepEP LL params (#31911)

Signed-off-by: Elvir Crncevic <elvircrn@gmail.com>
2026-01-07 23:47:54 +01:00
parent 10ef65eded
commit ffc0a2798b
1 changed files with 2 additions and 0 deletions
--- a/vllm/model_executor/layers/fused_moe/deepep_ll_prepare_finalize.py
+++ b/vllm/model_executor/layers/fused_moe/deepep_ll_prepare_finalize.py
@@ -295,6 +295,8 @@ class DeepEPLLPrepareAndFinalize(mk.FusedMoEPrepareAndFinalize):
            self.max_tokens_per_rank,
            num_experts,
            use_fp8=self.use_fp8_dispatch,
+            round_scale=self.use_ue8m0_dispatch,
+            use_ue8m0=self.use_ue8m0_dispatch,
            **(dict(use_nvfp4=True) if use_nvfp4 else dict()),
            **(
                dict(x_global_scale=qc_a1_gscale_or_scale)