Mercurial > repos > malex > gaussian
comparison gaussian_job_rules.py @ 2:ed472dc06c20 draft default tip
Uploaded 1.0.0
| author | malex |
|---|---|
| date | Tue, 18 Sep 2012 23:03:01 -0400 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 1:28ebcfcb10a2 | 2:ed472dc06c20 |
|---|---|
| 1 def gaussian(job): | |
| 2 inp_data = dict( [ ( da.name, da.dataset ) for da in job.input_datasets ] ) | |
| 3 inp_data.update( [ ( da.name, da.dataset ) for da in job.input_library_datasets ] ) | |
| 4 src_comfile = inp_data[ "infile" ].file_name | |
| 5 comfile = open(src_comfile, 'r') | |
| 6 ppn='1' | |
| 7 pmem = '900mb' | |
| 8 overhead = 200 | |
| 9 walltime = '11:59:59' | |
| 10 for line in comfile: | |
| 11 line = line.strip().lower() | |
| 12 if ( "%nproc=" in line): | |
| 13 try: | |
| 14 ppn = line.split('=')[1] | |
| 15 except Exception, e: | |
| 16 log.debug(e) | |
| 17 sys.stdout.write(e) | |
| 18 if ("%mem=" in line): | |
| 19 try: | |
| 20 mem_str = line.split('=')[1] | |
| 21 if ( "kb" in mem_str ): | |
| 22 mem_num = mem_str.split('kb')[0] | |
| 23 mem_num = int(int(mem_num) / 1024.0) + overhead | |
| 24 if ( "mb" in mem_str ): | |
| 25 mem_num = mem_str.split('mb')[0] | |
| 26 mem_num = int(mem_num) + overhead | |
| 27 elif ( "gb" in mem_str ): | |
| 28 mem_num = mem_str.split('gb')[0] | |
| 29 mem_num = (int(mem_num) * 1024) + overhead | |
| 30 elif ( "kw" in mem_str ): | |
| 31 mem_num = mem_str.split('kw')[0] | |
| 32 mem_num = int(int(mem_num) * 8 / 1024.0) + overhead | |
| 33 elif ( "mw" in mem_str ): | |
| 34 mem_num = mem_str.split('mw')[0] | |
| 35 mem_num = (int(mem_num) * 8) + overhead | |
| 36 elif ( "gw" in mem_str ): | |
| 37 mem_num = mem_str.split('gw')[0] | |
| 38 mem_num = (int(mem_num) * 8 * 1024) + overhead | |
| 39 else: | |
| 40 #Assume words | |
| 41 mem_num = int(int(mem_str) * 8 / 1024.0 / 1024.0) + overhead | |
| 42 pmem_num = int(mem_num / int(ppn)) | |
| 43 pmem = str(pmem_num) + 'mb' | |
| 44 except Exception, e: | |
| 45 log.debug(e) | |
| 46 sys.stdout.write(e) | |
| 47 if ("!walltime=" in line): | |
| 48 try: | |
| 49 walltime = line.split('=')[1] | |
| 50 except Exception, e: | |
| 51 log.debug(e) | |
| 52 sys.stdout.write(e) | |
| 53 request = "-l nodes=1:ppn=%s,pmem=%s,walltime=%s" % (ppn, pmem, walltime) | |
| 54 return 'drmaa://%s/' % request |