Better version check for Slurm

Author: vkarak

reframe/core/launchers/mpi.py

@@ -20,14 +20,14 @@ def __init__(self):
         self.use_cpus_per_task = True
         try:
             out = osext.run_command('srun --version')
-            match = re.search('slurm (\d+)\.(\d+)\.(\d+)', out.stdout)
+            match = re.search(r'slurm(-wlm)? (\d+)\.(\d+)\.(\d+)', out.stdout)
             if match:
                 # We cannot pass to semver strings like 22.05.1 directly
                 # because it is not a valid version string for semver. We
                 # need to remove all the leading zeros.
                 slurm_version = (
                     semver.VersionInfo(
-                        match.group(1), match.group(2), match.group(3)
+                        match.group(2), match.group(3), match.group(4)
                     )
                 )
                 if slurm_version < semver.VersionInfo(22, 5, 0):