added benchmark for qr with split 0 but non-tall skinny shape

helmholtz-analytics · Feb 14, 2025 · 3465c7c · 3465c7c
1 parent be331a8
commit 3465c7c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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)