使用GHC API进行优化编译程序

时间:2018-06-20 20:56:02

标签: haskell ghc ghc-api

我想在应用优化的情况下将Haskell模块编译为GHC Core,并使用生成的内核输出。但是,当我使用compileToCoreSimplified时,它似乎并没有运行所有常规优化。参加程序:

{-# OPTIONS_GHC -O2 #-}
module LensOpt(pick) where
import Control.Lens
data Record = Record {_field :: String}
field = lens _field $ \r x -> r{_field=x}
pick = Record "test" ^. field

通过ghc -ddump-simple运行时,我得到了最佳输出:

LensOpt.pick1 :: GHC.Prim.Addr#
LensOpt.pick1 = "test"#

pick :: String
pick = GHC.CString.unpackCString# LensOpt.pick1

但是,当使用compileToCoreSimplified进行编译时,会得到输出:

s1 :: Addr#
s1 = "test"#

s2 :: [Char]
s2 = unpackCString# s1

s3 :: Record
s3 = Record s2

pick :: String
pick = case s3 of { Record ds -> ds }

我通过以下方式调用compileToCoreSimplified

import GHC
import GhcPlugins
import GHC.Paths

main = runGhc (Just libdir) $ do
    setTargets []
    dflags <- getSessionDynFlags
    setSessionDynFlags dflags{hscTarget = HscNothing}
    res <- compileToCoreSimplified "LensOpt.hs"
    liftIO $ writeFile "lens_api.txt" $ showSDoc dflags $ ppr res

如何驱动GHC API来应用全部优化?

1 个答案:

答案 0 :(得分:1)

您需要打开要在DynFlags中运行的优化。

您可以通过指定优化级别(0..2)并基于该优化级别(updOptLevel)来设置,也可以通过启用诸如Opt_Specialise之类的单个优化来做到这一点:

main = runGhc (Just libdir) $ do
    setTargets []
    dflags <- getSessionDynFlags
    setSessionDynFlags $ updOptLevel 2 $ dflags{hscTarget = HscNothing}