Add low-rank benchmarks for larger MTTKRP

dahong67 · Mar 1, 2024 · 8f57eef · 8f57eef
1 parent 32319b9
commit 8f57eef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmark/suites/mttkrp-large.jl b/benchmark/suites/mttkrp-large.jl
@@ -13,7 +13,7 @@ append!(
     SETUPS,
     [
         (; size = sz, rank = r, mode = n) for sz in [ntuple(n -> In, 4) for In in 20:20:80],
-        r in 20:20:120, n in 1:4
+        r in [10; 20:20:120], n in 1:4
     ],
 )
 
@@ -22,7 +22,7 @@ append!(
     SETUPS,
     [
         (; size = sz, rank = r, mode = n) for sz in [(20, 40, 80, 500), (500, 80, 40, 20)],
-        r in 100:100:300, n in 1:4
+        r in [10; 100:100:300], n in 1:4
     ],
 )