1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
{-# OPTIONS -fbang-patterns #-}
-- compile as:
-- ghc --make -O2 -optc-O2 -fvia-C benchmarks.hs
-- ghc --make -O benchmarks.hs
import Numeric.LinearAlgebra
import System.Time
import System.CPUTime
import Text.Printf
import Data.List(foldl1')
time act = do
t0 <- getCPUTime
act
t1 <- getCPUTime
printf "%.3f s CPU\n" $ (fromIntegral (t1 - t0) / (10^12 :: Double)) :: IO ()
--------------------------------------------------------------------------------
main = sequence_ [bench1,bench2,bench3]
w :: Vector Double
w = constant 1 30000000
bench1 = do
putStrLn "Sum of a vector with 30M doubles:"
print$ vectorMax w -- evaluate it
time $ printf " BLAS: %.2f: " $ sumVB w
time $ printf " Haskell: %.2f: " $ sumVH w
time $ printf " BLAS: %.2f: " $ sumVB w
time $ printf " Haskell: %.2f: " $ sumVH w
sumVB v = constant 1 (dim v) <.> v
sumVH v = go (d - 1) 0
where
d = dim v
go :: Int -> Double -> Double
go 0 s = s + (v @> 0)
go !j !s = go (j - 1) (s + (v @> j))
--------------------------------------------------------------------------------
bench2 = do
putStrLn "-------------------------------------------------------"
putStrLn "Multiplication of 1M different 3x3 matrices:"
-- putStrLn "from [[]]"
-- time $ print $ fun (10^6) rot'
-- putStrLn "from []"
time $ print $ manymult (10^6) rot
print $ cos (10^6/2)
rot' :: Double -> Matrix Double
rot' a = matrix [[ c,0,s],
[ 0,1,0],
[-s,0,c]]
where c = cos a
s = sin a
matrix = fromLists
rot :: Double -> Matrix Double
rot a = (3><3) [ c,0,s
, 0,1,0
,-s,0,c ]
where c = cos a
s = sin a
manymult n r = foldl1' (<>) (map r angles)
where angles = toList $ linspace n (0,1)
--------------------------------------------------------------------------------
bench3 = do
putStrLn "-------------------------------------------------------"
putStrLn "foldVector:"
let v = flatten $ ident 555 :: Vector Double
print $ vectorMax v -- evaluate it
let getPos k s = if k `rem` 555 < 200 && v@>k > 0 then k:s else s
time $ print $ sum $ foldVector getPos [] v
time $ print $ foldVector (\k s -> w@>k + s) 0.0 w
foldVector f s v = go (d - 1) s
where
d = dim v
go 0 s = f (0::Int) s
go !j !s = go (j - 1) (f j s)
|
