Add low-rank benchmarks for MTTKRP

dahong67 · Mar 1, 2024 · 32319b9 · 32319b9
1 parent 468898e
commit 32319b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmark/suites/mttkrp.jl b/benchmark/suites/mttkrp.jl
@@ -13,7 +13,7 @@ append!(
     SETUPS,
     [
         (; size = sz, rank = r, mode = n) for
-        sz in [ntuple(n -> In, 3) for In in 50:50:200], r in 50:50:300, n in 1:3
+        sz in [ntuple(n -> In, 3) for In in 50:50:200], r in [10; 50:50:300], n in 1:3
     ],
 )
 
@@ -31,7 +31,7 @@ append!(
     SETUPS,
     [
         (; size = sz, rank = r, mode = n) for sz in [(30, 100, 1000), (1000, 100, 30)],
-        r in 100:100:300, n in 1:3
+        r in [10; 100:100:300], n in 1:3
     ],
 )