Speed up sumFull from 36ms to 82 microseconds

1 files changed, 9 insertions, 0 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index 8fe0fdc..185bef0 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -176,6 +176,15 @@ tests_compare =
     ,bench "sum Double [1e6]" $
       nf (\a -> runScalar (rsumOuter1Prim a))
          (riota @Double n)
+    ,bench "sumAll iota [1e6]" $
+      nf (\a -> rsumAllPrim a)
+         (riota @Double n)
+    ,bench "sumAll rev1(iota) [1e6]" $
+      nf (\a -> rsumAllPrim a)
+         (rrev1 $ riota @Double n)
+    ,bench "sumAll reshape(iota) [1e6]" $
+      nf (\a -> rsumAllPrim a)
+         (rreshape (1 :$: n :$: 1 :$: ZSR) $ riota @Double n)
     ]
   ,bgroup "NumElt"
     [bench "sum(+) Double [1e6]" $