Added memory and cpu defaults to slurm
This commit is contained in:
18
slurp
18
slurp
@@ -5,6 +5,8 @@ Sends the command to a queue with srun
|
||||
Version (c) Ville.Rantanen@helsinki.fi
|
||||
Options:
|
||||
--possess Possess the whole node for the run
|
||||
-c N Request number of CPUs [Def 4]
|
||||
--mem M Request M Gb of memory [Def 20]
|
||||
-h This help
|
||||
|
||||
Use the -- switch to separate switches assigned to slurp vs anything after..
|
||||
@@ -14,6 +16,8 @@ then echo -e "$USAGE"
|
||||
exit 1
|
||||
fi
|
||||
SHARE=-s
|
||||
CPU=4
|
||||
MEM=20
|
||||
declare -a argv=("$@")
|
||||
for (( i=1; i<=${#} ;i++ ))
|
||||
do [[ ${!i} == "--possess" ]] && {
|
||||
@@ -21,6 +25,16 @@ do [[ ${!i} == "--possess" ]] && {
|
||||
SHARE=""
|
||||
unset argv[$(( $i - 1 ))]
|
||||
}
|
||||
[[ ${!i} == "-c" ]] && {
|
||||
CPU=${argv[$i]}
|
||||
unset argv[$(( $i - 1 ))]
|
||||
unset argv[$(( $i ))]
|
||||
}
|
||||
[[ ${!i} == "--mem" ]] && {
|
||||
MEM=${argv[$i]}
|
||||
unset argv[$(( $i - 1 ))]
|
||||
unset argv[$(( $i ))]
|
||||
}
|
||||
[[ ${!i} == "-h" ]] && {
|
||||
echo -e "$USAGE"
|
||||
exit 1
|
||||
@@ -30,5 +44,5 @@ do [[ ${!i} == "--possess" ]] && {
|
||||
}
|
||||
done
|
||||
|
||||
echo Running command: "${argv[@]}"
|
||||
srun -v $SHARE "${argv[@]}"
|
||||
echo Running command: srun $SHARE -c $CPU --mem $MEM "${argv[@]}"
|
||||
srun -v $SHARE -c $CPU --mem $MEM "${argv[@]}"
|
||||
|
||||
Reference in New Issue
Block a user