From b3bb5226951adad6c3867e30740d9e077a17a7fc Mon Sep 17 00:00:00 2001
From: Caspar van Leeuwen <casparl@int6.local.snellius.surf.nl>
Date: Mon, 13 Jan 2025 13:14:35 +0100
Subject: [PATCH] Make sure eessi_container.sh will add --nv flag for the test
 step, when GPUs are available. This is needed to run the GPU test

---
 bot/test.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/bot/test.sh b/bot/test.sh
index 2e2c838dfa..4205ea8564 100755
--- a/bot/test.sh
+++ b/bot/test.sh
@@ -208,6 +208,12 @@ fi
 # Reframe configuration
 TEST_STEP_ARGS+=("--extra-bind-paths" "/sys/fs/cgroup:/hostsys/fs/cgroup:ro")
 
+# add options required to handle NVIDIA support
+if command_exists "nvidia-smi"; then
+    echo "Command 'nvidia-smi' found, using available GPU"
+    TEST_STEP_ARGS+=("--nvidia" "run")
+fi
+
 # prepare arguments to test_suite.sh (specific to test step)
 declare -a TEST_SUITE_ARGS=()
 if [[ ${EESSI_SOFTWARE_SUBDIR_OVERRIDE} =~ .*/generic$ ]]; then