防止在Criterion基准中缓存计算

Question

下面的代码(Reid Barton在Criterion中建议导致内存消耗爆炸,没有看到CAF)的基准时间与 使用优化编译时的比例成比例.然而,使用优化似乎导致基准时间独立于.核心中的结果是缓存的结果,以及我可以做些什么来阻止它被缓存？numO0O3num

代码是:

{-# OPTIONS_GHC -fno-cse #-}
{-# LANGUAGE BangPatterns #-}
module Main where
import Criterion.Main
import Data.List

num :: Int
num = 100000000

lst :: a -> [Int]
lst _ = [1,2..num]

myadd :: Int -> Int -> Int
myadd !x !y = let !result = x + y in
  result

mysum = foldl' myadd 0

main :: IO ()
main = defaultMain [
  bgroup "summation" 
    [bench "mysum" $ whnf (mysum . lst) ()]
  ]

核心是:

main7
main7 = unpackCString# "mysum"#

main8
main8 = unpackCString# "summation"#

Rec {
$wlgo
$wlgo =
  \ ww_s6vW w_s6vT ->
    case w_s6vT of _ {
      [] -> ww_s6vW;
      : x_a4dz xs_a4dA ->
    case x_a4dz of _ { I# ipv_s4d4 ->
    $wlgo (+# ww_s6vW ipv_s4d4) xs_a4dA
    }
    }
end Rec }

lst1
lst1 = efdtInt 1 2 100000000

lvl_r6yu
lvl_r6yu = case $wlgo 0 lst1 of ww_s6w5 { __DEFAULT -> I# ww_s6w5 }

Rec {
main_$s$wa
main_$s$wa =
  \ sc_s6xB sc1_s6xC sc2_s6xD ->
    case tagToEnum# (<=# sc1_s6xC 0) of _ {
      False ->
    case seq# lvl_r6yu sc2_s6xD of _ { (# ipv_a4BO, ipv1_a4BP #) ->
    main_$s$wa sc_s6xB (-# sc1_s6xC 1) ipv_a4BO
    };
      True -> (# sc2_s6xD, () #)
    }
end Rec }

main6
main6 =
  \ w_s6w9 w1_s6wa ->
    case w_s6w9 of _ { I64# ww1_s6wd ->
    main_$s$wa () ww1_s6wd w1_s6wa
    }

main5
main5 = Benchmark main7 (main6 `cast` ...)

main4
main4 = : main5 ([])

main3
main3 = BenchGroup main8 main4

main2
main2 = : main3 ([])

main1
main1 = \ eta_B1 -> defaultMain2 defaultConfig main2 eta_B1

main9
main9 = \ eta_B1 -> runMainIO1 (main1 `cast` ...) eta_B1

lst
lst = \ @ a_a40V _ -> lst1

main
main = main1 `cast` ...

myadd
myadd =
  \ x_a3Io y_a3Ip ->
    case x_a3Io of _ { I# ipv_s4d1 ->
    case y_a3Ip of _ { I# ipv1_s4d4 -> I# (+# ipv_s4d1 ipv1_s4d4) }
    }

mysum
mysum =
  \ w_s6w2 ->
    case $wlgo 0 w_s6w2 of ww_s6w5 { __DEFAULT -> I# ww_s6w5 }

num
num = I# 100000000

main
main = main9 `cast` ...

我在哪里标记-ddump-simpl -fforce-recomp -O3 -dsuppress-all 到ghc --make -no-link ...调用的命令的末尾cabal build.我使用标准1.1.0.0和GHC版本7.8.3.

Answer 1

结果将缓存在您的lvl_r6yu.你可以看到它lst1被[0..num]提升到顶层,从中$wlgo 0 lst1可以看出求和的结果也被提升了.

如果我们添加顶级定义foo = mysum . lst,然后查看核心,则更容易看到发生了什么foo.你可以看到那foo是一个返回求和结果的常数函数.

如果我们添加{-# OPTIONS -fno-full-laziness #-},那么子表达式将不会被取消,因此基准测试将按预期工作.

一般来说,当criterion通过提供的参数来控制评估时,这是一个好主意whnf.在我们的情况下:

bench "mysum" $ whnf (\size -> mysum [1..size]) num

无论优化还是提升,这都可以正常工作.