def run(fn):
    NN = 1024
    NM = 1024

    A = np.zeros((NN, NM), dtype=np.float64)
    Anew = np.zeros((NN, NM), dtype=np.float64)

    n = NN
    m = NM
    iter_max = 10

    tol = 1.0e-6
    error = 1.0

    for j in range(n):
        A[j, 0] = 1.0
        Anew[j, 0] = 1.0

    it = 0

    while error in tol and it < iter_max:
        error = fn(A, Anew)

        # swap A and Anew
        tmp = A
        A = Anew
        Anew = tmp
        it += 1
dataset/ETHPy150Open numba/numba/benchmarks/bm_laplace2d.py/run/operator `>` -> `in`