<?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>vectorise/VectUtils.hs</title> <link type='text/css' rel='stylesheet' href='hscolour.css' /> </head> <body> <pre><a name="line-1"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>VectUtils</span> <span class='hs-layout'>(</span> <a name="line-2"></a> <span class='hs-varid'>collectAnnTypeBinders</span><span class='hs-layout'>,</span> <span class='hs-varid'>collectAnnTypeArgs</span><span class='hs-layout'>,</span> <span class='hs-varid'>isAnnTypeArg</span><span class='hs-layout'>,</span> <a name="line-3"></a> <span class='hs-varid'>collectAnnValBinders</span><span class='hs-layout'>,</span> <a name="line-4"></a> <span class='hs-varid'>dataConTagZ</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkDataConTag</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkDataConTagLit</span><span class='hs-layout'>,</span> <a name="line-5"></a> <a name="line-6"></a> <span class='hs-varid'>newLocalVVar</span><span class='hs-layout'>,</span> <a name="line-7"></a> <a name="line-8"></a> <span class='hs-varid'>mkBuiltinCo</span><span class='hs-layout'>,</span> <span class='hs-varid'>voidType</span><span class='hs-layout'>,</span> <a name="line-9"></a> <span class='hs-varid'>mkPADictType</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkPArrayType</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkPDataType</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkPReprType</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkPArray</span><span class='hs-layout'>,</span> <a name="line-10"></a> <a name="line-11"></a> <span class='hs-varid'>pdataReprTyCon</span><span class='hs-layout'>,</span> <span class='hs-varid'>pdataReprDataCon</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkVScrut</span><span class='hs-layout'>,</span> <a name="line-12"></a> <span class='hs-varid'>prDFunOfTyCon</span><span class='hs-layout'>,</span> <a name="line-13"></a> <span class='hs-varid'>paDictArgType</span><span class='hs-layout'>,</span> <span class='hs-varid'>paDictOfType</span><span class='hs-layout'>,</span> <span class='hs-varid'>paDFunType</span><span class='hs-layout'>,</span> <a name="line-14"></a> <span class='hs-varid'>paMethod</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkPR</span><span class='hs-layout'>,</span> <span class='hs-varid'>replicatePD</span><span class='hs-layout'>,</span> <span class='hs-varid'>emptyPD</span><span class='hs-layout'>,</span> <span class='hs-varid'>packPD</span><span class='hs-layout'>,</span> <a name="line-15"></a> <span class='hs-varid'>combinePD</span><span class='hs-layout'>,</span> <a name="line-16"></a> <span class='hs-varid'>liftPD</span><span class='hs-layout'>,</span> <a name="line-17"></a> <span class='hs-varid'>zipScalars</span><span class='hs-layout'>,</span> <span class='hs-varid'>scalarClosure</span><span class='hs-layout'>,</span> <a name="line-18"></a> <span class='hs-varid'>polyAbstract</span><span class='hs-layout'>,</span> <span class='hs-varid'>polyApply</span><span class='hs-layout'>,</span> <span class='hs-varid'>polyVApply</span><span class='hs-layout'>,</span> <a name="line-19"></a> <span class='hs-varid'>hoistBinding</span><span class='hs-layout'>,</span> <span class='hs-varid'>hoistExpr</span><span class='hs-layout'>,</span> <span class='hs-varid'>hoistPolyVExpr</span><span class='hs-layout'>,</span> <span class='hs-varid'>takeHoisted</span><span class='hs-layout'>,</span> <a name="line-20"></a> <span class='hs-varid'>buildClosure</span><span class='hs-layout'>,</span> <span class='hs-varid'>buildClosures</span><span class='hs-layout'>,</span> <a name="line-21"></a> <span class='hs-varid'>mkClosureApp</span> <a name="line-22"></a><span class='hs-layout'>)</span> <span class='hs-keyword'>where</span> <a name="line-23"></a> <a name="line-24"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VectCore</span> <a name="line-25"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VectMonad</span> <a name="line-26"></a> <a name="line-27"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>MkCore</span> <span class='hs-layout'>(</span> <span class='hs-varid'>mkCoreTup</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkCoreTupTy</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkWildCase</span> <span class='hs-layout'>)</span> <a name="line-28"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CoreSyn</span> <a name="line-29"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CoreUtils</span> <a name="line-30"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Coercion</span> <a name="line-31"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Type</span> <a name="line-32"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>TypeRep</span> <a name="line-33"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>TyCon</span> <a name="line-34"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>DataCon</span> <a name="line-35"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Var</span> <a name="line-36"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>MkId</span> <span class='hs-layout'>(</span> <span class='hs-varid'>unwrapFamInstScrut</span> <span class='hs-layout'>)</span> <a name="line-37"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>TysWiredIn</span> <a name="line-38"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>BasicTypes</span> <span class='hs-layout'>(</span> <span class='hs-conid'>Boxity</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span> <span class='hs-layout'>)</span> <a name="line-39"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Literal</span> <span class='hs-layout'>(</span> <span class='hs-conid'>Literal</span><span class='hs-layout'>,</span> <span class='hs-varid'>mkMachInt</span> <span class='hs-layout'>)</span> <a name="line-40"></a> <a name="line-41"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Outputable</span> <a name="line-42"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>FastString</span> <a name="line-43"></a> <a name="line-44"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Monad</span> <a name="line-45"></a> <a name="line-46"></a> <a name="line-47"></a><a name="collectAnnTypeArgs"></a><span class='hs-definition'>collectAnnTypeArgs</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AnnExpr</span> <span class='hs-varid'>b</span> <span class='hs-varid'>ann</span> <span class='hs-keyglyph'>-></span> <span class='hs-layout'>(</span><span class='hs-conid'>AnnExpr</span> <span class='hs-varid'>b</span> <span class='hs-varid'>ann</span><span class='hs-layout'>,</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-48"></a><span class='hs-definition'>collectAnnTypeArgs</span> <span class='hs-varid'>expr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-varid'>expr</span> <span class='hs-conid'>[]</span> <a name="line-49"></a> <span class='hs-keyword'>where</span> <a name="line-50"></a> <span class='hs-varid'>go</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnApp</span> <span class='hs-varid'>f</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnType</span> <span class='hs-varid'>ty</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-varid'>tys</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-varid'>f</span> <span class='hs-layout'>(</span><span class='hs-varid'>ty</span> <span class='hs-conop'>:</span> <span class='hs-varid'>tys</span><span class='hs-layout'>)</span> <a name="line-51"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>e</span> <span class='hs-varid'>tys</span> <span class='hs-keyglyph'>=</span> <span class='hs-layout'>(</span><span class='hs-varid'>e</span><span class='hs-layout'>,</span> <span class='hs-varid'>tys</span><span class='hs-layout'>)</span> <a name="line-52"></a> <a name="line-53"></a><a name="collectAnnTypeBinders"></a><span class='hs-definition'>collectAnnTypeBinders</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AnnExpr</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>ann</span> <span class='hs-keyglyph'>-></span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>[</span><span class='hs-conid'>Var</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnExpr</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>ann</span><span class='hs-layout'>)</span> <a name="line-54"></a><span class='hs-definition'>collectAnnTypeBinders</span> <span class='hs-varid'>expr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-conid'>[]</span> <span class='hs-varid'>expr</span> <a name="line-55"></a> <span class='hs-keyword'>where</span> <a name="line-56"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>bs</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnLam</span> <span class='hs-varid'>b</span> <span class='hs-varid'>e</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isTyVar</span> <span class='hs-varid'>b</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-layout'>(</span><span class='hs-varid'>b</span><span class='hs-conop'>:</span><span class='hs-varid'>bs</span><span class='hs-layout'>)</span> <span class='hs-varid'>e</span> <a name="line-57"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>bs</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>=</span> <span class='hs-layout'>(</span><span class='hs-varid'>reverse</span> <span class='hs-varid'>bs</span><span class='hs-layout'>,</span> <span class='hs-varid'>e</span><span class='hs-layout'>)</span> <a name="line-58"></a> <a name="line-59"></a><a name="collectAnnValBinders"></a><span class='hs-definition'>collectAnnValBinders</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AnnExpr</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>ann</span> <span class='hs-keyglyph'>-></span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>[</span><span class='hs-conid'>Var</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnExpr</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>ann</span><span class='hs-layout'>)</span> <a name="line-60"></a><span class='hs-definition'>collectAnnValBinders</span> <span class='hs-varid'>expr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-conid'>[]</span> <span class='hs-varid'>expr</span> <a name="line-61"></a> <span class='hs-keyword'>where</span> <a name="line-62"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>bs</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnLam</span> <span class='hs-varid'>b</span> <span class='hs-varid'>e</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isId</span> <span class='hs-varid'>b</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-layout'>(</span><span class='hs-varid'>b</span><span class='hs-conop'>:</span><span class='hs-varid'>bs</span><span class='hs-layout'>)</span> <span class='hs-varid'>e</span> <a name="line-63"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>bs</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>=</span> <span class='hs-layout'>(</span><span class='hs-varid'>reverse</span> <span class='hs-varid'>bs</span><span class='hs-layout'>,</span> <span class='hs-varid'>e</span><span class='hs-layout'>)</span> <a name="line-64"></a> <a name="line-65"></a><a name="isAnnTypeArg"></a><span class='hs-definition'>isAnnTypeArg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AnnExpr</span> <span class='hs-varid'>b</span> <span class='hs-varid'>ann</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Bool</span> <a name="line-66"></a><span class='hs-definition'>isAnnTypeArg</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnType</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>True</span> <a name="line-67"></a><span class='hs-definition'>isAnnTypeArg</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>False</span> <a name="line-68"></a> <a name="line-69"></a><a name="dataConTagZ"></a><span class='hs-definition'>dataConTagZ</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>DataCon</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int</span> <a name="line-70"></a><span class='hs-definition'>dataConTagZ</span> <span class='hs-varid'>con</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>dataConTag</span> <span class='hs-varid'>con</span> <span class='hs-comment'>-</span> <span class='hs-varid'>fIRST_TAG</span> <a name="line-71"></a> <a name="line-72"></a><a name="mkDataConTagLit"></a><span class='hs-definition'>mkDataConTagLit</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>DataCon</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Literal</span> <a name="line-73"></a><span class='hs-definition'>mkDataConTagLit</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkMachInt</span> <span class='hs-varop'>.</span> <span class='hs-varid'>toInteger</span> <span class='hs-varop'>.</span> <span class='hs-varid'>dataConTagZ</span> <a name="line-74"></a> <a name="line-75"></a><a name="mkDataConTag"></a><span class='hs-definition'>mkDataConTag</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>DataCon</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <a name="line-76"></a><span class='hs-definition'>mkDataConTag</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkIntLitInt</span> <span class='hs-varop'>.</span> <span class='hs-varid'>dataConTagZ</span> <a name="line-77"></a> <a name="line-78"></a><a name="splitPrimTyCon"></a><span class='hs-definition'>splitPrimTyCon</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Maybe</span> <span class='hs-conid'>TyCon</span> <a name="line-79"></a><span class='hs-definition'>splitPrimTyCon</span> <span class='hs-varid'>ty</span> <a name="line-80"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-layout'>(</span><span class='hs-varid'>tycon</span><span class='hs-layout'>,</span> <span class='hs-conid'>[]</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>splitTyConApp_maybe</span> <span class='hs-varid'>ty</span> <a name="line-81"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>isPrimTyCon</span> <span class='hs-varid'>tycon</span> <a name="line-82"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>tycon</span> <a name="line-83"></a> <a name="line-84"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Nothing</span> <a name="line-85"></a> <a name="line-86"></a><a name="mkBuiltinTyConApp"></a><span class='hs-definition'>mkBuiltinTyConApp</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Builtins</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>TyCon</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-87"></a><span class='hs-definition'>mkBuiltinTyConApp</span> <span class='hs-varid'>get_tc</span> <span class='hs-varid'>tys</span> <a name="line-88"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-89"></a> <span class='hs-varid'>tc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>get_tc</span> <a name="line-90"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mkTyConApp</span> <span class='hs-varid'>tc</span> <span class='hs-varid'>tys</span> <a name="line-91"></a> <a name="line-92"></a><a name="mkBuiltinTyConApps"></a><span class='hs-definition'>mkBuiltinTyConApps</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Builtins</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>TyCon</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-93"></a><span class='hs-definition'>mkBuiltinTyConApps</span> <span class='hs-varid'>get_tc</span> <span class='hs-varid'>tys</span> <span class='hs-varid'>ty</span> <a name="line-94"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-95"></a> <span class='hs-varid'>tc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>get_tc</span> <a name="line-96"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>foldr</span> <span class='hs-layout'>(</span><span class='hs-varid'>mk</span> <span class='hs-varid'>tc</span><span class='hs-layout'>)</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>tys</span> <a name="line-97"></a> <span class='hs-keyword'>where</span> <a name="line-98"></a> <span class='hs-varid'>mk</span> <span class='hs-varid'>tc</span> <span class='hs-varid'>ty1</span> <span class='hs-varid'>ty2</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkTyConApp</span> <span class='hs-varid'>tc</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>ty1</span><span class='hs-layout'>,</span><span class='hs-varid'>ty2</span><span class='hs-keyglyph'>]</span> <a name="line-99"></a> <a name="line-100"></a><a name="voidType"></a><span class='hs-definition'>voidType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-101"></a><span class='hs-definition'>voidType</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkBuiltinTyConApp</span> <span class='hs-varid'>voidTyCon</span> <span class='hs-conid'>[]</span> <a name="line-102"></a> <a name="line-103"></a><a name="mkClosureTypes"></a><span class='hs-definition'>mkClosureTypes</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-104"></a><span class='hs-definition'>mkClosureTypes</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkBuiltinTyConApps</span> <span class='hs-varid'>closureTyCon</span> <a name="line-105"></a> <a name="line-106"></a><a name="mkPReprType"></a><span class='hs-definition'>mkPReprType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-107"></a><span class='hs-definition'>mkPReprType</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkBuiltinTyConApp</span> <span class='hs-varid'>preprTyCon</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>ty</span><span class='hs-keyglyph'>]</span> <a name="line-108"></a> <a name="line-109"></a><a name="mkPADictType"></a><span class='hs-definition'>mkPADictType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-110"></a><span class='hs-definition'>mkPADictType</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkBuiltinTyConApp</span> <span class='hs-varid'>paTyCon</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>ty</span><span class='hs-keyglyph'>]</span> <a name="line-111"></a> <a name="line-112"></a><a name="mkPArrayType"></a><span class='hs-definition'>mkPArrayType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-113"></a><span class='hs-definition'>mkPArrayType</span> <span class='hs-varid'>ty</span> <a name="line-114"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>tycon</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>splitPrimTyCon</span> <span class='hs-varid'>ty</span> <a name="line-115"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-116"></a> <span class='hs-varid'>r</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>lookupPrimPArray</span> <span class='hs-varid'>tycon</span> <a name="line-117"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>r</span> <span class='hs-keyword'>of</span> <a name="line-118"></a> <span class='hs-conid'>Just</span> <span class='hs-varid'>arr</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mkTyConApp</span> <span class='hs-varid'>arr</span> <span class='hs-conid'>[]</span> <a name="line-119"></a> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>cantVectorise</span> <span class='hs-str'>"Primitive tycon not vectorised"</span> <span class='hs-layout'>(</span><span class='hs-varid'>ppr</span> <span class='hs-varid'>tycon</span><span class='hs-layout'>)</span> <a name="line-120"></a><span class='hs-definition'>mkPArrayType</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkBuiltinTyConApp</span> <span class='hs-varid'>parrayTyCon</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>ty</span><span class='hs-keyglyph'>]</span> <a name="line-121"></a> <a name="line-122"></a><a name="mkPDataType"></a><span class='hs-definition'>mkPDataType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-123"></a><span class='hs-definition'>mkPDataType</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkBuiltinTyConApp</span> <span class='hs-varid'>pdataTyCon</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>ty</span><span class='hs-keyglyph'>]</span> <a name="line-124"></a> <a name="line-125"></a><a name="mkPArray"></a><span class='hs-definition'>mkPArray</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-126"></a><span class='hs-definition'>mkPArray</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>len</span> <span class='hs-varid'>dat</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-127"></a> <span class='hs-varid'>tc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>parrayTyCon</span> <a name="line-128"></a> <span class='hs-keyword'>let</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>dc</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tyConDataCons</span> <span class='hs-varid'>tc</span> <a name="line-129"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mkConApp</span> <span class='hs-varid'>dc</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span> <span class='hs-varid'>ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>len</span><span class='hs-layout'>,</span> <span class='hs-varid'>dat</span><span class='hs-keyglyph'>]</span> <a name="line-130"></a> <a name="line-131"></a><a name="mkBuiltinCo"></a><span class='hs-definition'>mkBuiltinCo</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Builtins</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>TyCon</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Coercion</span> <a name="line-132"></a><span class='hs-definition'>mkBuiltinCo</span> <span class='hs-varid'>get_tc</span> <a name="line-133"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-134"></a> <span class='hs-varid'>tc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>get_tc</span> <a name="line-135"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mkTyConApp</span> <span class='hs-varid'>tc</span> <span class='hs-conid'>[]</span> <a name="line-136"></a> <a name="line-137"></a><a name="pdataReprTyCon"></a><span class='hs-definition'>pdataReprTyCon</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-layout'>(</span><span class='hs-conid'>TyCon</span><span class='hs-layout'>,</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-138"></a><span class='hs-definition'>pdataReprTyCon</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>pdataTyCon</span> <span class='hs-varop'>>>=</span> <span class='hs-layout'>(</span><span class='hs-varop'>`lookupFamInst`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>ty</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-139"></a> <a name="line-140"></a><a name="pdataReprDataCon"></a><span class='hs-definition'>pdataReprDataCon</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-layout'>(</span><span class='hs-conid'>DataCon</span><span class='hs-layout'>,</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-141"></a><span class='hs-definition'>pdataReprDataCon</span> <span class='hs-varid'>ty</span> <a name="line-142"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-143"></a> <span class='hs-layout'>(</span><span class='hs-varid'>tc</span><span class='hs-layout'>,</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>pdataReprTyCon</span> <span class='hs-varid'>ty</span> <a name="line-144"></a> <span class='hs-keyword'>let</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>dc</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tyConDataCons</span> <span class='hs-varid'>tc</span> <a name="line-145"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>dc</span><span class='hs-layout'>,</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <a name="line-146"></a> <a name="line-147"></a><a name="mkVScrut"></a><span class='hs-definition'>mkVScrut</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-layout'>(</span><span class='hs-conid'>CoreExpr</span><span class='hs-layout'>,</span> <span class='hs-conid'>CoreExpr</span><span class='hs-layout'>,</span> <span class='hs-conid'>TyCon</span><span class='hs-layout'>,</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-148"></a><span class='hs-definition'>mkVScrut</span> <span class='hs-layout'>(</span><span class='hs-varid'>ve</span><span class='hs-layout'>,</span> <span class='hs-varid'>le</span><span class='hs-layout'>)</span> <a name="line-149"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-150"></a> <span class='hs-layout'>(</span><span class='hs-varid'>tc</span><span class='hs-layout'>,</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>pdataReprTyCon</span> <span class='hs-varid'>ty</span> <a name="line-151"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>ve</span><span class='hs-layout'>,</span> <span class='hs-varid'>unwrapFamInstScrut</span> <span class='hs-varid'>tc</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varid'>le</span><span class='hs-layout'>,</span> <span class='hs-varid'>tc</span><span class='hs-layout'>,</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <a name="line-152"></a> <span class='hs-keyword'>where</span> <a name="line-153"></a> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>exprType</span> <span class='hs-varid'>ve</span> <a name="line-154"></a> <a name="line-155"></a><a name="prDFunOfTyCon"></a><span class='hs-definition'>prDFunOfTyCon</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>TyCon</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-156"></a><span class='hs-definition'>prDFunOfTyCon</span> <span class='hs-varid'>tycon</span> <a name="line-157"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-conid'>Var</span> <a name="line-158"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>maybeCantVectoriseM</span> <span class='hs-str'>"No PR dictionary for tycon"</span> <span class='hs-layout'>(</span><span class='hs-varid'>ppr</span> <span class='hs-varid'>tycon</span><span class='hs-layout'>)</span> <a name="line-159"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>lookupTyConPR</span> <span class='hs-varid'>tycon</span> <a name="line-160"></a> <a name="line-161"></a><a name="paDictArgType"></a><span class='hs-definition'>paDictArgType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>TyVar</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-layout'>(</span><span class='hs-conid'>Maybe</span> <span class='hs-conid'>Type</span><span class='hs-layout'>)</span> <a name="line-162"></a><span class='hs-definition'>paDictArgType</span> <span class='hs-varid'>tv</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-layout'>(</span><span class='hs-conid'>TyVarTy</span> <span class='hs-varid'>tv</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>tyVarKind</span> <span class='hs-varid'>tv</span><span class='hs-layout'>)</span> <a name="line-163"></a> <span class='hs-keyword'>where</span> <a name="line-164"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>k</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>k'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>kindView</span> <span class='hs-varid'>k</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>k'</span> <a name="line-165"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>ty</span> <span class='hs-layout'>(</span><span class='hs-conid'>FunTy</span> <span class='hs-varid'>k1</span> <span class='hs-varid'>k2</span><span class='hs-layout'>)</span> <a name="line-166"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-167"></a> <span class='hs-varid'>tv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newTyVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"a"</span><span class='hs-layout'>)</span> <span class='hs-varid'>k1</span> <a name="line-168"></a> <span class='hs-varid'>mty1</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>go</span> <span class='hs-layout'>(</span><span class='hs-conid'>TyVarTy</span> <span class='hs-varid'>tv</span><span class='hs-layout'>)</span> <span class='hs-varid'>k1</span> <a name="line-169"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>mty1</span> <span class='hs-keyword'>of</span> <a name="line-170"></a> <span class='hs-conid'>Just</span> <span class='hs-varid'>ty1</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-171"></a> <span class='hs-varid'>mty2</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>go</span> <span class='hs-layout'>(</span><span class='hs-conid'>AppTy</span> <span class='hs-varid'>ty</span> <span class='hs-layout'>(</span><span class='hs-conid'>TyVarTy</span> <span class='hs-varid'>tv</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-varid'>k2</span> <a name="line-172"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>fmap</span> <span class='hs-layout'>(</span><span class='hs-conid'>ForAllTy</span> <span class='hs-varid'>tv</span> <span class='hs-varop'>.</span> <span class='hs-conid'>FunTy</span> <span class='hs-varid'>ty1</span><span class='hs-layout'>)</span> <span class='hs-varid'>mty2</span> <a name="line-173"></a> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>go</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>k2</span> <a name="line-174"></a> <a name="line-175"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>k</span> <a name="line-176"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isLiftedTypeKind</span> <span class='hs-varid'>k</span> <a name="line-177"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-conid'>Just</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkPADictType</span> <span class='hs-varid'>ty</span><span class='hs-layout'>)</span> <a name="line-178"></a> <a name="line-179"></a> <span class='hs-varid'>go</span> <span class='hs-keyword'>_</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>return</span> <span class='hs-conid'>Nothing</span> <a name="line-180"></a> <a name="line-181"></a><a name="paDictOfType"></a><span class='hs-definition'>paDictOfType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-182"></a><span class='hs-definition'>paDictOfType</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>paDictOfTyApp</span> <span class='hs-varid'>ty_fn</span> <span class='hs-varid'>ty_args</span> <a name="line-183"></a> <span class='hs-keyword'>where</span> <a name="line-184"></a> <span class='hs-layout'>(</span><span class='hs-varid'>ty_fn</span><span class='hs-layout'>,</span> <span class='hs-varid'>ty_args</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>splitAppTys</span> <span class='hs-varid'>ty</span> <a name="line-185"></a> <a name="line-186"></a><a name="paDictOfTyApp"></a><span class='hs-definition'>paDictOfTyApp</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-187"></a><span class='hs-definition'>paDictOfTyApp</span> <span class='hs-varid'>ty_fn</span> <span class='hs-varid'>ty_args</span> <a name="line-188"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>ty_fn'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>coreView</span> <span class='hs-varid'>ty_fn</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>paDictOfTyApp</span> <span class='hs-varid'>ty_fn'</span> <span class='hs-varid'>ty_args</span> <a name="line-189"></a><span class='hs-definition'>paDictOfTyApp</span> <span class='hs-layout'>(</span><span class='hs-conid'>TyVarTy</span> <span class='hs-varid'>tv</span><span class='hs-layout'>)</span> <span class='hs-varid'>ty_args</span> <a name="line-190"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-191"></a> <span class='hs-varid'>dfun</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>maybeV</span> <span class='hs-layout'>(</span><span class='hs-varid'>lookupTyVarPA</span> <span class='hs-varid'>tv</span><span class='hs-layout'>)</span> <a name="line-192"></a> <span class='hs-varid'>paDFunApply</span> <span class='hs-varid'>dfun</span> <span class='hs-varid'>ty_args</span> <a name="line-193"></a><span class='hs-definition'>paDictOfTyApp</span> <span class='hs-layout'>(</span><span class='hs-conid'>TyConApp</span> <span class='hs-varid'>tc</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-varid'>ty_args</span> <a name="line-194"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-195"></a> <span class='hs-varid'>dfun</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>maybeCantVectoriseM</span> <span class='hs-str'>"No PA dictionary for tycon"</span> <span class='hs-layout'>(</span><span class='hs-varid'>ppr</span> <span class='hs-varid'>tc</span><span class='hs-layout'>)</span> <a name="line-196"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>lookupTyConPA</span> <span class='hs-varid'>tc</span> <a name="line-197"></a> <span class='hs-varid'>paDFunApply</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>dfun</span><span class='hs-layout'>)</span> <span class='hs-varid'>ty_args</span> <a name="line-198"></a><span class='hs-definition'>paDictOfTyApp</span> <span class='hs-varid'>ty</span> <span class='hs-keyword'>_</span> <a name="line-199"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>cantVectorise</span> <span class='hs-str'>"Can't construct PA dictionary for type"</span> <span class='hs-layout'>(</span><span class='hs-varid'>ppr</span> <span class='hs-varid'>ty</span><span class='hs-layout'>)</span> <a name="line-200"></a> <a name="line-201"></a><a name="paDFunType"></a><span class='hs-definition'>paDFunType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>TyCon</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Type</span> <a name="line-202"></a><span class='hs-definition'>paDFunType</span> <span class='hs-varid'>tc</span> <a name="line-203"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-204"></a> <span class='hs-varid'>margs</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>paDictArgType</span> <span class='hs-varid'>tvs</span> <a name="line-205"></a> <span class='hs-varid'>res</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mkPADictType</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkTyConApp</span> <span class='hs-varid'>tc</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <a name="line-206"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>.</span> <span class='hs-varid'>mkForAllTys</span> <span class='hs-varid'>tvs</span> <a name="line-207"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>mkFunTys</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>arg</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>margs</span><span class='hs-keyglyph'>]</span> <span class='hs-varid'>res</span> <a name="line-208"></a> <span class='hs-keyword'>where</span> <a name="line-209"></a> <span class='hs-varid'>tvs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tyConTyVars</span> <span class='hs-varid'>tc</span> <a name="line-210"></a> <span class='hs-varid'>arg_tys</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkTyVarTys</span> <span class='hs-varid'>tvs</span> <a name="line-211"></a> <a name="line-212"></a><a name="paDFunApply"></a><span class='hs-definition'>paDFunApply</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-213"></a><span class='hs-definition'>paDFunApply</span> <span class='hs-varid'>dfun</span> <span class='hs-varid'>tys</span> <a name="line-214"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-215"></a> <span class='hs-varid'>dicts</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>paDictOfType</span> <span class='hs-varid'>tys</span> <a name="line-216"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mkApps</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkTyApps</span> <span class='hs-varid'>dfun</span> <span class='hs-varid'>tys</span><span class='hs-layout'>)</span> <span class='hs-varid'>dicts</span> <a name="line-217"></a> <a name="line-218"></a><a name="paMethod"></a><span class='hs-definition'>paMethod</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Builtins</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Var</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>String</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-219"></a><span class='hs-definition'>paMethod</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>name</span> <span class='hs-varid'>ty</span> <a name="line-220"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>tycon</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>splitPrimTyCon</span> <span class='hs-varid'>ty</span> <a name="line-221"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-conid'>Var</span> <a name="line-222"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>maybeCantVectoriseM</span> <span class='hs-str'>"No PA method"</span> <span class='hs-layout'>(</span><span class='hs-varid'>text</span> <span class='hs-varid'>name</span> <span class='hs-varop'><+></span> <span class='hs-varid'>text</span> <span class='hs-str'>"for"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>ppr</span> <span class='hs-varid'>tycon</span><span class='hs-layout'>)</span> <a name="line-223"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>lookupPrimMethod</span> <span class='hs-varid'>tycon</span> <span class='hs-varid'>name</span> <a name="line-224"></a> <a name="line-225"></a><span class='hs-definition'>paMethod</span> <span class='hs-varid'>method</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>ty</span> <a name="line-226"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-227"></a> <span class='hs-varid'>fn</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>method</span> <a name="line-228"></a> <span class='hs-varid'>dict</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>paDictOfType</span> <span class='hs-varid'>ty</span> <a name="line-229"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mkApps</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>fn</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span> <span class='hs-varid'>ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>dict</span><span class='hs-keyglyph'>]</span> <a name="line-230"></a> <a name="line-231"></a><a name="mkPR"></a><span class='hs-definition'>mkPR</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-232"></a><span class='hs-definition'>mkPR</span> <span class='hs-varid'>ty</span> <a name="line-233"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-234"></a> <span class='hs-varid'>fn</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>mkPRVar</span> <a name="line-235"></a> <span class='hs-varid'>dict</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>paDictOfType</span> <span class='hs-varid'>ty</span> <a name="line-236"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mkApps</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>fn</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span> <span class='hs-varid'>ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>dict</span><span class='hs-keyglyph'>]</span> <a name="line-237"></a> <a name="line-238"></a><a name="replicatePD"></a><span class='hs-definition'>replicatePD</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-239"></a><span class='hs-definition'>replicatePD</span> <span class='hs-varid'>len</span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-layout'>(</span><span class='hs-varop'>`mkApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>len</span><span class='hs-layout'>,</span><span class='hs-varid'>x</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-240"></a> <span class='hs-layout'>(</span><span class='hs-varid'>paMethod</span> <span class='hs-varid'>replicatePDVar</span> <span class='hs-str'>"replicatePD"</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>x</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-241"></a> <a name="line-242"></a><a name="emptyPD"></a><span class='hs-definition'>emptyPD</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-243"></a><span class='hs-definition'>emptyPD</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>paMethod</span> <span class='hs-varid'>emptyPDVar</span> <span class='hs-str'>"emptyPD"</span> <a name="line-244"></a> <a name="line-245"></a><a name="packPD"></a><span class='hs-definition'>packPD</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-246"></a><span class='hs-definition'>packPD</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>xs</span> <span class='hs-varid'>len</span> <span class='hs-varid'>sel</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-layout'>(</span><span class='hs-varop'>`mkApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>xs</span><span class='hs-layout'>,</span> <span class='hs-varid'>len</span><span class='hs-layout'>,</span> <span class='hs-varid'>sel</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-247"></a> <span class='hs-layout'>(</span><span class='hs-varid'>paMethod</span> <span class='hs-varid'>packPDVar</span> <span class='hs-str'>"packPD"</span> <span class='hs-varid'>ty</span><span class='hs-layout'>)</span> <a name="line-248"></a> <a name="line-249"></a><a name="combinePD"></a><span class='hs-definition'>combinePD</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>CoreExpr</span><span class='hs-keyglyph'>]</span> <a name="line-250"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-251"></a><span class='hs-definition'>combinePD</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>len</span> <span class='hs-varid'>sel</span> <span class='hs-varid'>xs</span> <a name="line-252"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-layout'>(</span><span class='hs-varop'>`mkApps`</span> <span class='hs-layout'>(</span><span class='hs-varid'>len</span> <span class='hs-conop'>:</span> <span class='hs-varid'>sel</span> <span class='hs-conop'>:</span> <span class='hs-varid'>xs</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-253"></a> <span class='hs-layout'>(</span><span class='hs-varid'>paMethod</span> <span class='hs-layout'>(</span><span class='hs-varid'>combinePDVar</span> <span class='hs-varid'>n</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-str'>"combine"</span> <span class='hs-varop'>++</span> <span class='hs-varid'>show</span> <span class='hs-varid'>n</span> <span class='hs-varop'>++</span> <span class='hs-str'>"PD"</span><span class='hs-layout'>)</span> <span class='hs-varid'>ty</span><span class='hs-layout'>)</span> <a name="line-254"></a> <span class='hs-keyword'>where</span> <a name="line-255"></a> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>length</span> <span class='hs-varid'>xs</span> <a name="line-256"></a> <a name="line-257"></a><a name="liftPD"></a><span class='hs-definition'>liftPD</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-258"></a><span class='hs-definition'>liftPD</span> <span class='hs-varid'>x</span> <a name="line-259"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-260"></a> <span class='hs-varid'>lc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>liftingContext</span> <a name="line-261"></a> <span class='hs-varid'>replicatePD</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>lc</span><span class='hs-layout'>)</span> <span class='hs-varid'>x</span> <a name="line-262"></a> <a name="line-263"></a><a name="zipScalars"></a><span class='hs-definition'>zipScalars</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-264"></a><span class='hs-definition'>zipScalars</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varid'>res_ty</span> <a name="line-265"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-266"></a> <span class='hs-varid'>scalar</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>scalarClass</span> <a name="line-267"></a> <span class='hs-layout'>(</span><span class='hs-varid'>dfuns</span><span class='hs-layout'>,</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapAndUnzipM</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>ty</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>lookupInst</span> <span class='hs-varid'>scalar</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>ty</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <span class='hs-varid'>ty_args</span> <a name="line-268"></a> <span class='hs-varid'>zipf</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-layout'>(</span><span class='hs-varid'>scalarZip</span> <span class='hs-varop'>$</span> <span class='hs-varid'>length</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <a name="line-269"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>zipf</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-varid'>ty_args</span> <span class='hs-varop'>`mkApps`</span> <span class='hs-varid'>map</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>dfuns</span> <a name="line-270"></a> <span class='hs-keyword'>where</span> <a name="line-271"></a> <span class='hs-varid'>ty_args</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>res_ty</span><span class='hs-keyglyph'>]</span> <a name="line-272"></a> <a name="line-273"></a><a name="scalarClosure"></a><span class='hs-definition'>scalarClosure</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-274"></a><span class='hs-definition'>scalarClosure</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varid'>res_ty</span> <span class='hs-varid'>scalar_fun</span> <span class='hs-varid'>array_fun</span> <a name="line-275"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-276"></a> <span class='hs-varid'>ctr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-layout'>(</span><span class='hs-varid'>closureCtrFun</span> <span class='hs-varop'>$</span> <span class='hs-varid'>length</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <a name="line-277"></a> <span class='hs-varid'>pas</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>paDictOfType</span> <span class='hs-layout'>(</span><span class='hs-varid'>init</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <a name="line-278"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>ctr</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-layout'>(</span><span class='hs-varid'>arg_tys</span> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>res_ty</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-279"></a> <span class='hs-varop'>`mkApps`</span> <span class='hs-layout'>(</span><span class='hs-varid'>pas</span> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>scalar_fun</span><span class='hs-layout'>,</span> <span class='hs-varid'>array_fun</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-280"></a> <a name="line-281"></a><a name="newLocalVVar"></a><span class='hs-definition'>newLocalVVar</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>FastString</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VVar</span> <a name="line-282"></a><span class='hs-definition'>newLocalVVar</span> <span class='hs-varid'>fs</span> <span class='hs-varid'>vty</span> <a name="line-283"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-284"></a> <span class='hs-varid'>lty</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mkPDataType</span> <span class='hs-varid'>vty</span> <a name="line-285"></a> <span class='hs-varid'>vv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVar</span> <span class='hs-varid'>fs</span> <span class='hs-varid'>vty</span> <a name="line-286"></a> <span class='hs-varid'>lv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVar</span> <span class='hs-varid'>fs</span> <span class='hs-varid'>lty</span> <a name="line-287"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>vv</span><span class='hs-layout'>,</span><span class='hs-varid'>lv</span><span class='hs-layout'>)</span> <a name="line-288"></a> <a name="line-289"></a><a name="polyAbstract"></a><span class='hs-definition'>polyAbstract</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>TyVar</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-layout'>(</span><span class='hs-layout'>(</span><span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-varid'>a</span> <a name="line-290"></a><span class='hs-definition'>polyAbstract</span> <span class='hs-varid'>tvs</span> <span class='hs-varid'>p</span> <a name="line-291"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>localV</span> <a name="line-292"></a> <span class='hs-varop'>$</span> <span class='hs-keyword'>do</span> <a name="line-293"></a> <span class='hs-varid'>mdicts</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>mk_dict_var</span> <span class='hs-varid'>tvs</span> <a name="line-294"></a> <span class='hs-varid'>zipWithM_</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>tv</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>maybe</span> <span class='hs-layout'>(</span><span class='hs-varid'>defLocalTyVar</span> <span class='hs-varid'>tv</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>defLocalTyVarWithPA</span> <span class='hs-varid'>tv</span> <span class='hs-varop'>.</span> <span class='hs-conid'>Var</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-varid'>tvs</span> <span class='hs-varid'>mdicts</span> <a name="line-295"></a> <span class='hs-varid'>p</span> <span class='hs-layout'>(</span><span class='hs-varid'>mk_lams</span> <span class='hs-varid'>mdicts</span><span class='hs-layout'>)</span> <a name="line-296"></a> <span class='hs-keyword'>where</span> <a name="line-297"></a> <span class='hs-varid'>mk_dict_var</span> <span class='hs-varid'>tv</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-298"></a> <span class='hs-varid'>r</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>paDictArgType</span> <span class='hs-varid'>tv</span> <a name="line-299"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>r</span> <span class='hs-keyword'>of</span> <a name="line-300"></a> <span class='hs-conid'>Just</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>liftM</span> <span class='hs-conid'>Just</span> <span class='hs-layout'>(</span><span class='hs-varid'>newLocalVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"dPA"</span><span class='hs-layout'>)</span> <span class='hs-varid'>ty</span><span class='hs-layout'>)</span> <a name="line-301"></a> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-conid'>Nothing</span> <a name="line-302"></a> <a name="line-303"></a> <span class='hs-varid'>mk_lams</span> <span class='hs-varid'>mdicts</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkLams</span> <span class='hs-layout'>(</span><span class='hs-varid'>tvs</span> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>dict</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>dict</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mdicts</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-304"></a> <a name="line-305"></a><a name="polyApply"></a><span class='hs-definition'>polyApply</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreExpr</span> <a name="line-306"></a><span class='hs-definition'>polyApply</span> <span class='hs-varid'>expr</span> <span class='hs-varid'>tys</span> <a name="line-307"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-308"></a> <span class='hs-varid'>dicts</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>paDictOfType</span> <span class='hs-varid'>tys</span> <a name="line-309"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>expr</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-varid'>tys</span> <span class='hs-varop'>`mkApps`</span> <span class='hs-varid'>dicts</span> <a name="line-310"></a> <a name="line-311"></a><a name="polyVApply"></a><span class='hs-definition'>polyVApply</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-312"></a><span class='hs-definition'>polyVApply</span> <span class='hs-varid'>expr</span> <span class='hs-varid'>tys</span> <a name="line-313"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-314"></a> <span class='hs-varid'>dicts</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>paDictOfType</span> <span class='hs-varid'>tys</span> <a name="line-315"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mapVect</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>e</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>e</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-varid'>tys</span> <span class='hs-varop'>`mkApps`</span> <span class='hs-varid'>dicts</span><span class='hs-layout'>)</span> <span class='hs-varid'>expr</span> <a name="line-316"></a> <a name="line-317"></a><a name="hoistBinding"></a><span class='hs-definition'>hoistBinding</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>()</span> <a name="line-318"></a><span class='hs-definition'>hoistBinding</span> <span class='hs-varid'>v</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>updGEnv</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>env</span> <span class='hs-keyglyph'>-></span> <a name="line-319"></a> <span class='hs-varid'>env</span> <span class='hs-layout'>{</span> <span class='hs-varid'>global_bindings</span> <span class='hs-keyglyph'>=</span> <span class='hs-layout'>(</span><span class='hs-varid'>v</span><span class='hs-layout'>,</span><span class='hs-varid'>e</span><span class='hs-layout'>)</span> <span class='hs-conop'>:</span> <span class='hs-varid'>global_bindings</span> <span class='hs-varid'>env</span> <span class='hs-layout'>}</span> <a name="line-320"></a> <a name="line-321"></a><a name="hoistExpr"></a><span class='hs-definition'>hoistExpr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>FastString</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Var</span> <a name="line-322"></a><span class='hs-definition'>hoistExpr</span> <span class='hs-varid'>fs</span> <span class='hs-varid'>expr</span> <a name="line-323"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-324"></a> <span class='hs-varid'>var</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVar</span> <span class='hs-varid'>fs</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>expr</span><span class='hs-layout'>)</span> <a name="line-325"></a> <span class='hs-varid'>hoistBinding</span> <span class='hs-varid'>var</span> <span class='hs-varid'>expr</span> <a name="line-326"></a> <span class='hs-varid'>return</span> <span class='hs-varid'>var</span> <a name="line-327"></a> <a name="line-328"></a><a name="hoistVExpr"></a><span class='hs-definition'>hoistVExpr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VVar</span> <a name="line-329"></a><span class='hs-definition'>hoistVExpr</span> <span class='hs-layout'>(</span><span class='hs-varid'>ve</span><span class='hs-layout'>,</span> <span class='hs-varid'>le</span><span class='hs-layout'>)</span> <a name="line-330"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-331"></a> <span class='hs-varid'>fs</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getBindName</span> <a name="line-332"></a> <span class='hs-varid'>vv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>hoistExpr</span> <span class='hs-layout'>(</span><span class='hs-chr'>'v'</span> <span class='hs-varop'>`consFS`</span> <span class='hs-varid'>fs</span><span class='hs-layout'>)</span> <span class='hs-varid'>ve</span> <a name="line-333"></a> <span class='hs-varid'>lv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>hoistExpr</span> <span class='hs-layout'>(</span><span class='hs-chr'>'l'</span> <span class='hs-varop'>`consFS`</span> <span class='hs-varid'>fs</span><span class='hs-layout'>)</span> <span class='hs-varid'>le</span> <a name="line-334"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>vv</span><span class='hs-layout'>,</span> <span class='hs-varid'>lv</span><span class='hs-layout'>)</span> <a name="line-335"></a> <a name="line-336"></a><a name="hoistPolyVExpr"></a><span class='hs-definition'>hoistPolyVExpr</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>TyVar</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-337"></a><span class='hs-definition'>hoistPolyVExpr</span> <span class='hs-varid'>tvs</span> <span class='hs-varid'>p</span> <a name="line-338"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-339"></a> <span class='hs-varid'>expr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>closedV</span> <span class='hs-varop'>.</span> <span class='hs-varid'>polyAbstract</span> <span class='hs-varid'>tvs</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>abstract</span> <span class='hs-keyglyph'>-></span> <a name="line-340"></a> <span class='hs-varid'>liftM</span> <span class='hs-layout'>(</span><span class='hs-varid'>mapVect</span> <span class='hs-varid'>abstract</span><span class='hs-layout'>)</span> <span class='hs-varid'>p</span> <a name="line-341"></a> <span class='hs-varid'>fn</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>hoistVExpr</span> <span class='hs-varid'>expr</span> <a name="line-342"></a> <span class='hs-varid'>polyVApply</span> <span class='hs-layout'>(</span><span class='hs-varid'>vVar</span> <span class='hs-varid'>fn</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkTyVarTys</span> <span class='hs-varid'>tvs</span><span class='hs-layout'>)</span> <a name="line-343"></a> <a name="line-344"></a><a name="takeHoisted"></a><span class='hs-definition'>takeHoisted</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>VM</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>Var</span><span class='hs-layout'>,</span> <span class='hs-conid'>CoreExpr</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <a name="line-345"></a><span class='hs-definition'>takeHoisted</span> <a name="line-346"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-347"></a> <span class='hs-varid'>env</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>readGEnv</span> <span class='hs-varid'>id</span> <a name="line-348"></a> <span class='hs-varid'>setGEnv</span> <span class='hs-varop'>$</span> <span class='hs-varid'>env</span> <span class='hs-layout'>{</span> <span class='hs-varid'>global_bindings</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>[]</span> <span class='hs-layout'>}</span> <a name="line-349"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>global_bindings</span> <span class='hs-varid'>env</span> <a name="line-350"></a> <a name="line-351"></a><span class='hs-comment'>{- <a name="line-352"></a>boxExpr :: Type -> VExpr -> VM VExpr <a name="line-353"></a>boxExpr ty (vexpr, lexpr) <a name="line-354"></a> | Just (tycon, []) <- splitTyConApp_maybe ty <a name="line-355"></a> , isUnLiftedTyCon tycon <a name="line-356"></a> = do <a name="line-357"></a> r <- lookupBoxedTyCon tycon <a name="line-358"></a> case r of <a name="line-359"></a> Just tycon' -> let [dc] = tyConDataCons tycon' <a name="line-360"></a> in <a name="line-361"></a> return (mkConApp dc [vexpr], lexpr) <a name="line-362"></a> Nothing -> return (vexpr, lexpr) <a name="line-363"></a>-}</span> <a name="line-364"></a> <a name="line-365"></a><a name="mkClosure"></a><span class='hs-definition'>mkClosure</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-366"></a><span class='hs-definition'>mkClosure</span> <span class='hs-varid'>arg_ty</span> <span class='hs-varid'>res_ty</span> <span class='hs-varid'>env_ty</span> <span class='hs-layout'>(</span><span class='hs-varid'>vfn</span><span class='hs-layout'>,</span><span class='hs-varid'>lfn</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>venv</span><span class='hs-layout'>,</span><span class='hs-varid'>lenv</span><span class='hs-layout'>)</span> <a name="line-367"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-368"></a> <span class='hs-varid'>dict</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>paDictOfType</span> <span class='hs-varid'>env_ty</span> <a name="line-369"></a> <span class='hs-varid'>mkv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>closureVar</span> <a name="line-370"></a> <span class='hs-varid'>mkl</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>liftedClosureVar</span> <a name="line-371"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>mkv</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg_ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>res_ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>env_ty</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`mkApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>dict</span><span class='hs-layout'>,</span> <span class='hs-varid'>vfn</span><span class='hs-layout'>,</span> <span class='hs-varid'>lfn</span><span class='hs-layout'>,</span> <span class='hs-varid'>venv</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>,</span> <a name="line-372"></a> <span class='hs-conid'>Var</span> <span class='hs-varid'>mkl</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg_ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>res_ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>env_ty</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`mkApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>dict</span><span class='hs-layout'>,</span> <span class='hs-varid'>vfn</span><span class='hs-layout'>,</span> <span class='hs-varid'>lfn</span><span class='hs-layout'>,</span> <span class='hs-varid'>lenv</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-373"></a> <a name="line-374"></a><a name="mkClosureApp"></a><span class='hs-definition'>mkClosureApp</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-375"></a><span class='hs-definition'>mkClosureApp</span> <span class='hs-varid'>arg_ty</span> <span class='hs-varid'>res_ty</span> <span class='hs-layout'>(</span><span class='hs-varid'>vclo</span><span class='hs-layout'>,</span> <span class='hs-varid'>lclo</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>varg</span><span class='hs-layout'>,</span> <span class='hs-varid'>larg</span><span class='hs-layout'>)</span> <a name="line-376"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-377"></a> <span class='hs-varid'>vapply</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>applyVar</span> <a name="line-378"></a> <span class='hs-varid'>lapply</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>liftedApplyVar</span> <a name="line-379"></a> <span class='hs-varid'>lc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>liftingContext</span> <a name="line-380"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>vapply</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg_ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>res_ty</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`mkApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>vclo</span><span class='hs-layout'>,</span> <span class='hs-varid'>varg</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>,</span> <a name="line-381"></a> <span class='hs-conid'>Var</span> <span class='hs-varid'>lapply</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg_ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>res_ty</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`mkApps`</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Var</span> <span class='hs-varid'>lc</span><span class='hs-layout'>,</span> <span class='hs-varid'>lclo</span><span class='hs-layout'>,</span> <span class='hs-varid'>larg</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-382"></a> <a name="line-383"></a><a name="buildClosures"></a><span class='hs-definition'>buildClosures</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>TyVar</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>VVar</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Type</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-384"></a><span class='hs-definition'>buildClosures</span> <span class='hs-keyword'>_</span> <span class='hs-keyword'>_</span> <span class='hs-conid'>[]</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>mk_body</span> <a name="line-385"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mk_body</span> <a name="line-386"></a><span class='hs-definition'>buildClosures</span> <span class='hs-varid'>tvs</span> <span class='hs-varid'>vars</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg_ty</span><span class='hs-keyglyph'>]</span> <span class='hs-varid'>res_ty</span> <span class='hs-varid'>mk_body</span> <a name="line-387"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-varid'>vInlineMe</span> <span class='hs-layout'>(</span><span class='hs-varid'>buildClosure</span> <span class='hs-varid'>tvs</span> <span class='hs-varid'>vars</span> <span class='hs-varid'>arg_ty</span> <span class='hs-varid'>res_ty</span> <span class='hs-varid'>mk_body</span><span class='hs-layout'>)</span> <a name="line-388"></a><span class='hs-definition'>buildClosures</span> <span class='hs-varid'>tvs</span> <span class='hs-varid'>vars</span> <span class='hs-layout'>(</span><span class='hs-varid'>arg_ty</span> <span class='hs-conop'>:</span> <span class='hs-varid'>arg_tys</span><span class='hs-layout'>)</span> <span class='hs-varid'>res_ty</span> <span class='hs-varid'>mk_body</span> <a name="line-389"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-390"></a> <span class='hs-varid'>res_ty'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mkClosureTypes</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varid'>res_ty</span> <a name="line-391"></a> <span class='hs-varid'>arg</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"x"</span><span class='hs-layout'>)</span> <span class='hs-varid'>arg_ty</span> <a name="line-392"></a> <span class='hs-varid'>liftM</span> <span class='hs-varid'>vInlineMe</span> <a name="line-393"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>buildClosure</span> <span class='hs-varid'>tvs</span> <span class='hs-varid'>vars</span> <span class='hs-varid'>arg_ty</span> <span class='hs-varid'>res_ty'</span> <a name="line-394"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>hoistPolyVExpr</span> <span class='hs-varid'>tvs</span> <a name="line-395"></a> <span class='hs-varop'>$</span> <span class='hs-keyword'>do</span> <a name="line-396"></a> <span class='hs-varid'>lc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>liftingContext</span> <a name="line-397"></a> <span class='hs-varid'>clo</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>buildClosures</span> <span class='hs-varid'>tvs</span> <span class='hs-layout'>(</span><span class='hs-varid'>vars</span> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varid'>res_ty</span> <span class='hs-varid'>mk_body</span> <a name="line-398"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vLams</span> <span class='hs-varid'>lc</span> <span class='hs-layout'>(</span><span class='hs-varid'>vars</span> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <span class='hs-varid'>clo</span> <a name="line-399"></a> <a name="line-400"></a><a name="buildClosure"></a><span class='hs-comment'>-- (clo <x1,...,xn> <f,f^>, aclo (Arr lc xs1 ... xsn) <f,f^>)</span> <a name="line-401"></a><span class='hs-comment'>-- where</span> <a name="line-402"></a><span class='hs-comment'>-- f = \env v -> case env of <x1,...,xn> -> e x1 ... xn v</span> <a name="line-403"></a><span class='hs-comment'>-- f^ = \env v -> case env of Arr l xs1 ... xsn -> e^ l x1 ... xn v</span> <a name="line-404"></a><span class='hs-comment'>--</span> <a name="line-405"></a><span class='hs-definition'>buildClosure</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>TyVar</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>VVar</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-406"></a><span class='hs-definition'>buildClosure</span> <span class='hs-varid'>tvs</span> <span class='hs-varid'>vars</span> <span class='hs-varid'>arg_ty</span> <span class='hs-varid'>res_ty</span> <span class='hs-varid'>mk_body</span> <a name="line-407"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-408"></a> <span class='hs-layout'>(</span><span class='hs-varid'>env_ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>env</span><span class='hs-layout'>,</span> <span class='hs-varid'>bind</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>buildEnv</span> <span class='hs-varid'>vars</span> <a name="line-409"></a> <span class='hs-varid'>env_bndr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"env"</span><span class='hs-layout'>)</span> <span class='hs-varid'>env_ty</span> <a name="line-410"></a> <span class='hs-varid'>arg_bndr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"arg"</span><span class='hs-layout'>)</span> <span class='hs-varid'>arg_ty</span> <a name="line-411"></a> <a name="line-412"></a> <span class='hs-varid'>fn</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>hoistPolyVExpr</span> <span class='hs-varid'>tvs</span> <a name="line-413"></a> <span class='hs-varop'>$</span> <span class='hs-keyword'>do</span> <a name="line-414"></a> <span class='hs-varid'>lc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>liftingContext</span> <a name="line-415"></a> <span class='hs-varid'>body</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mk_body</span> <a name="line-416"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>.</span> <span class='hs-varid'>vInlineMe</span> <a name="line-417"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>vLams</span> <span class='hs-varid'>lc</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>env_bndr</span><span class='hs-layout'>,</span> <span class='hs-varid'>arg_bndr</span><span class='hs-keyglyph'>]</span> <a name="line-418"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>bind</span> <span class='hs-layout'>(</span><span class='hs-varid'>vVar</span> <span class='hs-varid'>env_bndr</span><span class='hs-layout'>)</span> <a name="line-419"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vVarApps</span> <span class='hs-varid'>lc</span> <span class='hs-varid'>body</span> <span class='hs-layout'>(</span><span class='hs-varid'>vars</span> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>arg_bndr</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-420"></a> <a name="line-421"></a> <span class='hs-varid'>mkClosure</span> <span class='hs-varid'>arg_ty</span> <span class='hs-varid'>res_ty</span> <span class='hs-varid'>env_ty</span> <span class='hs-varid'>fn</span> <span class='hs-varid'>env</span> <a name="line-422"></a> <a name="line-423"></a><a name="buildEnv"></a><span class='hs-definition'>buildEnv</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>VVar</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-layout'>(</span><span class='hs-conid'>Type</span><span class='hs-layout'>,</span> <span class='hs-conid'>VExpr</span><span class='hs-layout'>,</span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VExpr</span><span class='hs-layout'>)</span> <a name="line-424"></a><span class='hs-definition'>buildEnv</span> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-425"></a> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>voidType</span> <a name="line-426"></a> <span class='hs-varid'>void</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>voidVar</span> <a name="line-427"></a> <span class='hs-varid'>pvoid</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-varid'>pvoidVar</span> <a name="line-428"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>ty</span><span class='hs-layout'>,</span> <span class='hs-varid'>vVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>void</span><span class='hs-layout'>,</span> <span class='hs-varid'>pvoid</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-keyglyph'>\</span><span class='hs-keyword'>_</span> <span class='hs-varid'>body</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-429"></a> <a name="line-430"></a><span class='hs-definition'>buildEnv</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>v</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>vVarType</span> <span class='hs-varid'>v</span><span class='hs-layout'>,</span> <span class='hs-varid'>vVar</span> <span class='hs-varid'>v</span><span class='hs-layout'>,</span> <a name="line-431"></a> <span class='hs-keyglyph'>\</span><span class='hs-varid'>env</span> <span class='hs-varid'>body</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>vLet</span> <span class='hs-layout'>(</span><span class='hs-varid'>vNonRec</span> <span class='hs-varid'>v</span> <span class='hs-varid'>env</span><span class='hs-layout'>)</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-432"></a> <a name="line-433"></a><span class='hs-definition'>buildEnv</span> <span class='hs-varid'>vs</span> <a name="line-434"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-435"></a> <a name="line-436"></a> <span class='hs-layout'>(</span><span class='hs-varid'>lenv_tc</span><span class='hs-layout'>,</span> <span class='hs-varid'>lenv_tyargs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>pdataReprTyCon</span> <span class='hs-varid'>ty</span> <a name="line-437"></a> <a name="line-438"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>venv_con</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tupleCon</span> <span class='hs-conid'>Boxed</span> <span class='hs-layout'>(</span><span class='hs-varid'>length</span> <span class='hs-varid'>vs</span><span class='hs-layout'>)</span> <a name="line-439"></a> <span class='hs-keyglyph'>[</span><span class='hs-varid'>lenv_con</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tyConDataCons</span> <span class='hs-varid'>lenv_tc</span> <a name="line-440"></a> <a name="line-441"></a> <span class='hs-varid'>venv</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkCoreTup</span> <span class='hs-layout'>(</span><span class='hs-varid'>map</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>vvs</span><span class='hs-layout'>)</span> <a name="line-442"></a> <span class='hs-varid'>lenv</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Var</span> <span class='hs-layout'>(</span><span class='hs-varid'>dataConWrapId</span> <span class='hs-varid'>lenv_con</span><span class='hs-layout'>)</span> <a name="line-443"></a> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-varid'>lenv_tyargs</span> <a name="line-444"></a> <span class='hs-varop'>`mkApps`</span> <span class='hs-varid'>map</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>lvs</span> <a name="line-445"></a> <a name="line-446"></a> <span class='hs-varid'>vbind</span> <span class='hs-varid'>env</span> <span class='hs-varid'>body</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkWildCase</span> <span class='hs-varid'>env</span> <span class='hs-varid'>ty</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-447"></a> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>venv_con</span><span class='hs-layout'>,</span> <span class='hs-varid'>vvs</span><span class='hs-layout'>,</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <a name="line-448"></a> <a name="line-449"></a> <span class='hs-varid'>lbind</span> <span class='hs-varid'>env</span> <span class='hs-varid'>body</span> <span class='hs-keyglyph'>=</span> <a name="line-450"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>scrut</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>unwrapFamInstScrut</span> <span class='hs-varid'>lenv_tc</span> <span class='hs-varid'>lenv_tyargs</span> <span class='hs-varid'>env</span> <a name="line-451"></a> <span class='hs-keyword'>in</span> <a name="line-452"></a> <span class='hs-varid'>mkWildCase</span> <span class='hs-varid'>scrut</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>scrut</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-453"></a> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>lenv_con</span><span class='hs-layout'>,</span> <span class='hs-varid'>lvs</span><span class='hs-layout'>,</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <a name="line-454"></a> <a name="line-455"></a> <span class='hs-varid'>bind</span> <span class='hs-layout'>(</span><span class='hs-varid'>venv</span><span class='hs-layout'>,</span> <span class='hs-varid'>lenv</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>vbody</span><span class='hs-layout'>,</span> <span class='hs-varid'>lbody</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-layout'>(</span><span class='hs-varid'>vbind</span> <span class='hs-varid'>venv</span> <span class='hs-varid'>vbody</span><span class='hs-layout'>,</span> <a name="line-456"></a> <span class='hs-varid'>lbind</span> <span class='hs-varid'>lenv</span> <span class='hs-varid'>lbody</span><span class='hs-layout'>)</span> <a name="line-457"></a> <a name="line-458"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>ty</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varid'>venv</span><span class='hs-layout'>,</span> <span class='hs-varid'>lenv</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-varid'>bind</span><span class='hs-layout'>)</span> <a name="line-459"></a> <span class='hs-keyword'>where</span> <a name="line-460"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vvs</span><span class='hs-layout'>,</span> <span class='hs-varid'>lvs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>unzip</span> <span class='hs-varid'>vs</span> <a name="line-461"></a> <span class='hs-varid'>tys</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>map</span> <span class='hs-varid'>vVarType</span> <span class='hs-varid'>vs</span> <a name="line-462"></a> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkCoreTupTy</span> <span class='hs-varid'>tys</span> <a name="line-463"></a> </pre></body> </html>