minor fix

Signed-off-by: Yukun He <[email protected]>

Author: hyukn

tensorrt_llm/_torch/custom_ops/torch_custom_ops.py

@@ -1260,6 +1260,7 @@ def _(
     strategy: int,
     op: int,
     eps: float,
+    tp_size: int,
     trigger_completion_at_end: bool,
 ) -> torch.Tensor:
     if op == int(AllReduceFusionOp.NONE):