Mateusz Suski
61 lines
1.6 KiB
YAML
61 lines
1.6 KiB
YAML
---
|
|
- name: Generate measurable Slurm usage for sreport
|
|
hosts: slurm_controller
|
|
become: true
|
|
gather_facts: false
|
|
|
|
tasks:
|
|
- name: Submit CPU usage job
|
|
ansible.builtin.shell: |
|
|
set -euo pipefail
|
|
|
|
job_id="$(
|
|
sudo -iu slurmuser sbatch --parsable <<'SBATCH'
|
|
#!/bin/bash
|
|
#SBATCH --job-name=sreport-usage
|
|
#SBATCH --partition=debug
|
|
#SBATCH --cpus-per-task=2
|
|
#SBATCH --mem=512M
|
|
#SBATCH --time=00:03:00
|
|
#SBATCH --output=/shared/sreport-usage-%j.out
|
|
|
|
echo "HOST=$(hostname)"
|
|
echo "SLURM_JOB_ID=$SLURM_JOB_ID"
|
|
echo "SLURM_CPUS_PER_TASK=${SLURM_CPUS_PER_TASK:-}"
|
|
echo "CPUS_ALLOWED=$(grep Cpus_allowed_list /proc/self/status)"
|
|
echo "Burning CPU for 90 seconds"
|
|
|
|
timeout 90 bash -c 'while true; do :; done' &
|
|
timeout 90 bash -c 'while true; do :; done' &
|
|
wait
|
|
|
|
echo "Done"
|
|
date
|
|
SBATCH
|
|
)"
|
|
|
|
echo "JOB_ID=$job_id"
|
|
|
|
for i in $(seq 1 150); do
|
|
if squeue -h -j "$job_id" | grep -q .; then
|
|
squeue -j "$job_id"
|
|
sleep 2
|
|
else
|
|
break
|
|
fi
|
|
done
|
|
|
|
echo "### sacct"
|
|
sacct -j "$job_id" --format=JobID,JobName,User,Partition,State,ExitCode,Elapsed,AllocCPUS,ReqMem,NodeList
|
|
|
|
echo "### output"
|
|
cat "/shared/sreport-usage-${job_id}.out"
|
|
args:
|
|
executable: /bin/bash
|
|
register: sreport_usage_job
|
|
changed_when: true
|
|
|
|
- name: Show usage job result
|
|
ansible.builtin.debug:
|
|
var: sreport_usage_job.stdout_lines
|