TensorRT-LLMs/examples/layer_wise_benchmarks/middleware/exclude_slurm_envs
Tailing Yuan a7fe043b13
[None][feat] Layer-wise benchmarks: support TEP balance, polish slurm scripts (#10237)
Signed-off-by: Tailing Yuan <yuantailing@gmail.com>
2026-01-05 11:23:04 +08:00

9 lines
150 B
Bash
Executable File

#!/bin/bash
set -euo pipefail
# Clear slurm envs
unset $(env | awk -F'=' '{print $1}' | (grep -E "SLURM_|SLURMD_|slurm_|MPI_|PMIX_" || true))
"$@"