hcartiaux/dotfiles
1
0
Fork 0
mirror of https://github.com/hcartiaux/dotfiles.git synced 2024-10-18 09:15:24 +02:00
Code Activity

[bash] Slurm integration improved

Browse source
This commit is contained in:
Hyacinthe Cartiaux 2017-08-24 15:01:32 +02:00
parent f59deb8096
commit 5afcdae976
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
bash/bash_slurm
View file

@ -3,8 +3,8 @@ __slurm_ps1_remaining_time(){
if [ -n "$SLURM_JOB_ID" ]; then
DATE_NOW=$(date +%s)
DATE_JOB_START=$(stat -c %Y "/sys/fs/cgroup/cpu/slurm/uid_${UID}/job_${SLURM_JOB_ID}")
DATE_TMP=$(sacct -j $SLURM_JOB_ID --format Timelimit | awk -F: '/:/ { print ($1 * 3600) + ($2 * 60) + $3 }')
((DATE_TMP = (DATE_TMP - DATE_NOW + DATE_JOB_START) / 60))
[[ -z "$TIME_LIMIT" ]] && TIME_LIMIT=$(scontrol show job $SLURM_JOB_ID | grep -Po 'TimeLimit=\K[^ ]*' | awk -F: '/:/ { print ($1 * 3600) + ($2 * 60) + $3 }')
((DATE_TMP = (TIME_LIMIT - DATE_NOW + DATE_JOB_START) / 60))
echo -n "[SLURM$SLURM_JOB_ID->$DATE_TMP]"
fi
}
@ -12,6 +12,6 @@ __slurm_ps1_remaining_time(){
test -n "$SLURM_PTY_PORT" && (
echo "[SLURM] SLURM_JOB_ID=$SLURM_JOB_ID"
echo "[SLURM] Your nodes are:"
srun bash -c 'echo " $(hostname)*$SLURM_CPUS_ON_NODE"' | sort
srun bash -c 'echo " $(hostname)*$SLURM_CPUS_ON_NODE"' | sort -u
) || true