From a0010622885dcb55a916bf3514c0e9040f6871e9 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Thu, 23 May 2024 00:18:17 +0200
Subject: Fast numeric operations for Num

---
 bench/Main.hs | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

(limited to 'bench')

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)))
+    ]
+  ]
-- 
cgit v1.2.3-70-g09d2