Change of behaviour in Slurm version 22.05

[ekouts]

From the release notes: https://slurm.schedmd.com/news.html

RELEASE NOTES FOR SLURM VERSION 22.05
...
-- srun will no longer read in SLURM_CPUS_PER_TASK. This means you will
    implicitly have to specify --cpus-per-task on your srun calls, or set the
    new SRUN_CPUS_PER_TASK env var to accomplish the same thing.

To achieve the same behavior as before we need either to pass explicitly --cpus-per-task=$SLURM_CPUS_PER_TASK or set the new SRUN_CPUS_PER_TASK env var.

export SRUN_CPUS_PER_TASK=$SLURM_CPUS_PER_TASK

As far as I understand we can pass the option for older version of slurm as well so there is no need to check for the version.