From c9cac432dc96d581913de7629534ebd2ede8ad98 Mon Sep 17 00:00:00 2001 From: Yukun He <23156053+hyukn@users.noreply.github.com> Date: Fri, 9 May 2025 12:51:02 +0800 Subject: [PATCH] chore: Fix pipeline break caused by previous PR (#4081) rebase + pipeline reuse (#4169) Fix import break caused by rebase. Signed-off-by: Yukun He <23156053+hyukn@users.noreply.github.com> --- tensorrt_llm/_torch/distributed/__init__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tensorrt_llm/_torch/distributed/__init__.py b/tensorrt_llm/_torch/distributed/__init__.py index ba0ccd77fb..82f5a23b61 100644 --- a/tensorrt_llm/_torch/distributed/__init__.py +++ b/tensorrt_llm/_torch/distributed/__init__.py @@ -1,7 +1,8 @@ +from tensorrt_llm.functional import AllReduceFusionOp + from .communicator import Distributed, MPIDist, PPComm, TorchDist -from .ops import (AllReduce, AllReduceFusionOp, AllReduceParams, - AllReduceStrategy, MoEAllReduce, allgather, reducescatter, - userbuffers_allreduce_finalize) +from .ops import (AllReduce, AllReduceParams, AllReduceStrategy, MoEAllReduce, + allgather, reducescatter, userbuffers_allreduce_finalize) __all__ = [ "allgather",