From 0b76aa8583256c34bbd74257beae78aac1aaa0b3 Mon Sep 17 00:00:00 2001
From: inter <inter@qq.com>
Date: Sun, 21 Sep 2025 20:19:36 +0800
Subject: [PATCH] Add File

---
 tools/scripts/slurm_test_single.sh | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 tools/scripts/slurm_test_single.sh

diff --git a/tools/scripts/slurm_test_single.sh b/tools/scripts/slurm_test_single.sh
new file mode 100644
index 0000000..2dea3ce
--- /dev/null
+++ b/tools/scripts/slurm_test_single.sh
@@ -0,0 +1,19 @@
+#!/usr/bin/env bash
+
+set -x
+
+PARTITION=$1
+GPUS=1
+GPUS_PER_NODE=1
+PY_ARGS=${@:2}
+JOB_NAME=eval
+SRUN_ARGS=${SRUN_ARGS:-""}
+
+srun -p ${PARTITION} \
+    --job-name=${JOB_NAME} \
+    --gres=gpu:${GPUS_PER_NODE} \
+    --ntasks=${GPUS} \
+    --ntasks-per-node=${GPUS_PER_NODE} \
+    --kill-on-bad-exit=1 \
+    ${SRUN_ARGS} \
+    python -u test.py ${PY_ARGS}