Select Git revision

Jayesh Badwaik authored
value.py 769 B
#!/usr/bin/env python3
import argparse
import random
from mpi4py import MPI
def parser():
parser = argparse.ArgumentParser(description="Value of a number")
parser.add_argument("--number", type=int, help="Number to use for multiplication")
parser.add_argument("--input", type=str, help="Input file")
return parser
if __name__ == "__main__":
comm = MPI.COMM_WORLD
nproc = comm.Get_size()
args = parser().parse_args()
with open(args.input, "r") as f:
data = f.read()
problem_size = float(data) * float(args.number) * 12
runtime = problem_size / nproc * random.uniform(0.8, 1.2)
string = " " * random.randint(1, 10)
if comm.Get_rank() == 0:
print("Runtime={}{}".format(string,runtime))