Skip to content

Commit 3465c7c

Browse files
author
Hoppe
committed
added benchmark for qr with split 0 but non-tall skinny shape
1 parent be331a8 commit 3465c7c

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

benchmarks/cb/linalg.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,11 @@ def qr_split_0(a):
1919
qr = ht.linalg.qr(a)
2020

2121

22+
@monitor()
23+
def qr_split_0_square(a):
24+
qr = ht.linalg.qr(a)
25+
26+
2227
@monitor()
2328
def qr_split_1(a):
2429
qr = ht.linalg.qr(a)
@@ -57,6 +62,11 @@ def run_linalg_benchmarks():
5762
qr_split_0(a_0)
5863
del a_0
5964

65+
n = 2000
66+
a_0 = ht.random.random((n, n), split=0)
67+
qr_split_0_square(a_0)
68+
del a_0
69+
6070
n = 2000
6171
a_1 = ht.random.random((n, n), split=1)
6272
qr_split_1(a_1)

0 commit comments

Comments
 (0)