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

Add DeepEP backend.

Browse Source 
Signed-off-by: Bo Li <22713281+bobboli@users.noreply.github.com>
This commit is contained in:
Bo Li 2026-01-13 09:10:49 +00:00
parent 3479d73b4f
commit 7077b9133b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/microbenchmarks/bench_moe_comm.py
View File

@ -313,6 +313,7 @@ def parse_args() -> argparse.Namespace:
"ALLGATHER",
"NVLINK_ONE_SIDED",
"NVLINK_TWO_SIDED",
"DEEPEP",
"DEEPEPLOWLATENCY",
],
help="Which communication backend to benchmark (default: run all backends).",
@ -524,7 +525,7 @@ def _run_benchmark_worker_under_current_mpi(args: argparse.Namespace) -> None:
)
backends = (
["ALLGATHER", "NVLINK_ONE_SIDED", "NVLINK_TWO_SIDED", "DEEPEPLOWLATENCY"]
["ALLGATHER", "NVLINK_ONE_SIDED", "NVLINK_TWO_SIDED", "DEEPEP", "DEEPEPLOWLATENCY"]
if args.backend is None
else [args.backend]
)