Commit 12061c89 authored by Maude Le Jeune's avatar Maude Le Jeune
Browse files

+ mem option for pbs launcher

parent 8ecbf6b3
......@@ -184,7 +184,7 @@ def launch_ssh(pipe, host_list, address=None, authkey='secret',
w.wait()
def launch_pbs(pipe, n, address=None, authkey='secret', job_name="job_", log_level=logging.WARNING, cpu_time="2:00:00", server=False, job_header="",ppn=1 ):
def launch_pbs(pipe, n, address=None, authkey='secret', job_name="job_", log_level=logging.WARNING, cpu_time="2:00:00", server=False, job_header="",ppn=1, mem=4 ):
""" Launch a bunch of distant workers through a PBS batch system.
"""
......@@ -206,7 +206,7 @@ def launch_pbs(pipe, n, address=None, authkey='secret', job_name="job_", log_lev
f.write ("#PBS -o %s\n"%logfile)
f.write ("#PBS -e %s\n"%errfile)
f.write ("#PBS -N %s%d\n"%(job_name,i))
f.write ("#PBS -l nodes=1:ppn=%d,walltime=%s\n"%(ppn, cpu_time))
f.write ("#PBS -l nodes=1:ppn=%d,walltime=%s,mem=%dgb\n"%(ppn, cpu_time,mem))
f.write (job_header+"\n")
f.write ("python -m pipelet.launchers -H %s -p %s -s %s -l %s"%(address[0],address[1],authkey,jobfile.replace('job','worker')))
f.close()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment