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

fix [nvbug5351244]: test_mpi_session submit sync/async (#5608)

Browse Source 
Signed-off-by: Superjomn <328693+Superjomn@users.noreply.github.com>
This commit is contained in:
Yan Chunwei 2025-07-01 00:48:59 +08:00 committed by GitHub
parent 647e070ed6
commit d6c81bad97
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/unittest/llmapi/_test_remote_mpi_session.sh Normal file
View File

@ -0,0 +1,12 @@
#!/bin/bash
set -ex
task=$1
echo "Starting remote MPI session test with task: $task"
echo "MPI processes: 2"
# Add timeout to prevent infinite hanging
timeout 60 mpirun -np 2 trtllm-llmapi-launch python3 _run_mpi_comm_task.py --task_type $task
echo "Remote MPI session test completed"

1
tests/unittest/llmapi/test_mpi_session.py
View File

@ -54,7 +54,6 @@ def run_client(server_addr, values_to_process):
return f"Error in client: {str(e)}"
@pytest.mark.skip(reason="https://nvbugs/5351244")
@pytest.mark.parametrize("task_type", ["submit", "submit_sync"])
def test_remote_mpi_session(task_type: Literal["submit", "submit_sync"]):
"""Test RemoteMpiPoolSessionClient and RemoteMpiPoolSessionServer interaction"""