addition_dense_dense.py

import numpy as np
import benchmarkutils

name = "addition_dense_dense"

samples = 2
evals = 5
cutoffs = range(50, 801, 50)


def setup(N):
    op1 = np.random.rand(N, N) * 0.2j
    op2 = np.random.rand(N, N) * 0.1j
    return op1, op2


def f(op1, op2):
    return op1 + op2


print("Benchmarking:", name)
print("Cutoff: ", end="", flush=True)
results = []
for N in cutoffs:
    print(N, "", end="", flush=True)
    op1, op2 = setup(N)
    t = benchmarkutils.run_benchmark(f, op1, op2, samples=samples, evals=evals)
    results.append({"N": N, "t": t})
print()

benchmarkutils.save(name, results)