qstat

# Authors: Mozhgan Kabiri Chimeh, Paul Richmond
# Contact: m.kabiri-chimeh@sheffield.ac.uk
#
# Startup script to allocate GPU devices.
# Based on https://github.com/kyamagu/sge-gpuprolog
#
# The script is written to emulate the 'qstat' command. It gets the number of GPUs as the JOB_ID and replaces it with the number of GPUs in 'userTest' file and echos it.
# Note that it gets environment varible that would normally be the $JOB_ID. We use the Job ID as the number of devices that qstat should return

# To execute, simply run qstat -j $2, where $2 would be the number of GPUs

NUM_GPUS=$2 # this is normaly the $JOB_ID, but here we are using it as the NUM_GPUS

# replaces the number of gpus already existed in the Twins qstat output with the one recieved from the user
replace_func(){
sed -i -e "s/gpu=[[:digit:]]/gpu=$NUM_GPUS/g" userTest
}


# Loop until all parameters are used up
if [ "$2" != "" ]; then
   replace_func
fi

cat userTest