mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
11 lines
205 B
Bash
Executable File
11 lines
205 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
# Clear slurm envs
|
|
unset $(env | grep -i slurm | awk -F'=' '{print $1}')
|
|
unset $(env | grep MPI | awk -F'=' '{print $1}')
|
|
|
|
set -x
|
|
mpirun --allow-run-as-root --np ${NP} "$@"
|