kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

chore: Fix pipeline break caused by previous PR (#4081) rebase + pipeline reuse (#4169)

Browse Source 
Fix import break caused by rebase.

Signed-off-by: Yukun He <23156053+hyukn@users.noreply.github.com>
This commit is contained in:
Yukun He 2025-05-09 12:51:02 +08:00 committed by GitHub
parent d80dc40135
commit c9cac432dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorrt_llm/_torch/distributed/__init__.py
View File

@ -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",