<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd"> <html> <head> <!-- Generated by HsColour, http://www.cs.york.ac.uk/fp/darcs/hscolour/ --> <title>simplStg/SimplStg.lhs</title> <link type='text/css' rel='stylesheet' href='hscolour.css' /> </head> <body> % % (c) The GRASP/AQUA Project, Glasgow University, 1993-1998 % \section[SimplStg]{Driver for simplifying @STG@ programs} \begin{code} <pre><a name="line-1"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>SimplStg</span> <span class='hs-layout'>(</span> <span class='hs-varid'>stg2stg</span> <span class='hs-layout'>)</span> <span class='hs-keyword'>where</span> <a name="line-2"></a> <a name="line-3"></a><span class='hs-cpp'>#include "HsVersions.h"</span> <a name="line-4"></a> <a name="line-5"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>StgSyn</span> <a name="line-6"></a> <a name="line-7"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CostCentre</span> <span class='hs-layout'>(</span> <span class='hs-conid'>CollectedCCs</span> <span class='hs-layout'>)</span> <a name="line-8"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>SCCfinal</span> <span class='hs-layout'>(</span> <span class='hs-varid'>stgMassageForProfiling</span> <span class='hs-layout'>)</span> <a name="line-9"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>StgLint</span> <span class='hs-layout'>(</span> <span class='hs-varid'>lintStgBindings</span> <span class='hs-layout'>)</span> <a name="line-10"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>StgStats</span> <span class='hs-layout'>(</span> <span class='hs-varid'>showStgStats</span> <span class='hs-layout'>)</span> <a name="line-11"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>SRT</span> <span class='hs-layout'>(</span> <span class='hs-varid'>computeSRTs</span> <span class='hs-layout'>)</span> <a name="line-12"></a> <a name="line-13"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>DynFlags</span> <span class='hs-layout'>(</span> <span class='hs-conid'>DynFlags</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-conid'>DynFlag</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-varid'>dopt</span><span class='hs-layout'>,</span> <span class='hs-conid'>StgToDo</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <a name="line-14"></a> <span class='hs-varid'>getStgToDo</span> <span class='hs-layout'>)</span> <a name="line-15"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Id</span> <span class='hs-layout'>(</span> <span class='hs-conid'>Id</span> <span class='hs-layout'>)</span> <a name="line-16"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Module</span> <span class='hs-layout'>(</span> <span class='hs-conid'>Module</span> <span class='hs-layout'>)</span> <a name="line-17"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>ErrUtils</span> <span class='hs-layout'>(</span> <span class='hs-varid'>doIfSet_dyn</span><span class='hs-layout'>,</span> <span class='hs-varid'>dumpIfSet_dyn</span><span class='hs-layout'>,</span> <span class='hs-varid'>showPass</span> <span class='hs-layout'>)</span> <a name="line-18"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>UniqSupply</span> <span class='hs-layout'>(</span> <span class='hs-varid'>mkSplitUniqSupply</span><span class='hs-layout'>,</span> <span class='hs-varid'>splitUniqSupply</span> <span class='hs-layout'>)</span> <a name="line-19"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Outputable</span> </pre>\end{code} \begin{code} <pre><a name="line-1"></a><a name="stg2stg"></a><span class='hs-definition'>stg2stg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>DynFlags</span> <span class='hs-comment'>-- includes spec of what stg-to-stg passes to do</span> <a name="line-2"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Module</span> <span class='hs-comment'>-- module name (profiling only)</span> <a name="line-3"></a> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>StgBinding</span><span class='hs-keyglyph'>]</span> <span class='hs-comment'>-- input...</span> <a name="line-4"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-layout'>(</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>StgBinding</span><span class='hs-layout'>,</span><span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>Id</span><span class='hs-layout'>,</span><span class='hs-keyglyph'>[</span><span class='hs-conid'>Id</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <span class='hs-comment'>-- output program...</span> <a name="line-5"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>CollectedCCs</span><span class='hs-layout'>)</span> <span class='hs-comment'>-- cost centre information (declared and used)</span> <a name="line-6"></a> <a name="line-7"></a><span class='hs-definition'>stg2stg</span> <span class='hs-varid'>dflags</span> <span class='hs-varid'>module_name</span> <span class='hs-varid'>binds</span> <a name="line-8"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <span class='hs-layout'>{</span> <span class='hs-varid'>showPass</span> <span class='hs-varid'>dflags</span> <span class='hs-str'>"Stg2Stg"</span> <a name="line-9"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>us</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mkSplitUniqSupply</span> <span class='hs-chr'>'g'</span> <a name="line-10"></a> <a name="line-11"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>doIfSet_dyn</span> <span class='hs-varid'>dflags</span> <span class='hs-conid'>Opt_D_verbose_stg2stg</span> <a name="line-12"></a> <span class='hs-layout'>(</span><span class='hs-varid'>printDump</span> <span class='hs-layout'>(</span><span class='hs-varid'>text</span> <span class='hs-str'>"VERBOSE STG-TO-STG:"</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-13"></a> <a name="line-14"></a> <span class='hs-layout'>;</span> <span class='hs-layout'>(</span><span class='hs-varid'>binds'</span><span class='hs-layout'>,</span> <span class='hs-varid'>us'</span><span class='hs-layout'>,</span> <span class='hs-varid'>ccs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>end_pass</span> <span class='hs-varid'>us</span> <span class='hs-str'>"Stg2Stg"</span> <span class='hs-layout'>(</span><span class='hs-conid'>[]</span><span class='hs-layout'>,</span><span class='hs-conid'>[]</span><span class='hs-layout'>,</span><span class='hs-conid'>[]</span><span class='hs-layout'>)</span> <span class='hs-varid'>binds</span> <a name="line-15"></a> <a name="line-16"></a> <span class='hs-comment'>-- Do the main business!</span> <a name="line-17"></a> <span class='hs-layout'>;</span> <span class='hs-layout'>(</span><span class='hs-varid'>processed_binds</span><span class='hs-layout'>,</span> <span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-varid'>cost_centres</span><span class='hs-layout'>)</span> <a name="line-18"></a> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>foldl_mn</span> <span class='hs-varid'>do_stg_pass</span> <span class='hs-layout'>(</span><span class='hs-varid'>binds'</span><span class='hs-layout'>,</span> <span class='hs-varid'>us'</span><span class='hs-layout'>,</span> <span class='hs-varid'>ccs</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>getStgToDo</span> <span class='hs-varid'>dflags</span><span class='hs-layout'>)</span> <a name="line-19"></a> <a name="line-20"></a> <span class='hs-layout'>;</span> <span class='hs-keyword'>let</span> <span class='hs-varid'>srt_binds</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>computeSRTs</span> <span class='hs-varid'>processed_binds</span> <a name="line-21"></a> <a name="line-22"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>dumpIfSet_dyn</span> <span class='hs-varid'>dflags</span> <span class='hs-conid'>Opt_D_dump_stg</span> <span class='hs-str'>"STG syntax:"</span> <a name="line-23"></a> <span class='hs-layout'>(</span><span class='hs-varid'>pprStgBindingsWithSRTs</span> <span class='hs-varid'>srt_binds</span><span class='hs-layout'>)</span> <a name="line-24"></a> <a name="line-25"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>srt_binds</span><span class='hs-layout'>,</span> <span class='hs-varid'>cost_centres</span><span class='hs-layout'>)</span> <a name="line-26"></a> <span class='hs-layout'>}</span> <a name="line-27"></a> <a name="line-28"></a> <span class='hs-keyword'>where</span> <a name="line-29"></a> <span class='hs-varid'>stg_linter</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>if</span> <span class='hs-varid'>dopt</span> <span class='hs-conid'>Opt_DoStgLinting</span> <span class='hs-varid'>dflags</span> <a name="line-30"></a> <span class='hs-keyword'>then</span> <span class='hs-varid'>lintStgBindings</span> <a name="line-31"></a> <span class='hs-keyword'>else</span> <span class='hs-layout'>(</span> <span class='hs-keyglyph'>\</span> <span class='hs-sel'>_whodunnit</span> <span class='hs-varid'>binds</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>binds</span> <span class='hs-layout'>)</span> <a name="line-32"></a> <a name="line-33"></a> <span class='hs-comment'>-------------------------------------------</span> <a name="line-34"></a> <span class='hs-varid'>do_stg_pass</span> <span class='hs-layout'>(</span><span class='hs-varid'>binds</span><span class='hs-layout'>,</span> <span class='hs-varid'>us</span><span class='hs-layout'>,</span> <span class='hs-varid'>ccs</span><span class='hs-layout'>)</span> <span class='hs-varid'>to_do</span> <a name="line-35"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>let</span> <a name="line-36"></a> <span class='hs-layout'>(</span><span class='hs-varid'>us1</span><span class='hs-layout'>,</span> <span class='hs-varid'>us2</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>splitUniqSupply</span> <span class='hs-varid'>us</span> <a name="line-37"></a> <span class='hs-keyword'>in</span> <a name="line-38"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>to_do</span> <span class='hs-keyword'>of</span> <a name="line-39"></a> <span class='hs-conid'>D_stg_stats</span> <span class='hs-keyglyph'>-></span> <a name="line-40"></a> <span class='hs-varid'>trace</span> <span class='hs-layout'>(</span><span class='hs-varid'>showStgStats</span> <span class='hs-varid'>binds</span><span class='hs-layout'>)</span> <a name="line-41"></a> <span class='hs-varid'>end_pass</span> <span class='hs-varid'>us2</span> <span class='hs-str'>"StgStats"</span> <span class='hs-varid'>ccs</span> <span class='hs-varid'>binds</span> <a name="line-42"></a> <a name="line-43"></a> <span class='hs-conid'>StgDoMassageForProfiling</span> <span class='hs-keyglyph'>-></span> <a name="line-44"></a> <span class='hs-comment'>{-# SCC "ProfMassage" #-}</span> <a name="line-45"></a> <span class='hs-keyword'>let</span> <a name="line-46"></a> <span class='hs-layout'>(</span><span class='hs-varid'>collected_CCs</span><span class='hs-layout'>,</span> <span class='hs-varid'>binds3</span><span class='hs-layout'>)</span> <a name="line-47"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>stgMassageForProfiling</span> <span class='hs-varid'>dflags</span> <span class='hs-varid'>this_pkg</span> <span class='hs-varid'>module_name</span> <span class='hs-varid'>us1</span> <span class='hs-varid'>binds</span> <a name="line-48"></a> <span class='hs-varid'>this_pkg</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>thisPackage</span> <span class='hs-varid'>dflags</span> <a name="line-49"></a> <span class='hs-keyword'>in</span> <a name="line-50"></a> <span class='hs-varid'>end_pass</span> <span class='hs-varid'>us2</span> <span class='hs-str'>"ProfMassage"</span> <span class='hs-varid'>collected_CCs</span> <span class='hs-varid'>binds3</span> <a name="line-51"></a> <a name="line-52"></a> <span class='hs-varid'>end_pass</span> <span class='hs-varid'>us2</span> <span class='hs-varid'>what</span> <span class='hs-varid'>ccs</span> <span class='hs-varid'>binds2</span> <a name="line-53"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <span class='hs-comment'>-- report verbosely, if required</span> <a name="line-54"></a> <span class='hs-varid'>dumpIfSet_dyn</span> <span class='hs-varid'>dflags</span> <span class='hs-conid'>Opt_D_verbose_stg2stg</span> <span class='hs-varid'>what</span> <a name="line-55"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vcat</span> <span class='hs-layout'>(</span><span class='hs-varid'>map</span> <span class='hs-varid'>ppr</span> <span class='hs-varid'>binds2</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-56"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>linted_binds</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>stg_linter</span> <span class='hs-varid'>what</span> <span class='hs-varid'>binds2</span> <a name="line-57"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>linted_binds</span><span class='hs-layout'>,</span> <span class='hs-varid'>us2</span><span class='hs-layout'>,</span> <span class='hs-varid'>ccs</span><span class='hs-layout'>)</span> <a name="line-58"></a> <span class='hs-comment'>-- return: processed binds</span> <a name="line-59"></a> <span class='hs-comment'>-- UniqueSupply for the next guy to use</span> <a name="line-60"></a> <span class='hs-comment'>-- cost-centres to be declared/registered (specialised)</span> <a name="line-61"></a> <span class='hs-comment'>-- add to description of what's happened (reverse order)</span> <a name="line-62"></a> <a name="line-63"></a><a name="foldl_mn"></a><span class='hs-comment'>-- here so it can be inlined...</span> <a name="line-64"></a><span class='hs-definition'>foldl_mn</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-varid'>b</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-varid'>b</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>b</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>a</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-varid'>b</span> <a name="line-65"></a><span class='hs-definition'>foldl_mn</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>z</span> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>return</span> <span class='hs-varid'>z</span> <a name="line-66"></a><span class='hs-definition'>foldl_mn</span> <span class='hs-varid'>f</span> <span class='hs-varid'>z</span> <span class='hs-layout'>(</span><span class='hs-varid'>x</span><span class='hs-conop'>:</span><span class='hs-varid'>xs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>f</span> <span class='hs-varid'>z</span> <span class='hs-varid'>x</span> <span class='hs-varop'>>>=</span> <span class='hs-keyglyph'>\</span> <span class='hs-varid'>zz</span> <span class='hs-keyglyph'>-></span> <a name="line-67"></a> <span class='hs-varid'>foldl_mn</span> <span class='hs-varid'>f</span> <span class='hs-varid'>zz</span> <span class='hs-varid'>xs</span> </pre>\end{code} </body> </html>