diff --git a/benchmarks/cb/linalg.py b/benchmarks/cb/linalg.py index 3596d4916..9202ca0d5 100644 --- a/benchmarks/cb/linalg.py +++ b/benchmarks/cb/linalg.py @@ -19,6 +19,11 @@ def qr_split_0(a): qr = ht.linalg.qr(a) +@monitor() +def qr_split_0_square(a): + qr = ht.linalg.qr(a) + + @monitor() def qr_split_1(a): qr = ht.linalg.qr(a) @@ -57,6 +62,11 @@ def run_linalg_benchmarks(): qr_split_0(a_0) del a_0 + n = 2000 + a_0 = ht.random.random((n, n), split=0) + qr_split_0_square(a_0) + del a_0 + n = 2000 a_1 = ht.random.random((n, n), split=1) qr_split_1(a_1)