mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-13 22:18:36 +08:00
21 lines
338 B
Bash
Executable File
21 lines
338 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
# ACCOUNT=
|
|
# PARTITION=
|
|
# EXTRA_ARGS="--gres gpu:4"
|
|
TIME=${TIME:-01:00:00}
|
|
|
|
set -x
|
|
salloc -A "$ACCOUNT" \
|
|
-p "$PARTITION" \
|
|
-N "$NODES" \
|
|
--segment "$NODES" \
|
|
$EXTRA_ARGS \
|
|
-t "$TIME" \
|
|
--no-shell \
|
|
2>&1 \
|
|
| tee >(cat >&2) \
|
|
| awk '/Granted job allocation/ {print $NF}'
|