WIP benchmark file

1 files changed, 20 insertions, 0 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
new file mode 100644
index 0000000..d8582fe
--- /dev/null
+++ b/bench/Main.hs
@@ -0,0 +1,20 @@
+{-# LANGUAGE NumericUnderscores #-}
+{-# LANGUAGE TypeApplications #-}
+module Main where
+
+import Test.Tasty.Bench
+
+import Data.Array.Nested
+
+
+main :: IO ()
+main = defaultMain
+  [bgroup "Num"
+    [bench "sum(+) Double [1e6]" $
+      let n = 1_000_000
+      in nf (\(a, b) -> runScalar (rsumOuter1 (a + b)))
+            (riota @Double n, riota n)
+    ,bench "sum(*) Double [1e6]" $
+      let n = 1_000_000
+      in nf (\(a, b) -> runScalar (rsumOuter1 (a + b)))
+            (riota @Double n, riota n)]]