From bb1deceb98b4c7bfcd35372e0289566cb593d8a9 Mon Sep 17 00:00:00 2001
From: Mikolaj Konarski <mikolaj.konarski@funktory.com>
Date: Thu, 15 Jan 2026 21:30:26 +0100
Subject: Speed up sumFull from 36ms to 82 microseconds

---
 bench/Main.hs | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'bench/Main.hs')

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]" $
-- 
cgit v1.2.3-70-g09d2