Exemplo n.º 1
0
def main(hex_string):
    """Handle command line parameters"""
    
    try:
        
    except Exception as e:
        script_name = basename(__file__)
        print 'Usage: ' + script_name + ' <hex-value>'
        for max_cluster_size in max_cluster_sizes:
            for scale_factor in scale_factors:
                log10_samples = int(numpy.round(3 - numpy.log10(c13_factor)))

                sim_args = [ sim_type,
                             "--static_Bz", static_Bz,
                             "--c13_factor", c13_factor,
                             "--max_cluster_size", max_cluster_size,
                             "--scale_factor", scale_factor ] + sim_opts \
                             + [ log10_samples_hook, log10_samples ]

                if test_jobs:
                    subprocess.call(cmd_args(sim_args,3))

                else:
                    test_job_name = "./jobs/" + basename(sim_args) + ".o_feedback"
                    fname_candidates = glob.glob(test_job_name)
                    if not len(fname_candidates) == 1:
                        print("please run test jobs first")
                        exit(1)
                    fname = fname_candidates[0]

                    with open(fname,"r") as f:
                        for line in f:
                            if "WallTime" in line:
                                test_time_text = line.split()[2]
                                break

                    hh_mm_ss = [ int(t) for t in test_time_text.split(":") ]
                    test_time = hh_mm_ss[-1] + hh_mm_ss[-2]*60 + hh_mm_ss[-3]*3600
Exemplo n.º 3
0
    print("must specify number of samples to simulate (via {} [num])"
          .format(log10_samples_hook))
    exit(1)
log10_samples = cmd_args[cmd_args.index(log10_samples_hook)+1]
del cmd_args[cmd_args.index(log10_samples_hook)+1]
del cmd_args[cmd_args.index(log10_samples_hook)]

print_period = 1800 # seconds

# identify some directories and names
project_dir = os.path.dirname(os.path.realpath(__file__))
data_dir = "data"
sim_file = "simulate.exe"
summary_script = "fidelity-summary.py"
basename_module = "basename.py"
out_file = "{}/{}.txt".format(data_dir,basename(sim_args))
base_cmd = ["./"+sim_file] + cmd_args

if "TMPDIR" not in os.environ:
    print("please set the TMPDIR environment variable")
    exit(1)
job_dir = os.environ["TMPDIR"] + "/" + os.environ["USER"] + "/" + "".join(sys.argv[1:])

# move into job directory
os.makedirs(job_dir)
for fname in [ sim_file, summary_script, basename_module ]:
    shutil.copy2(project_dir+"/"+fname, job_dir+"/"+fname)
os.chdir(job_dir)
os.mkdir(data_dir)

# method to execute a single simulation
nodes = 1
tasks_per_node = 16
task_num = str(int(nodes)*tasks_per_node)

if not log10_samples_hook in sim_args:
    print("must specify number of samples to simulate (via {} [num])"
          .format(log10_samples_hook))
    exit(1)
log10_samples = sim_args[sim_args.index(log10_samples_hook)+1]
mem_per_task_estimate_in_mb = int(0.5 * 4**float(log10_samples))

project_dir = os.path.dirname(os.path.realpath(__file__))
job_dir = "jobs"
mkfac = "mkfac.py"
script = "fidelity-sweep.py"
job_file = job_dir+"/"+basename(sim_args)+".sh"
out_file = job_dir+"/"+basename(sim_args)+".o"

resources = ["nodes={}:ppn={}".format(nodes,tasks_per_node),
             "walltime={}:00:00".format(walltime_in_hours),
             "pmem={}mb".format(mem_per_task_estimate_in_mb)]
options = ["-N "+basename(sim_args), "-m e", "-j oe", "-o "+out_file]
for resource in resources:
    options += ["-l "+resource]

job_text = "#!/usr/bin/env sh\n"
for option in options:
    job_text += "#MSUB {}\n".format(option)
job_text += "\n"
job_text += "python {}/{} {} {}\n".format(project_dir,script,task_num," ".join(sim_args))