Add the binary-trees shootout benchmark
authorJohan Tibell <johan.tibell@gmail.com>
Tue, 5 Feb 2013 21:01:52 +0000 (13:01 -0800)
committerJohan Tibell <johan.tibell@gmail.com>
Tue, 5 Feb 2013 21:01:52 +0000 (13:01 -0800)
.gitignore
shootout/Makefile
shootout/binary-trees/Main.hs [new file with mode: 0644]
shootout/binary-trees/Makefile [new file with mode: 0644]
shootout/binary-trees/binary-trees.faststdout [new file with mode: 0644]
shootout/binary-trees/binary-trees.slowstdout [new file with mode: 0644]
shootout/binary-trees/binary-trees.stdout [new file with mode: 0644]

index 32e5c30..8d0effb 100644 (file)
@@ -51,6 +51,7 @@ real/scs/scs
 real/symalg/symalg
 real/veritas/veritas
 
+shootout/binary-trees/binary-trees
 shootout/fannkuch-redux/fannkuch-redux
 shootout/pidigits/pidigits
 shootout/spectral-norm/spectral-norm
index fe0072a..16733c9 100644 (file)
@@ -1,7 +1,7 @@
 TOP = ..
 include $(TOP)/mk/boilerplate.mk
 
-SUBDIRS = fannkuch-redux pidigits spectral-norm
+SUBDIRS = binary-trees fannkuch-redux pidigits spectral-norm
 
 include $(TOP)/mk/target.mk
 
diff --git a/shootout/binary-trees/Main.hs b/shootout/binary-trees/Main.hs
new file mode 100644 (file)
index 0000000..7912dd4
--- /dev/null
@@ -0,0 +1,74 @@
+--
+-- The Computer Language Benchmarks Game
+-- http://benchmarksgame.alioth.debian.org/
+--
+-- Contributed by Don Stewart
+-- Parallelized by Louis Wasserman
+
+import System.Environment
+import Control.Monad
+import System.Mem
+import Data.Bits
+import Text.Printf
+import GHC.Conc
+
+--
+-- an artificially strict tree.
+--
+-- normally you would ensure the branches are lazy, but this benchmark
+-- requires strict allocation.
+--
+data Tree = Nil | Node !Int !Tree !Tree
+
+minN = 4
+
+io s n t = printf "%s of depth %d\t check: %d\n" s n t
+
+main = do
+    n <- getArgs >>= readIO . head
+    let maxN     = max (minN + 2) n
+        stretchN = maxN + 1
+    -- stretch memory tree
+    let c = {-# SCC "stretch" #-} check (make 0 stretchN)
+    io "stretch tree" stretchN c
+
+    -- allocate a long lived tree
+    let !long    = make 0 maxN
+
+    -- allocate, walk, and deallocate many bottom-up binary trees
+    let vs = depth minN maxN
+    mapM_ (\((m,d,i)) -> io (show m ++ "\t trees") d i) vs
+
+    -- confirm the the long-lived binary tree still exists
+    io "long lived tree" maxN (check long)
+
+-- generate many trees
+depth :: Int -> Int -> [(Int,Int,Int)]
+depth d m
+    | d <= m    = let 
+       s = sumT d n 0
+       rest = depth (d+2) m
+       in s `par` ((2*n,d,s) : rest)
+    | otherwise = []
+  where n = bit (m - d + minN)
+
+-- allocate and check lots of trees
+sumT :: Int -> Int -> Int -> Int       
+sumT d 0 t = t
+sumT  d i t = a `par` b `par` sumT d (i-1) ans
+  where a = check (make i    d)
+        b = check (make (-i) d)
+        ans = a + b + t
+
+check = check' True 0
+
+-- traverse the tree, counting up the nodes
+check' :: Bool -> Int -> Tree -> Int
+check' !b !z Nil          = z
+check' b z (Node i l r)          = check' (not b) (check' b (if b then z+i else z-i) l) r
+
+-- build a tree
+make :: Int -> Int -> Tree
+make i 0 = Node i Nil Nil
+make i d = Node i (make (i2-1) d2) (make i2 d2)
+  where i2 = 2*i; d2 = d-1
diff --git a/shootout/binary-trees/Makefile b/shootout/binary-trees/Makefile
new file mode 100644 (file)
index 0000000..8a27f6d
--- /dev/null
@@ -0,0 +1,12 @@
+TOP = ../..
+include $(TOP)/mk/boilerplate.mk
+include $(TOP)/mk/target.mk
+
+FAST_OPTS = 12
+NORM_OPTS = 16
+SLOW_OPTS = 20  # official shootout setting
+
+# The benchmark game also uses -fllvm, which we can't since it might
+# not be available on the developer's machine.
+HC_OPTS += -XBangPatterns -O2 -funbox-strict-fields
+SRC_RUNTEST_OPTS += +RTS -K128M -H -RTS
diff --git a/shootout/binary-trees/binary-trees.faststdout b/shootout/binary-trees/binary-trees.faststdout
new file mode 100644 (file)
index 0000000..9dfe135
--- /dev/null
@@ -0,0 +1,7 @@
+stretch tree of depth 13        check: -1
+8192    trees of depth 4        check: -8192
+2048    trees of depth 6        check: -2048
+512     trees of depth 8        check: -512
+128     trees of depth 10       check: -128
+32      trees of depth 12       check: -32
+long lived tree of depth 12     check: -1
diff --git a/shootout/binary-trees/binary-trees.slowstdout b/shootout/binary-trees/binary-trees.slowstdout
new file mode 100644 (file)
index 0000000..897eba5
--- /dev/null
@@ -0,0 +1,11 @@
+stretch tree of depth 21        check: -1
+2097152         trees of depth 4        check: -2097152
+524288  trees of depth 6        check: -524288
+131072  trees of depth 8        check: -131072
+32768   trees of depth 10       check: -32768
+8192    trees of depth 12       check: -8192
+2048    trees of depth 14       check: -2048
+512     trees of depth 16       check: -512
+128     trees of depth 18       check: -128
+32      trees of depth 20       check: -32
+long lived tree of depth 20     check: -1
diff --git a/shootout/binary-trees/binary-trees.stdout b/shootout/binary-trees/binary-trees.stdout
new file mode 100644 (file)
index 0000000..696bd5c
--- /dev/null
@@ -0,0 +1,9 @@
+stretch tree of depth 17        check: -1
+131072  trees of depth 4        check: -131072
+32768   trees of depth 6        check: -32768
+8192    trees of depth 8        check: -8192
+2048    trees of depth 10       check: -2048
+512     trees of depth 12       check: -512
+128     trees of depth 14       check: -128
+32      trees of depth 16       check: -32
+long lived tree of depth 16     check: -1