mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-02-05 02:31:33 +08:00
22 lines
393 B
Bash
Executable File
22 lines
393 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" \
|
|
-J "$ACCOUNT-tensorrt_llm.layer_wise_benchmarks" \
|
|
$EXTRA_ARGS \
|
|
-t "$TIME" \
|
|
--no-shell \
|
|
2>&1 \
|
|
| tee >(cat >&2) \
|
|
| awk '/Granted job allocation/ {print $NF}'
|