diff options
| author | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2026-01-15 21:30:26 +0100 |
|---|---|---|
| committer | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2026-01-31 13:40:04 +0100 |
| commit | d4352fdcecacd204b52e1c1a3646633e963e5490 (patch) | |
| tree | 9950c7c1af5bcd9664a0b6a55f7f8cc40cc32e0a /bench/Main.hs | |
| parent | f06186b2005ca2b939c219f23ebefe7d4ea5eb99 (diff) | |
Add benchmarks for sumFull
Diffstat (limited to 'bench/Main.hs')
| -rw-r--r-- | bench/Main.hs | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/bench/Main.hs b/bench/Main.hs index 2058e77..ce3a9df 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]" $ |