NVIDIA / NVIDIA/TensorRT-LLM

disaggregated service MPI communicate failed,pls help to check,thanks

Open
#5,012 2 comments 0 reactions 2 assignees View on GitHub

@Shixiaowei02 is already working on this.

Since Jun 11, 2025.

bug Disaggregated serving Investigating triaged
Dominant language
Python
Stars
14.7k
Forks
2.8k
Avg merge
2d 23h
Merged PRs (30d)
489

Description

version:0.19.0
GPU:NVIDIA GeForce RTX 4090
cmd:
export TRTLLM_USE_MPI_KVCACHE=1
mpirun --allow-run-as-root -n 3 --oversubscribe ./executorExampleDisaggregated --context_engine_dir qwen2 --context_rank_size 1 --generation_engine_dir qwen2 --generation_rank_size 1
--input_tokens_csv_file inputTokens.csv

log:
[TensorRT-LLM][DEBUG][2] Start calling requestSync for request ID: 1, context request ID: 1.
[TensorRT-LLM][DEBUG] start MPI_Send with size 8
[TensorRT-LLM][DEBUG] end MPI_Send with size 8
[TensorRT-LLM][DEBUG] start MPI_Send with size 8
[TensorRT-LLM][DEBUG] end MPI_Send with size 8
[TensorRT-LLM][DEBUG] start MPI_Send with size 204
[TensorRT-LLM][DEBUG] end MPI_Send with size 204
[TensorRT-LLM][TRACE] Created event 0x7faf24001940
[TensorRT-LLM][INFO] Set logger level to TRACE
[TensorRT-LLM][DEBUG] start MPI_Recv with size 8
[TensorRT-LLM][DEBUG] end MPI_Recv with size 8
[TensorRT-LLM][DEBUG] start MPI_Recv with size 204
[TensorRT-LLM][TRACE] Created stream 0x7faf24001360
[TensorRT-LLM][TRACE] static std::shared_ptr<tensorrt_llm::runtime::CudaMemPool> tensorrt_llm::runtime::CudaMemPool::getPrimaryPoolForDevice(int) start
[TensorRT-LLM][TRACE] static std::shared_ptr<tensorrt_llm::runtime::CudaMemPool> tensorrt_llm::runtime::CudaMemPool::getPrimaryPoolForDevice(int) stop
[TensorRT-LLM][DEBUG] end MPI_Recv with size 204
[TensorRT-LLM][TRACE] static const tensorrt_llm::mpi::MpiComm& tensorrt_llm::mpi::MpiComm::world() start
[TensorRT-LLM][TRACE] static const tensorrt_llm::mpi::MpiComm& tensorrt_llm::mpi::MpiComm::world() stop
[TensorRT-LLM][DEBUG][2] Start receiving KV cache for request ID: 1, context request ID: 1.
[TensorRT-LLM][DEBUG] start MPI_Recv with size 1835008
[TensorRT-LLM][TRACE] static const tensorrt_llm::mpi::MpiComm& tensorrt_llm::mpi::MpiComm::world() start
[TensorRT-LLM][TRACE] static const tensorrt_llm::mpi::MpiComm& tensorrt_llm::mpi::MpiComm::world() stop
[TensorRT-LLM][DEBUG][1] Start sending KV cache for request ID: 1.
[TensorRT-LLM][DEBUG] splitKVCache: numLayers: 28, headNum: 4, DomainPPSize:1, DomainTPSize:1, layerNumDomainPP:28, headNumDomainTP:4
[TensorRT-LLM][DEBUG] start MPI_Send with size 1835008
[test-59bcdcdc88-r2hqg:2557333] Read -1, expected 1835008, errno = 14

MPI_ABORT was invoked on rank 1 in communicator MPI_COMM_WORLD
with errorcode 1.

NOTE: invoking MPI_ABORT causes Open MPI to kill all MPI processes.
You may or may not see output from other processes, depending on
exactly when Open MPI kills them.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.