Skip to content

Commit 09380f4

Browse files
authored
Fix
1 parent b9329a1 commit 09380f4

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

src/slurmmanager.jl

+8-4
Original file line numberDiff line numberDiff line change
@@ -212,11 +212,11 @@ function Distributed.launch(manager::SlurmManager, params::Dict, instances_arr::
212212
exehome = params[:dir]
213213
exename = params[:exename]
214214
exeflags = params[:exeflags]
215-
custom_worker_flag = params[:custom_worker_flag]
216-
217-
_srun_cmd_without_env = `srun -D $exehome $exename $exeflags $custom_worker_flag`
218215

219216
@static if Base.VERSION >= v"1.6.0"
217+
custom_worker_flag = params[:custom_worker_flag]
218+
_srun_cmd_without_env = `srun -D $exehome $exename $exeflags $custom_worker_flag`
219+
220220
# Pass the key-value pairs from `params[:env]` to the `srun` command:
221221
env2 = _new_environment_additions(Dict{String,String}(params[:env]))
222222
srun_cmd_with_env = addenv(_srun_cmd_without_env, env2)
@@ -225,7 +225,11 @@ function Distributed.launch(manager::SlurmManager, params::Dict, instances_arr::
225225
if haskey(params, :env)
226226
@warn "SlurmClusterManager.jl does not support params[:env] on Julia 1.5 and earlier" Base.VERSION
227227
end
228-
srun_cmd_with_env = _srun_cmd_without_env
228+
if haskey(params, :custom_worker_flag)
229+
@warn "SlurmClusterManager.jl does not support params[:custom_worker_flag] on Julia 1.5 and earlier" Base.VERSION
230+
end
231+
232+
srun_cmd_with_env = `srun -D $exehome $exename $exeflags --worker`
229233
end
230234

231235
# Pass cookie as stdin to srun; srun forwards stdin to process

0 commit comments

Comments
 (0)