Allow specifying additional job args

start_vscode.sh

@@ -36,33 +36,36 @@ VSC_HOSTNAME="euler.ethz.ch"
 # 2. Command line options overwrite defaults
 # 3. Config file options  overwrite command line options
 
-# Configuration file default    : $HOME/.vsc_config
+# Configuration file default        : $HOME/.vsc_config
 VSC_CONFIG_FILE="$HOME/.vsc_config"
 
-# Username default              : no default
+# Username default                  : no default
 VSC_USERNAME=""
 
-# Number of CPU cores default   : 1 CPU core
+# Number of CPU cores default       : 1 CPU core
 VSC_NUM_CPU=1
 
-# Runtime limit default         : 1:00 hour
+# Runtime limit default             : 1:00 hour
 VSC_RUN_TIME="01:00"
 
-# Memory default                : 1024 MB per core
+# Memory default                    : 1024 MB per core
 VSC_MEM_PER_CPU_CORE=1024
 
-# Number of GPUs default        : 0 GPUs
+# Number of GPUs default            : 0 GPUs
 VSC_NUM_GPU=0
 
-# Waiting interval default      : 60 seconds
+# Waiting interval default          : 60 seconds
 VSC_WAITING_INTERVAL=60
 
-# SSH key location default      : no default
+# SSH key location default          : no default
 VSC_SSH_KEY_PATH=""
 
-# Batch system                  : Slurm
+# Batch system                      : Slurm
 VSC_BATCH_SYSTEM="SLURM"
 
+# Additional job arguments default  : no default
+VSC_JOB_ARGS=""
+
 ###############################################################################
 # Usage instructions                                                          #
 ###############################################################################
@@ -89,6 +92,7 @@ Optional arguments:
         -i | --interval       INTERVAL         Time interval for checking if the job on the cluster already started
         -k | --key            SSH_KEY_PATH     Path to SSH key with non-standard name
         -v | --version                         Display version of the script and exit
+        -j | --jobargs        JOB_ARGS         Additional job arguments
 
 
 Examples:
@@ -109,6 +113,7 @@ VSC_MEM_PER_CPU_CORE=1024   # Memory limit in MB per core
 VSC_WAITING_INTERVAL=60     # Time interval to check if the job on the cluster already started
 VSC_SSH_KEY_PATH=""         # Path to SSH key with non-standard name
 VSC_BATCH_SYSTEM="SLURM"    # Batch system to use (SLURM or LSF)
+VSC_JOB_ARGS=""             # Additional job arguments
 
 EOF
 exit 1
@@ -173,6 +178,11 @@ do
                 shift
                 shift
                 ;;
+                -j|--jobargs)
+                VSC_JOB_ARGS=$2
+                shift
+                shift
+                ;;
                 *)
                 echo -e "Warning: ignoring unknown option $1 \n"
                 shift
@@ -326,7 +336,7 @@ ENDSSH
 echo -e "Connecting to $VSC_HOSTNAME to start the code-server in a $BATCH_SYS batch job"
 case $VSC_BATCH_SYSTEM in
         "LSF")
-        VSC_BJOB_OUT=$(ssh $VSC_SSH_OPT bsub -n $VSC_NUM_CPU -W $VSC_RUN_TIME -R "rusage[mem=$VSC_MEM_PER_CPU_CORE]" $VSC_SNUM_GPU<<ENDBSUB
+        VSC_BJOB_OUT=$(ssh $VSC_SSH_OPT bsub -n $VSC_NUM_CPU -W $VSC_RUN_TIME -R "rusage[mem=$VSC_MEM_PER_CPU_CORE]" $VSC_SNUM_GPU $VSC_JOB_ARGS<<ENDBSUB
         module load $VSC_MODULE_COMMAND
         export XDG_RUNTIME_DIR="\$HOME/vsc_runtime"
         VSC_IP_REMOTE="\$(hostname -i)"
@@ -344,7 +354,7 @@ ENDBSUB
         
         fi
 
-        VSC_BJOB_OUT=$(ssh $VSC_SSH_OPT sbatch --ntasks=1 --cpus-per-task=$VSC_NUM_CPU  "--time=$VSC_RUN_TIME" "--mem-per-cpu=$VSC_MEM_PER_CPU_CORE" -e "error.dat" $VSC_SNUM_GPU<<ENDBSUB
+        VSC_BJOB_OUT=$(ssh $VSC_SSH_OPT sbatch --ntasks=1 --cpus-per-task=$VSC_NUM_CPU  "--time=$VSC_RUN_TIME" "--mem-per-cpu=$VSC_MEM_PER_CPU_CORE" -e "error.dat" $VSC_SNUM_GPU $VSC_JOB_ARGS<<ENDBSUB
 #!/bin/bash
 module load $VSC_MODULE_COMMAND
 export XDG_RUNTIME_DIR="\$HOME/vsc_runtime"

vsconfig.example

@@ -5,3 +5,4 @@ VSC_RUN_TIME="01:00"        # Run time limit for the code-server in hours and mi
 VSC_MEM_PER_CPU_CORE=1024   # Memory limit in MB per core
 VSC_WAITING_INTERVAL=60     # Time interval to check if the job on the cluster already started
 VSC_SSH_KEY_PATH=""         # Path to SSH key with non-standard name
+VSC_JOB_ARGS=""             # Additional arguments when submitting the job