mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
13 lines
309 B
Bash
13 lines
309 B
Bash
#!/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 --allow-run-as-root -np 2 trtllm-llmapi-launch python3 _run_mpi_comm_task.py --task_type $task
|
|
|
|
echo "Remote MPI session test completed"
|