<?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>nativeGen/RegAlloc/Linear/Base.hs</title> <link type='text/css' rel='stylesheet' href='hscolour.css' /> </head> <body> <pre><a name="line-1"></a> <a name="line-2"></a><span class='hs-comment'>-- | Put common type definitions here to break recursive module dependencies.</span> <a name="line-3"></a> <a name="line-4"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>RegAlloc</span><span class='hs-varop'>.</span><span class='hs-conid'>Linear</span><span class='hs-varop'>.</span><span class='hs-conid'>Base</span> <span class='hs-layout'>(</span> <a name="line-5"></a> <span class='hs-conid'>BlockAssignment</span><span class='hs-layout'>,</span> <a name="line-6"></a> <a name="line-7"></a> <span class='hs-conid'>Loc</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <a name="line-8"></a> <span class='hs-varid'>regsOfLoc</span><span class='hs-layout'>,</span> <a name="line-9"></a> <a name="line-10"></a> <span class='hs-comment'>-- for stats</span> <a name="line-11"></a> <span class='hs-conid'>SpillReason</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <a name="line-12"></a> <span class='hs-conid'>RegAllocStats</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <a name="line-13"></a> <a name="line-14"></a> <span class='hs-comment'>-- the allocator monad</span> <a name="line-15"></a> <span class='hs-conid'>RA_State</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <a name="line-16"></a> <span class='hs-conid'>RegM</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span> <a name="line-17"></a><span class='hs-layout'>)</span> <a name="line-18"></a> <a name="line-19"></a><span class='hs-keyword'>where</span> <a name="line-20"></a> <a name="line-21"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>RegAlloc</span><span class='hs-varop'>.</span><span class='hs-conid'>Linear</span><span class='hs-varop'>.</span><span class='hs-conid'>FreeRegs</span> <a name="line-22"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>RegAlloc</span><span class='hs-varop'>.</span><span class='hs-conid'>Linear</span><span class='hs-varop'>.</span><span class='hs-conid'>StackMap</span> <a name="line-23"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>RegAlloc</span><span class='hs-varop'>.</span><span class='hs-conid'>Liveness</span> <a name="line-24"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Reg</span> <a name="line-25"></a> <a name="line-26"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Outputable</span> <a name="line-27"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Unique</span> <a name="line-28"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>UniqFM</span> <a name="line-29"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>UniqSupply</span> <a name="line-30"></a> <a name="line-31"></a> <a name="line-32"></a><a name="BlockAssignment"></a><span class='hs-comment'>-- | Used to store the register assignment on entry to a basic block.</span> <a name="line-33"></a><a name="BlockAssignment"></a><span class='hs-comment'>-- We use this to handle join points, where multiple branch instructions</span> <a name="line-34"></a><a name="BlockAssignment"></a><span class='hs-comment'>-- target a particular label. We have to insert fixup code to make</span> <a name="line-35"></a><a name="BlockAssignment"></a><span class='hs-comment'>-- the register assignments from the different sources match up.</span> <a name="line-36"></a><a name="BlockAssignment"></a><span class='hs-comment'>--</span> <a name="line-37"></a><a name="BlockAssignment"></a><span class='hs-keyword'>type</span> <span class='hs-conid'>BlockAssignment</span> <a name="line-38"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>BlockMap</span> <span class='hs-layout'>(</span><span class='hs-conid'>FreeRegs</span><span class='hs-layout'>,</span> <span class='hs-conid'>RegMap</span> <span class='hs-conid'>Loc</span><span class='hs-layout'>)</span> <a name="line-39"></a> <a name="line-40"></a> <a name="line-41"></a><span class='hs-comment'>-- | Where a vreg is currently stored</span> <a name="line-42"></a><span class='hs-comment'>-- A temporary can be marked as living in both a register and memory</span> <a name="line-43"></a><span class='hs-comment'>-- (InBoth), for example if it was recently loaded from a spill location.</span> <a name="line-44"></a><span class='hs-comment'>-- This makes it cheap to spill (no save instruction required), but we</span> <a name="line-45"></a><span class='hs-comment'>-- have to be careful to turn this into InReg if the value in the</span> <a name="line-46"></a><span class='hs-comment'>-- register is changed.</span> <a name="line-47"></a> <a name="line-48"></a><a name="Loc"></a><span class='hs-comment'>-- This is also useful when a temporary is about to be clobbered. We</span> <a name="line-49"></a><a name="Loc"></a><span class='hs-comment'>-- save it in a spill location, but mark it as InBoth because the current</span> <a name="line-50"></a><a name="Loc"></a><span class='hs-comment'>-- instruction might still want to read it.</span> <a name="line-51"></a><a name="Loc"></a><span class='hs-comment'>--</span> <a name="line-52"></a><a name="Loc"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>Loc</span> <a name="line-53"></a> <span class='hs-comment'>-- | vreg is in a register</span> <a name="line-54"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>InReg</span> <span class='hs-comment'>{-# UNPACK #-}</span> <span class='hs-varop'>!</span><span class='hs-conid'>RealReg</span> <a name="line-55"></a> <a name="line-56"></a> <span class='hs-comment'>-- | vreg is held in a stack slot</span> <a name="line-57"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>InMem</span> <span class='hs-comment'>{-# UNPACK #-}</span> <span class='hs-varop'>!</span><span class='hs-conid'>StackSlot</span> <a name="line-58"></a> <a name="line-59"></a> <a name="line-60"></a> <span class='hs-comment'>-- | vreg is held in both a register and a stack slot</span> <a name="line-61"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>InBoth</span> <span class='hs-comment'>{-# UNPACK #-}</span> <span class='hs-varop'>!</span><span class='hs-conid'>RealReg</span> <a name="line-62"></a> <span class='hs-comment'>{-# UNPACK #-}</span> <span class='hs-varop'>!</span><span class='hs-conid'>StackSlot</span> <a name="line-63"></a> <span class='hs-keyword'>deriving</span> <span class='hs-layout'>(</span><span class='hs-conid'>Eq</span><span class='hs-layout'>,</span> <span class='hs-conid'>Show</span><span class='hs-layout'>,</span> <span class='hs-conid'>Ord</span><span class='hs-layout'>)</span> <a name="line-64"></a> <a name="line-65"></a><span class='hs-keyword'>instance</span> <span class='hs-conid'>Outputable</span> <span class='hs-conid'>Loc</span> <span class='hs-keyword'>where</span> <a name="line-66"></a> <span class='hs-varid'>ppr</span> <span class='hs-varid'>l</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>text</span> <span class='hs-layout'>(</span><span class='hs-varid'>show</span> <span class='hs-varid'>l</span><span class='hs-layout'>)</span> <a name="line-67"></a> <a name="line-68"></a> <a name="line-69"></a><a name="regsOfLoc"></a><span class='hs-comment'>-- | Get the reg numbers stored in this Loc.</span> <a name="line-70"></a><span class='hs-definition'>regsOfLoc</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Loc</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>RealReg</span><span class='hs-keyglyph'>]</span> <a name="line-71"></a><span class='hs-definition'>regsOfLoc</span> <span class='hs-layout'>(</span><span class='hs-conid'>InReg</span> <span class='hs-varid'>r</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>r</span><span class='hs-keyglyph'>]</span> <a name="line-72"></a><span class='hs-definition'>regsOfLoc</span> <span class='hs-layout'>(</span><span class='hs-conid'>InBoth</span> <span class='hs-varid'>r</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>r</span><span class='hs-keyglyph'>]</span> <a name="line-73"></a><span class='hs-definition'>regsOfLoc</span> <span class='hs-layout'>(</span><span class='hs-conid'>InMem</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>[]</span> <a name="line-74"></a> <a name="line-75"></a> <a name="line-76"></a><a name="SpillReason"></a><span class='hs-comment'>-- | Reasons why instructions might be inserted by the spiller.</span> <a name="line-77"></a><a name="SpillReason"></a><span class='hs-comment'>-- Used when generating stats for -ddrop-asm-stats.</span> <a name="line-78"></a><a name="SpillReason"></a><span class='hs-comment'>--</span> <a name="line-79"></a><a name="SpillReason"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>SpillReason</span> <a name="line-80"></a> <span class='hs-comment'>-- | vreg was spilled to a slot so we could use its</span> <a name="line-81"></a> <span class='hs-comment'>-- current hreg for another vreg</span> <a name="line-82"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>SpillAlloc</span> <span class='hs-varop'>!</span><span class='hs-conid'>Unique</span> <a name="line-83"></a> <a name="line-84"></a> <span class='hs-comment'>-- | vreg was moved because its hreg was clobbered</span> <a name="line-85"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>SpillClobber</span> <span class='hs-varop'>!</span><span class='hs-conid'>Unique</span> <a name="line-86"></a> <a name="line-87"></a> <span class='hs-comment'>-- | vreg was loaded from a spill slot</span> <a name="line-88"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>SpillLoad</span> <span class='hs-varop'>!</span><span class='hs-conid'>Unique</span> <a name="line-89"></a> <a name="line-90"></a> <span class='hs-comment'>-- | reg-reg move inserted during join to targets</span> <a name="line-91"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>SpillJoinRR</span> <span class='hs-varop'>!</span><span class='hs-conid'>Unique</span> <a name="line-92"></a> <a name="line-93"></a> <span class='hs-comment'>-- | reg-mem move inserted during join to targets</span> <a name="line-94"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>SpillJoinRM</span> <span class='hs-varop'>!</span><span class='hs-conid'>Unique</span> <a name="line-95"></a> <a name="line-96"></a> <a name="line-97"></a><a name="RegAllocStats"></a><span class='hs-comment'>-- | Used to carry interesting stats out of the register allocator.</span> <a name="line-98"></a><a name="RegAllocStats"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>RegAllocStats</span> <a name="line-99"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>RegAllocStats</span> <a name="line-100"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>ra_spillInstrs</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>UniqFM</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Int</span><span class='hs-keyglyph'>]</span> <span class='hs-layout'>}</span> <a name="line-101"></a> <a name="line-102"></a> <a name="line-103"></a><a name="RA_State"></a><span class='hs-comment'>-- | The register alloctor state</span> <a name="line-104"></a><a name="RA_State"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>RA_State</span> <a name="line-105"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>RA_State</span> <a name="line-106"></a> <a name="line-107"></a> <span class='hs-layout'>{</span> <a name="line-108"></a> <span class='hs-comment'>-- | the current mapping from basic blocks to </span> <a name="line-109"></a> <span class='hs-comment'>-- the register assignments at the beginning of that block.</span> <a name="line-110"></a> <span class='hs-varid'>ra_blockassig</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>BlockAssignment</span> <a name="line-111"></a> <a name="line-112"></a> <span class='hs-comment'>-- | free machine registers</span> <a name="line-113"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>ra_freeregs</span> <span class='hs-keyglyph'>::</span> <span class='hs-comment'>{-#UNPACK#-}</span><span class='hs-varop'>!</span><span class='hs-conid'>FreeRegs</span> <a name="line-114"></a> <a name="line-115"></a> <span class='hs-comment'>-- | assignment of temps to locations</span> <a name="line-116"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>ra_assig</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>RegMap</span> <span class='hs-conid'>Loc</span> <a name="line-117"></a> <a name="line-118"></a> <span class='hs-comment'>-- | current stack delta</span> <a name="line-119"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>ra_delta</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int</span> <a name="line-120"></a> <a name="line-121"></a> <span class='hs-comment'>-- | free stack slots for spilling</span> <a name="line-122"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>ra_stack</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>StackMap</span> <a name="line-123"></a> <a name="line-124"></a> <span class='hs-comment'>-- | unique supply for generating names for join point fixup blocks.</span> <a name="line-125"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>ra_us</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>UniqSupply</span> <a name="line-126"></a> <a name="line-127"></a> <span class='hs-comment'>-- | Record why things were spilled, for -ddrop-asm-stats.</span> <a name="line-128"></a> <span class='hs-comment'>-- Just keep a list here instead of a map of regs -> reasons.</span> <a name="line-129"></a> <span class='hs-comment'>-- We don't want to slow down the allocator if we're not going to emit the stats.</span> <a name="line-130"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>ra_spills</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>SpillReason</span><span class='hs-keyglyph'>]</span> <span class='hs-layout'>}</span> <a name="line-131"></a> <a name="line-132"></a> <a name="line-133"></a><a name="RegM"></a><span class='hs-comment'>-- | The register allocator monad type.</span> <a name="line-134"></a><a name="RegM"></a><span class='hs-keyword'>newtype</span> <span class='hs-conid'>RegM</span> <span class='hs-varid'>a</span> <a name="line-135"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>RegM</span> <span class='hs-layout'>{</span> <span class='hs-varid'>unReg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>RA_State</span> <span class='hs-keyglyph'>-></span> <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-conid'>RA_State</span><span class='hs-layout'>,</span> <span class='hs-varid'>a</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span> <a name="line-136"></a> <a name="line-137"></a> </pre></body> </html>