diff --git a/python/sglang/srt/layers/flashinfer_comm_fusion.py b/python/sglang/srt/layers/flashinfer_comm_fusion.py index 086a2a46395b..325c487135d1 100644 --- a/python/sglang/srt/layers/flashinfer_comm_fusion.py +++ b/python/sglang/srt/layers/flashinfer_comm_fusion.py @@ -490,6 +490,7 @@ def flashinfer_allreduce_residual_rmsnorm( workspace=workspace_manager.workspace, pattern=_flashinfer_comm.AllReduceFusionPattern.kARResidualRMSNorm, launch_with_pdl=True, + trigger_completion_at_end=trigger_completion_at_end, residual_out=residual_out, norm_out=norm_out, residual_in=residual,