File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -20,25 +20,25 @@ def __init__(self):
2020 self .use_cpus_per_task = True
2121 try :
2222 out = osext .run_command ('srun --version' )
23- match = re .search ('slurm (\d+)\.(\d+)\.(\d+)' , out .stdout )
23+ match = re .search (r 'slurm(-wlm)? (\d+)\.(\d+)\.(\d+)' , out .stdout )
2424 if match :
2525 # We cannot pass to semver strings like 22.05.1 directly
2626 # because it is not a valid version string for semver. We
2727 # need to remove all the leading zeros.
2828 slurm_version = (
2929 semver .VersionInfo (
30- match .group (1 ), match .group (2 ), match .group (3 )
30+ match .group (2 ), match .group (3 ), match .group (4 )
3131 )
3232 )
3333 if slurm_version < semver .VersionInfo (22 , 5 , 0 ):
3434 self .use_cpus_per_task = False
3535 else :
36- getlogger ().warning (
36+ getlogger ().debug (
3737 'could not get version of Slurm, --cpus-per-task will be '
3838 'set according to the num_cpus_per_task attribute'
3939 )
4040 except Exception :
41- getlogger ().warning (
41+ getlogger ().debug (
4242 'could not get version of Slurm, --cpus-per-task will be set '
4343 'according to the num_cpus_per_task attribute'
4444 )
You can’t perform that action at this time.
0 commit comments