Run benchmark helper scripts via Docker containers

Author: semihguresci

README.md

@@ -24,10 +24,13 @@ docker run --rm --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix
 docker run --rm --gpus all vk-bench --headless --frames 300 --out results.json
 ```
 
+
+Scripts default to Docker image `vk-bench`. Override with `VK_BENCH_IMAGE=<image>` if needed.
+
 ### Bench all 3 scenes
 
 ```bash
-scripts/run_bench.sh results
+scripts/run_bench.sh results  # runs inside Docker image vk-bench
 ```
 
 ## Example results
@@ -51,14 +54,15 @@ scripts/run_bench.sh results
 ## Nsight steps (exact command)
 
 ```bash
-scripts/nsight_capture.sh results/nsight_capture
+scripts/nsight_capture.sh results/nsight_capture  # profiles docker run
 ```
 
 Or directly:
 
 ```bash
 nsys profile --trace=vulkan,nvtx,cuda --output results/nsight_capture \
-  vk-bench --headless --scene million-tris --warmup 20 --frames 120 --out results/nsight_capture.json
+  docker run --rm --gpus all -v "$(pwd)/results:/results" vk-bench \
+  --headless --scene million-tris --warmup 20 --frames 120 --out /results/nsight_capture.json
 ```
 
 ![Nsight capture screenshot](docs/nsight-capture.svg)

scripts/collect_system_info.sh

@@ -1,21 +1,27 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
+IMAGE="${VK_BENCH_IMAGE:-vk-bench}"
 OUT="${1:-results/system_info.txt}"
 mkdir -p "$(dirname "$OUT")"
 
+if ! command -v docker >/dev/null 2>&1; then
+  echo "docker is required for scripts/collect_system_info.sh" >&2
+  exit 1
+fi
+
 {
   echo "== Date =="
   date -Iseconds
   echo
   echo "== uname =="
   uname -a
   echo
-  echo "== nvidia-smi =="
+  echo "== nvidia-smi (host) =="
   nvidia-smi || true
   echo
-  echo "== vulkaninfo --summary =="
-  vulkaninfo --summary || true
+  echo "== vulkaninfo --summary (container) =="
+  docker run --rm --gpus all "$IMAGE" vulkaninfo --summary || true
 } > "$OUT"
 
 echo "Wrote $OUT"

scripts/nsight_capture.sh

@@ -1,17 +1,26 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
+IMAGE="${VK_BENCH_IMAGE:-vk-bench}"
 OUT="${1:-results/nsight_capture}"
 mkdir -p "$(dirname "$OUT")"
 
+if ! command -v docker >/dev/null 2>&1; then
+  echo "docker is required for scripts/nsight_capture.sh" >&2
+  exit 1
+fi
+
 if ! command -v nsys >/dev/null 2>&1; then
-  echo "nsys not found. Install NVIDIA Nsight Systems to use this script." >&2
+  echo "nsys not found on host. Install NVIDIA Nsight Systems to use this script." >&2
   exit 1
 fi
 
 nsys profile \
   --trace=vulkan,nvtx,cuda \
   --output "$OUT" \
-  vk-bench --headless --scene million-tris --warmup 20 --frames 120 --vsync 0 --out "${OUT}.json"
+  docker run --rm --gpus all \
+    -v "$(pwd)/results:/results" \
+    "$IMAGE" \
+    --headless --scene million-tris --warmup 20 --frames 120 --vsync 0 --out /results/nsight_capture.json
 
 echo "Capture written to ${OUT}.qdrep (or .nsys-rep depending on version)."

scripts/run_bench.sh

@@ -1,12 +1,21 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
+IMAGE="${VK_BENCH_IMAGE:-vk-bench}"
 RESULT_DIR="${1:-results}"
 mkdir -p "$RESULT_DIR"
 
+if ! command -v docker >/dev/null 2>&1; then
+  echo "docker is required for scripts/run_bench.sh" >&2
+  exit 1
+fi
+
 for scene in triangle million-tris compute-copy; do
-  echo "Running scene: $scene"
-  vk-bench --headless --scene "$scene" --warmup 30 --frames 300 --vsync 0 --out "$RESULT_DIR/${scene}.json"
+  echo "Running scene in Docker: $scene"
+  docker run --rm --gpus all \
+    -v "$(pwd)/$RESULT_DIR:/results" \
+    "$IMAGE" \
+    --headless --scene "$scene" --warmup 30 --frames 300 --vsync 0 --out "/results/${scene}.json"
 done
 
 echo "Saved benchmark outputs to $RESULT_DIR"