Commit f02a8490 authored by Yiannis Tsiouris's avatar Yiannis Tsiouris

Fix types in Bench module

parent 9e6903ee
...@@ -6,11 +6,14 @@ module Bench( -- sequential benchmarks ...@@ -6,11 +6,14 @@ module Bench( -- sequential benchmarks
, dist, dist_seq , dist, dist_seq
) where ) where
import Control.Distributed.Process (Process, NodeId)
import Data.List (lookup) import Data.List (lookup)
import Data.Maybe (fromMaybe) import Data.Maybe (fromMaybe)
import Prelude hiding (seq) import Prelude hiding (seq)
import MasterWorker (HostInfo(..), MaybeHosts(..), orbit) import MasterWorker (HostInfo(..), MaybeHosts(..),
GenClos, MasterStats, orbit)
import Table (Vertex)
import Utils import Utils
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
...@@ -20,30 +23,40 @@ import Utils ...@@ -20,30 +23,40 @@ import Utils
-- * number of processors P > 0 (per node) -- * number of processors P > 0 (per node)
-- * list of Workers (in short node name format 'name@host') -- * list of Workers (in short node name format 'name@host')
-- sequential orbit computation -- sequential orbit computation
seq :: (Int -> GenClos) -> Int -> Process String
seq generators n = seq generators n =
sz $ orbit (generators n) [0] (Seq (2 * n)) orbit (generators n) [0] (Seq (2 * n)) >>= return . sz . snd
-- parallel orbit computation (par_seq/3 does not spawn image computation) -- parallel orbit computation (par_seq/3 does not spawn image computation)
par :: (Int -> GenClos) -> Int -> Int -> Process String
par generators n p = par generators n p =
sz $ orbit (generators n) [0] orbit (generators n) [0]
(Par (JustOne (p, ((2 * n) `div` p) + 1, 0, True))) (Par (JustOne (p, ((2 * n) `div` p) + 1, 0, True)))
>>= return . sz . snd
par_seq :: (Int -> GenClos) -> Int -> Int -> Process String
par_seq generators n p = par_seq generators n p =
sz $ orbit (generators n) [0] orbit (generators n) [0]
(Par (JustOne (p, ((2 * n) `div` p) + 1, 0, False))) (Par (JustOne (p, ((2 * n) `div` p) + 1, 0, False)))
>>= return . sz . snd
-- distributed orbit computation (dist_seq/4 does not spawn image computation) -- distributed orbit computation (dist_seq/4 does not spawn image computation)
dist :: (Int -> GenClos) -> Int -> Int -> [NodeId] -> Process String
dist generators n p workers = dist generators n p workers =
sz $ orbit (generators n) [0] orbit (generators n) [0]
(Par (Many [(h, p, (2 * n) `div` (w * p) + 1, 1, True) | h <- workers])) (Par (Many [(h, p, (2 * n) `div` (w * p) + 1, 1, True) | h <- workers]))
>>= return . sz . snd
where w = length workers where w = length workers
dist_seq :: (Int -> GenClos) -> Int -> Int -> [NodeId] -> Process String
dist_seq generators n p workers = dist_seq generators n p workers =
sz $ orbit (generators n) [0] orbit (generators n) [0]
(Par (Many [(h, p, (2 * n) `div` (w * p) + 1, 1, False) | h <- workers])) (Par (Many [(h, p, (2 * n) `div` (w * p) + 1, 1, False) | h <- workers]))
>>= return . sz . snd
where w = length workers where w = length workers
sz (_, mainStats : _) = sz :: [MasterStats] -> String
sz (mainStats : _) =
case "size" `lookup` mainStats of case "size" `lookup` mainStats of
Nothing -> "false" Nothing -> "false"
Just s -> "{size," ++ s ++ "}" Just s -> "{size," ++ s ++ "}"
...@@ -5,6 +5,7 @@ ...@@ -5,6 +5,7 @@
module MasterWorker( -- Master module MasterWorker( -- Master
GenClos GenClos
, HostInfo (..) , HostInfo (..)
, MasterStats
, MaybeHosts(..) , MaybeHosts(..)
, orbit , orbit
, get_gens, get_master, get_workers, get_spawn_img_comp , get_gens, get_master, get_workers, get_spawn_img_comp
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment