Использование GHC API для компиляции программы с оптимизацией

Question

Я хочу скомпилировать модуль 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 применять полную оптимизацию?

Answer 1

Вам нужно включить оптимизацию, которую вы хотите запустить в DynFlags,

Вы можете сделать это, либо указав уровень оптимизации (0..2) и установив оптимизации на его основе (updOptLevel) или путем включения отдельных оптимизаций, таких как Opt_Specialise:

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