Add support for params[:env]

Author: DilumAluthge

src/slurmmanager.jl

@@ -49,10 +49,17 @@ function launch(manager::SlurmManager, params::Dict, instances_arr::Array, c::Co
         exename = params[:exename]
         exeflags = params[:exeflags]
 
+        # Pass the key-value pairs from `params[:env]` to the `srun` command:
+        srun_cmd_without_env = `srun -D $exehome $exename $exeflags --worker`
+        env2 = Dict{String,String}()
+        for (name, value) in pairs(Dict{String,String}(params[:env]))
+            env2[name] = value
+        end
+        srun_cmd_with_env = addenv(srun_cmd_without_env, env2)
+
         # Pass cookie as stdin to srun; srun forwards stdin to process
         # This way the cookie won't be visible in ps, top, etc on the compute node
-        srun_cmd = `srun -D $exehome $exename $exeflags --worker`
-        manager.srun_proc = open(srun_cmd, write=true, read=true)
+        manager.srun_proc = open(srun_cmd_with_env, write=true, read=true)
         write(manager.srun_proc, cluster_cookie())
         write(manager.srun_proc, "\n")