From ac97d3094a2a443a6e384eac5d9837b3c36e18f0 Mon Sep 17 00:00:00 2001
From: Paul Brinkmeier <hallo@pbrinkmeier.de>
Date: Fri, 5 Jan 2024 18:16:32 +0100
Subject: [PATCH] Add --unbuffered to batch configs

---
 experiments/batch/cuda               | 2 +-
 experiments/batch/cuda-singleblock   | 3 +--
 experiments/batch/nocuda             | 2 +-
 experiments/batch/nocuda-singleblock | 6 ++----
 4 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/experiments/batch/cuda b/experiments/batch/cuda
index 8dac19e..ec7a2d0 100644
--- a/experiments/batch/cuda
+++ b/experiments/batch/cuda
@@ -27,6 +27,6 @@ source "${SOURCE_DIR}/activate-nastja-modules"
 # However I'm leaving this here for documentation reasons
 export CUDA_VISIBLE_DEVICES=0
 
-srun "${SOURCE_DIR}/nastja/build-cuda/nastja" \
+srun --unbuffered "${SOURCE_DIR}/nastja/build-cuda/nastja" \
   -c "${SOURCE_DIR}/ma/experiments/configs/genesis.json" \
   -o "${OUTPUT_DIR}"
diff --git a/experiments/batch/cuda-singleblock b/experiments/batch/cuda-singleblock
index 5a235ee..44480b2 100644
--- a/experiments/batch/cuda-singleblock
+++ b/experiments/batch/cuda-singleblock
@@ -9,7 +9,6 @@
 #SBATCH --nodes=1
 # Number of MPI processes
 #SBATCH --ntasks=1
-#SBATCH --cpus-per-task=1
 # For now, we are using a single GPU only
 #SBATCH --gres=gpu:1
 #SBATCH --time=01:00:00
@@ -26,6 +25,6 @@ source "${SOURCE_DIR}/activate-nastja-modules"
 # However I'm leaving this here for documentation reasons
 export CUDA_VISIBLE_DEVICES=0
 
-srun "${SOURCE_DIR}/nastja/build-cuda/nastja" \
+srun --unbuffered "${SOURCE_DIR}/nastja/build-cuda/nastja" \
   -c "${SOURCE_DIR}/ma/experiments/configs/genesis-singleblock.json" \
   -o "${OUTPUT_DIR}"
diff --git a/experiments/batch/nocuda b/experiments/batch/nocuda
index dda8ef8..93fc970 100644
--- a/experiments/batch/nocuda
+++ b/experiments/batch/nocuda
@@ -21,6 +21,6 @@ OUTPUT_DIR=/p/scratch/cellsinsilico/paul/nastja-out/nocuda
 mkdir -p "${OUTPUT_DIR}"
 source "${SOURCE_DIR}/activate-nastja-modules"
 
-srun "${SOURCE_DIR}/nastja/build-nocuda/nastja" \
+srun --unbuffered "${SOURCE_DIR}/nastja/build-nocuda/nastja" \
   -c "${SOURCE_DIR}/ma/experiments/configs/genesis.json" \
   -o "${OUTPUT_DIR}"
diff --git a/experiments/batch/nocuda-singleblock b/experiments/batch/nocuda-singleblock
index 3fd9976..3d8ba66 100644
--- a/experiments/batch/nocuda-singleblock
+++ b/experiments/batch/nocuda-singleblock
@@ -9,9 +9,7 @@
 #SBATCH --nodes=1
 # Number of MPI processes
 #SBATCH --ntasks=1
-#SBATCH --cpus-per-task=1
-# For now, we are using a single GPU only
-#SBATCH --time=01:00:00
+#SBATCH --time=06:00:00
 #SBATCH --output=logs/nocuda-singleblock-%j.log
 #SBATCH --error=logs/nocuda-singleblock-%j.log
 
@@ -21,6 +19,6 @@ OUTPUT_DIR=/p/scratch/cellsinsilico/paul/nastja-out/nocuda-singleblock
 mkdir -p "${OUTPUT_DIR}"
 source "${SOURCE_DIR}/activate-nastja-modules"
 
-srun "${SOURCE_DIR}/nastja/build-nocuda/nastja" \
+srun --unbuffered "${SOURCE_DIR}/nastja/build-nocuda/nastja" \
   -c "${SOURCE_DIR}/ma/experiments/configs/genesis-singleblock.json" \
   -o "${OUTPUT_DIR}"