blob: 02252ffbe08de98945da6177bb50d78a258d832d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
{-# OPTIONS_GHC -fno-warn-name-shadowing #-}
module Parallel
( streamsOf,
parMapM,
)
where
import Control.Concurrent.ParallelIO.Local (parallel, withPool)
import GHC.Conc (numCapabilities)
streamsOf :: Int -> [a] -> [[a]]
streamsOf 1 xs = [xs]
streamsOf n xs
| n > 0 = [everyN k xs | k <- [0 .. n - 1]]
| otherwise = []
where
everyN k xs = map snd $ filter ((== k) . (`mod` n) . fst) $ zip [0 ..] xs
parMapM :: (a -> IO b) -> [a] -> IO [b]
parMapM f xs = do
withPool numCapabilities $ \pool -> parallel pool (map f xs)
|
