Skip to content

GitLab

Unverified Commit 910d74f5 authored by Ryan McCormick's avatar Ryan McCormick Committed by GitHub
Browse files

fix: Add mpi arg to all srun commands (#5948)

parent 16956ac6
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
examples/basics/multinode/trtllm/srun_aggregated.sh
View file @ 910d74f5
...@@ -41,6 +41,7 @@ fi ...@@ -41,6 +41,7 @@ fi
# the stdout/stderr to files. # the stdout/stderr to files.
echo "Launching frontend services in background." echo "Launching frontend services in background."
srun \ srun \
--mpi pmix \
--overlap \ --overlap \
--container-image "${IMAGE}" \ --container-image "${IMAGE}" \
--container-mounts "${MOUNTS}" \ --container-mounts "${MOUNTS}" \
......
examples/basics/multinode/trtllm/srun_disaggregated.sh
View file @ 910d74f5
...@@ -44,6 +44,7 @@ fi ...@@ -44,6 +44,7 @@ fi
# the stdout/stderr to files. # the stdout/stderr to files.
echo "Launching frontend services in background." echo "Launching frontend services in background."
srun \ srun \
--mpi pmix \
--overlap \ --overlap \
--container-image "${IMAGE}" \ --container-image "${IMAGE}" \
--container-mounts "${MOUNTS}" \ --container-mounts "${MOUNTS}" \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment