TensorRT-LLMs/tensorrt_llm/_torch/distributed
dongxuy04 b057fc9593
[None][fix] cherrypick to main: Fix possible mpi broadcast and gather issue on large object (#7854)
Signed-off-by: Dongxu Yang <78518666+dongxuy04@users.noreply.github.com>
2025-09-22 10:17:23 +08:00
..
__init__.py [TRTLLM-3927] [feat] Finalize + Allreduce + add + rmsnorm fusion (#4756) 2025-06-10 19:55:16 +08:00
communicator.py [None][fix] cherrypick to main: Fix possible mpi broadcast and gather issue on large object (#7854) 2025-09-22 10:17:23 +08:00
ops.py [https://nvbugs/5489015][fix] Support communicator split in MNNVL allreduce and fix the binding issues. (#7387) 2025-09-17 07:43:20 +08:00