prepareBenchmark.py

Go to the documentation of this file.

 
# File to prepare the configs and commands to be executed for the benchmark
 
 
def prepareConfig(
    template_filename, n_threads=10, n_parallel_events=10, n_parallel_algorithms=10
):
    template = open(template_filename)
    new_filename = f"{template_filename.rstrip('.py')}_{n_threads}_{n_parallel_events}_{n_parallel_algorithms}.py"
    new_config = open(new_filename, "w")
    for line in template.readlines():
        if line.startswith("n_threads"):
            line = f"n_threads = {n_threads}\n"
        elif line.startswith("n_parallel_events"):
            line = f"n_parallel_events = {n_parallel_events}\n"
        elif line.startswith("n_parallel_algorithms"):
            line = f"n_parallel_algorithms = {n_parallel_algorithms}\n"
        new_config.write(line)
    new_config.close()
    return new_filename
 
 

if __name__ == "__main__":
    n_threads = 10
    for n_algos in range(1, 11):
        for n_events in range(1, n_algos + 1):
            config = prepareConfig(
                "../options/BrunelScenario.py",
                n_threads=n_threads,
                n_parallel_events=n_events,
                n_parallel_algorithms=n_algos,
            )
            # config.replace(".py",".log"))
            print(
                "/usr/bin/time -f %S -o {}.time `alias gaudirun` {} > {}".format(
                    config.replace(".py", ""), config, "/dev/null"
                )
            )