Update vllm/model_executor/layers/fused_moe/shared_fused_moe.py

SageMoore · yewentao256 · SageMoore · commit 87bef6c0b10f · 2025-11-10T18:26:52.000Z
Co-authored-by: Wentao Ye &lt;44945378+yewentao256@users.noreply.github.com&gt;
Signed-off-by: Sage Moore &lt;sagemoore@utexas.edu&gt;
Signed-off-by: Sage Moore &lt;sage@neuralmagic.com&gt;
diff --git a/vllm/model_executor/layers/fused_moe/shared_fused_moe.py b/vllm/model_executor/layers/fused_moe/shared_fused_moe.py
@@ -36,6 +36,7 @@ def __init__(
         self.use_overlapped = (
             use_overlapped
             and not (
+                TODO(wentao): find the root cause and remove this condition
                 self.enable_eplb
                 or (self.use_flashinfer_cutlass_kernels and self.dp_size > 1)
             )

Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,7 @@ def __init__(`
`36`	`36`	`self.use_overlapped = (`
`37`	`37`	`use_overlapped`
`38`	`38`	`and not (`
	`39`	`+ TODO(wentao): find the root cause and remove this condition`
`39`	`40`	`self.enable_eplb`
`40`	`41`	`or (self.use_flashinfer_cutlass_kernels and self.dp_size > 1)`
`41`	`42`	`)`