我使用 rpar/rseq 關注了 Simon Marlow 關于并行 haskell 的書(第 1 章)。
下面是代碼(解決魷魚游戲橋模擬):
{-# LANGUAGE FlexibleContexts #-}
import Control.DeepSeq (force)
import Control.Exception (evaluate)
import Control.Parallel.Strategies
import Data.Array.IO
( IOUArray,
getAssocs,
newListArray,
readArray,
writeArray,
)
import Data.Functor ((<&>))
import System.Environment (getArgs)
import System.Random (randomRIO)
game ::
Int -> -- number of steps
Int -> -- number of glass at each step
Int -> -- number of players
IO Int -- return the number of survivors
game totalStep totalGlass = go 1 totalGlass
where
go currentStep currentGlass numSurvivors
| numSurvivors == 0 || currentStep > totalStep = return numSurvivors
| otherwise = do
r <- randomRIO (1, currentGlass)
if r == 1
then go (currentStep 1) totalGlass numSurvivors
else go currentStep (currentGlass - 1) (numSurvivors - 1)
simulate :: Int -> IO Int -> IO [(Int, Int)]
simulate n game =
(newListArray (0, 16) (replicate 17 0) :: IO (IOUArray Int Int))
>>= go 1
>>= getAssocs
where
go i marr
| i <= n = do
r <- game
readArray marr r >>= writeArray marr r . ( 1)
go (i 1) marr
| otherwise = return marr
main1 :: IO ()
main1 = do
[n, steps, glassNum, playNum] <- getArgs <&> Prelude.map read
res <- simulate n (game steps glassNum playNum)
mapM_ print res
main2 :: IO ()
main2 = do
putStrLn "Running main2"
[n, steps, glassNum, playNum] <- getArgs <&> Prelude.map read
res <- runEval $ do
r1 <- rpar $ simulate (div n 2) (game steps glassNum playNum) >>= evaluate . force
r2 <- rpar $ simulate (div n 2) (game steps glassNum playNum) >>= evaluate . force
rseq r1
rseq r2
return $
(\l1 l2 -> zipWith (\e1 e2 -> (fst e1, snd e1 snd e2)) l1 l2)
<$> r1
<*> r2
mapM_ print res
main = main2
對于 main2,我已經使用以下方法進行編譯:
ghc -O2 -threaded ./squid.hs
并運行為:
./squid 10000000 18 2 16 RTS -N2
我不明白為什么 main1 比 main2 快,而 main2 在其中具有并行性。
如果這是并行性的正確使用,誰能給我一些關于我的代碼的評論?
uj5u.com熱心網友回復:
您實際上并沒有使用任何并行性。你寫
r1 <- rpar $ simulate (div n 2) (game steps glassNum playNum) >>= evaluate . force
這會觸發一個執行緒來評估一個IO動作,而不是運行它。那沒用。
由于您simulate的本質上是pure,您應該通過交換適當的陣列型別等將其從 轉換IO為ST s。然后您就可以rpar (runST $ simulate ...)并實際并行作業。我認為這些force呼叫在背景關系中沒有用/不合適;他們會更快地釋放陣列,但成本很高。
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/342912.html
標籤:哈斯克尔
上一篇:如何捕獲實體化特定類的所有例外?
