Benchmark dot product

2 files changed, 25 insertions, 0 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index fdb7f25..eb3c6d7 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -3,6 +3,7 @@
 {-# LANGUAGE TypeApplications #-}
 module Main where
 
+import Data.Array.RankedS qualified as RS
 import Numeric.LinearAlgebra qualified as LA
 import Test.Tasty.Bench
 
@@ -64,6 +65,14 @@ main = defaultMain
       let n = 1_000_000
       in nf (\a -> runScalar (rsumOuter1 a))
             (riota @Double n)
+    ,bench "dotprod Double [1e6]" $
+      let n = 1_000_000
+      in nf (\(a, b) -> rdot a b)
+            (riota @Double n, riota @Double n)
+    ,bench "dotprod Double [1e6] stride 1; -1" $
+      let n = 1_000_000
+      in nf (\(a, b) -> rdot a b)
+            (riota @Double n, rrev1 (riota @Double n))
     ]
   ,bgroup "hmatrix"
     [bench "sum(+) Double [1e6]" $
@@ -94,5 +103,20 @@ main = defaultMain
       let n = 1_000_000
       in nf (\a -> LA.sumElements a)
             (LA.linspace @Double n (0.0, fromIntegral (n - 1)))
+    ,bench "dotprod Double [1e6]" $
+      let n = 1_000_000
+      in nf (\(a, b) -> a LA.<.> b)
+            (LA.linspace @Double n (0.0, fromIntegral (n - 1))
+            ,LA.linspace @Double n (fromIntegral (n - 1), 0.0))
+    ]
+  ,bgroup "orthotope"
+    [bench "normalize [1e6]" $
+      let n = 1_000_000
+      in nf (\a -> RS.normalize a)
+            (RS.rev [0] (RS.iota @Double n))
+    ,bench "normalize noop [1e6]" $
+      let n = 1_000_000
+      in nf (\a -> RS.normalize a)
+            (RS.rev [0] (RS.rev [0] (RS.iota @Double n)))
     ]
   ]
diff --git a/ox-arrays.cabal b/ox-arrays.cabal
index 8e4b710..9153336 100644
--- a/ox-arrays.cabal
+++ b/ox-arrays.cabal
@@ -100,6 +100,7 @@ benchmark bench
     ox-arrays,
     base,
     hmatrix,
+    orthotope,
     tasty-bench
   hs-source-dirs: bench
   default-language: Haskell2010