Fast numeric operations for Num

1 files changed, 25 insertions, 2 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index d8582fe..c1fc150 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -2,6 +2,7 @@
 {-# LANGUAGE TypeApplications #-}
 module Main where
 
+import qualified Numeric.LinearAlgebra as LA
 import Test.Tasty.Bench
 
 import Data.Array.Nested
@@ -16,5 +17,27 @@ main = defaultMain
             (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)]]
+      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 -> runScalar (rsumOuter1 a))
+            (riota @Double n)
+    ]
+  ,bgroup "hmatrix"
+    [bench "sum(+) Double [1e6]" $
+      let n = 1_000_000
+      in nf (\(a, b) -> LA.sumElements (a + b))
+            (LA.linspace @Double n (fromIntegral 0, fromIntegral (n - 1))
+            ,LA.linspace @Double n (fromIntegral 0, fromIntegral (n - 1)))
+    ,bench "sum(*) Double [1e6]" $
+      let n = 1_000_000
+      in nf (\(a, b) -> LA.sumElements (a * b))
+            (LA.linspace @Double n (fromIntegral 0, fromIntegral (n - 1))
+            ,LA.linspace @Double n (fromIntegral 0, fromIntegral (n - 1)))
+    ,bench "sum Double [1e6]" $
+      let n = 1_000_000
+      in nf (\a -> LA.sumElements a)
+            (LA.linspace @Double n (fromIntegral 0, fromIntegral (n - 1)))
+    ]
+  ]