np.linalg.solve vs np.linalg.solve (twice)

inv-vs-double-solve.py

import perfplot
import numpy as np


def inv(AX):
    A, X = AX
    Ainv = np.linalg.inv(A)
    return Ainv @ X @ Ainv.T


def double_solve(AX):
    A, X = AX
    return np.linalg.solve(A, np.linalg.solve(A, X).T).T


b = perfplot.bench(
    setup=lambda n: (np.random.rand(n, n), np.random.rand(n, n)),
    kernels=[inv, double_solve],
    n_range=[2 ** k for k in range(10)],
)
b.save("out.png")
b.show()