<?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/Vectorise.hs</title> <link type='text/css' rel='stylesheet' href='hscolour.css' /> </head> <body> <pre><a name="line-1"></a> <a name="line-2"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>Vectorise</span><span class='hs-layout'>(</span> <span class='hs-varid'>vectorise</span> <span class='hs-layout'>)</span> <a name="line-3"></a><span class='hs-keyword'>where</span> <a name="line-4"></a> <a name="line-5"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VectMonad</span> <a name="line-6"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VectUtils</span> <a name="line-7"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VectType</span> <a name="line-8"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VectCore</span> <a name="line-9"></a> <a name="line-10"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>HscTypes</span> <span class='hs-varid'>hiding</span> <span class='hs-layout'>(</span> <span class='hs-conid'>MonadThings</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span> <span class='hs-layout'>)</span> <a name="line-11"></a> <a name="line-12"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Module</span> <span class='hs-layout'>(</span> <span class='hs-conid'>PackageId</span> <span class='hs-layout'>)</span> <a name="line-13"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CoreSyn</span> <a name="line-14"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CoreUtils</span> <a name="line-15"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>MkCore</span> <span class='hs-layout'>(</span> <span class='hs-varid'>mkWildCase</span> <span class='hs-layout'>)</span> <a name="line-16"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CoreFVs</span> <a name="line-17"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CoreMonad</span> <span class='hs-layout'>(</span> <span class='hs-conid'>CoreM</span><span class='hs-layout'>,</span> <span class='hs-varid'>getHscEnv</span> <span class='hs-layout'>)</span> <a name="line-18"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>DataCon</span> <a name="line-19"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>TyCon</span> <a name="line-20"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Type</span> <a name="line-21"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>FamInstEnv</span> <span class='hs-layout'>(</span> <span class='hs-varid'>extendFamInstEnvList</span> <span class='hs-layout'>)</span> <a name="line-22"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Var</span> <a name="line-23"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VarEnv</span> <a name="line-24"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VarSet</span> <a name="line-25"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Id</span> <a name="line-26"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>OccName</span> <a name="line-27"></a> <a name="line-28"></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-29"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>TysWiredIn</span> <a name="line-30"></a> <a name="line-31"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Outputable</span> <a name="line-32"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>FastString</span> <a name="line-33"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Monad</span> <span class='hs-layout'>(</span> <span class='hs-varid'>liftM</span><span class='hs-layout'>,</span> <span class='hs-varid'>liftM2</span><span class='hs-layout'>,</span> <span class='hs-varid'>zipWithM</span> <span class='hs-layout'>)</span> <a name="line-34"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>List</span> <span class='hs-layout'>(</span> <span class='hs-varid'>sortBy</span><span class='hs-layout'>,</span> <span class='hs-varid'>unzip4</span> <span class='hs-layout'>)</span> <a name="line-35"></a> <a name="line-36"></a><a name="vectorise"></a><span class='hs-definition'>vectorise</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>PackageId</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>ModGuts</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreM</span> <span class='hs-conid'>ModGuts</span> <a name="line-37"></a><span class='hs-definition'>vectorise</span> <span class='hs-varid'>backend</span> <span class='hs-varid'>guts</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-38"></a> <span class='hs-varid'>hsc_env</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getHscEnv</span> <a name="line-39"></a> <span class='hs-varid'>liftIO</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vectoriseIO</span> <span class='hs-varid'>backend</span> <span class='hs-varid'>hsc_env</span> <span class='hs-varid'>guts</span> <a name="line-40"></a> <a name="line-41"></a><a name="vectoriseIO"></a><span class='hs-definition'>vectoriseIO</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>PackageId</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>HscEnv</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>ModGuts</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-conid'>ModGuts</span> <a name="line-42"></a><span class='hs-definition'>vectoriseIO</span> <span class='hs-varid'>backend</span> <span class='hs-varid'>hsc_env</span> <span class='hs-varid'>guts</span> <a name="line-43"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-44"></a> <span class='hs-varid'>eps</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>hscEPS</span> <span class='hs-varid'>hsc_env</span> <a name="line-45"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>info</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>hptVectInfo</span> <span class='hs-varid'>hsc_env</span> <span class='hs-varop'>`plusVectInfo`</span> <span class='hs-varid'>eps_vect_info</span> <span class='hs-varid'>eps</span> <a name="line-46"></a> <span class='hs-conid'>Just</span> <span class='hs-layout'>(</span><span class='hs-varid'>info'</span><span class='hs-layout'>,</span> <span class='hs-varid'>guts'</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>initV</span> <span class='hs-varid'>backend</span> <span class='hs-varid'>hsc_env</span> <span class='hs-varid'>guts</span> <span class='hs-varid'>info</span> <span class='hs-layout'>(</span><span class='hs-varid'>vectModule</span> <span class='hs-varid'>guts</span><span class='hs-layout'>)</span> <a name="line-47"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>guts'</span> <span class='hs-layout'>{</span> <span class='hs-varid'>mg_vect_info</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>info'</span> <span class='hs-layout'>}</span><span class='hs-layout'>)</span> <a name="line-48"></a> <a name="line-49"></a><a name="vectModule"></a><span class='hs-definition'>vectModule</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ModGuts</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>ModGuts</span> <a name="line-50"></a><span class='hs-definition'>vectModule</span> <span class='hs-varid'>guts</span> <a name="line-51"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-52"></a> <span class='hs-layout'>(</span><span class='hs-varid'>types'</span><span class='hs-layout'>,</span> <span class='hs-varid'>fam_insts</span><span class='hs-layout'>,</span> <span class='hs-varid'>tc_binds</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectTypeEnv</span> <span class='hs-layout'>(</span><span class='hs-varid'>mg_types</span> <span class='hs-varid'>guts</span><span class='hs-layout'>)</span> <a name="line-53"></a> <a name="line-54"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>fam_inst_env'</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>extendFamInstEnvList</span> <span class='hs-layout'>(</span><span class='hs-varid'>mg_fam_inst_env</span> <span class='hs-varid'>guts</span><span class='hs-layout'>)</span> <span class='hs-varid'>fam_insts</span> <a name="line-55"></a> <span class='hs-varid'>updGEnv</span> <span class='hs-layout'>(</span><span class='hs-varid'>setFamInstEnv</span> <span class='hs-varid'>fam_inst_env'</span><span class='hs-layout'>)</span> <a name="line-56"></a> <a name="line-57"></a> <span class='hs-comment'>-- dicts <- mapM buildPADict pa_insts</span> <a name="line-58"></a> <span class='hs-comment'>-- workers <- mapM vectDataConWorkers pa_insts</span> <a name="line-59"></a> <span class='hs-varid'>binds'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>vectTopBind</span> <span class='hs-layout'>(</span><span class='hs-varid'>mg_binds</span> <span class='hs-varid'>guts</span><span class='hs-layout'>)</span> <a name="line-60"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>guts</span> <span class='hs-layout'>{</span> <span class='hs-varid'>mg_types</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>types'</span> <a name="line-61"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>mg_binds</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Rec</span> <span class='hs-varid'>tc_binds</span> <span class='hs-conop'>:</span> <span class='hs-varid'>binds'</span> <a name="line-62"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>mg_fam_inst_env</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>fam_inst_env'</span> <a name="line-63"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>mg_fam_insts</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mg_fam_insts</span> <span class='hs-varid'>guts</span> <span class='hs-varop'>++</span> <span class='hs-varid'>fam_insts</span> <a name="line-64"></a> <span class='hs-layout'>}</span> <a name="line-65"></a> <a name="line-66"></a><a name="vectTopBind"></a><span class='hs-definition'>vectTopBind</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CoreBind</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>CoreBind</span> <a name="line-67"></a><span class='hs-definition'>vectTopBind</span> <span class='hs-varid'>b</span><span class='hs-keyglyph'>@</span><span class='hs-layout'>(</span><span class='hs-conid'>NonRec</span> <span class='hs-varid'>var</span> <span class='hs-varid'>expr</span><span class='hs-layout'>)</span> <a name="line-68"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-69"></a> <span class='hs-varid'>var'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectTopBinder</span> <span class='hs-varid'>var</span> <a name="line-70"></a> <span class='hs-varid'>expr'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectTopRhs</span> <span class='hs-varid'>var</span> <span class='hs-varid'>expr</span> <a name="line-71"></a> <span class='hs-varid'>hs</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>takeHoisted</span> <a name="line-72"></a> <span class='hs-varid'>cexpr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>tryConvert</span> <span class='hs-varid'>var</span> <span class='hs-varid'>var'</span> <span class='hs-varid'>expr</span> <a name="line-73"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>.</span> <span class='hs-conid'>Rec</span> <span class='hs-varop'>$</span> <span class='hs-layout'>(</span><span class='hs-varid'>var</span><span class='hs-layout'>,</span> <span class='hs-varid'>cexpr</span><span class='hs-layout'>)</span> <span class='hs-conop'>:</span> <span class='hs-layout'>(</span><span class='hs-varid'>var'</span><span class='hs-layout'>,</span> <span class='hs-varid'>expr'</span><span class='hs-layout'>)</span> <span class='hs-conop'>:</span> <span class='hs-varid'>hs</span> <a name="line-74"></a> <span class='hs-varop'>`orElseV`</span> <a name="line-75"></a> <span class='hs-varid'>return</span> <span class='hs-varid'>b</span> <a name="line-76"></a> <a name="line-77"></a><span class='hs-definition'>vectTopBind</span> <span class='hs-varid'>b</span><span class='hs-keyglyph'>@</span><span class='hs-layout'>(</span><span class='hs-conid'>Rec</span> <span class='hs-varid'>bs</span><span class='hs-layout'>)</span> <a name="line-78"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-79"></a> <span class='hs-varid'>vars'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>vectTopBinder</span> <span class='hs-varid'>vars</span> <a name="line-80"></a> <span class='hs-varid'>exprs'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>zipWithM</span> <span class='hs-varid'>vectTopRhs</span> <span class='hs-varid'>vars</span> <span class='hs-varid'>exprs</span> <a name="line-81"></a> <span class='hs-varid'>hs</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>takeHoisted</span> <a name="line-82"></a> <span class='hs-varid'>cexprs</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>sequence</span> <span class='hs-varop'>$</span> <span class='hs-varid'>zipWith3</span> <span class='hs-varid'>tryConvert</span> <span class='hs-varid'>vars</span> <span class='hs-varid'>vars'</span> <span class='hs-varid'>exprs</span> <a name="line-83"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>.</span> <span class='hs-conid'>Rec</span> <span class='hs-varop'>$</span> <span class='hs-varid'>zip</span> <span class='hs-varid'>vars</span> <span class='hs-varid'>cexprs</span> <span class='hs-varop'>++</span> <span class='hs-varid'>zip</span> <span class='hs-varid'>vars'</span> <span class='hs-varid'>exprs'</span> <span class='hs-varop'>++</span> <span class='hs-varid'>hs</span> <a name="line-84"></a> <span class='hs-varop'>`orElseV`</span> <a name="line-85"></a> <span class='hs-varid'>return</span> <span class='hs-varid'>b</span> <a name="line-86"></a> <span class='hs-keyword'>where</span> <a name="line-87"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vars</span><span class='hs-layout'>,</span> <span class='hs-varid'>exprs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>unzip</span> <span class='hs-varid'>bs</span> <a name="line-88"></a> <a name="line-89"></a><a name="vectTopBinder"></a><span class='hs-definition'>vectTopBinder</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>Var</span> <a name="line-90"></a><span class='hs-definition'>vectTopBinder</span> <span class='hs-varid'>var</span> <a name="line-91"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-92"></a> <span class='hs-varid'>vty</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectType</span> <span class='hs-layout'>(</span><span class='hs-varid'>idType</span> <span class='hs-varid'>var</span><span class='hs-layout'>)</span> <a name="line-93"></a> <span class='hs-varid'>var'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>cloneId</span> <span class='hs-varid'>mkVectOcc</span> <span class='hs-varid'>var</span> <span class='hs-varid'>vty</span> <a name="line-94"></a> <span class='hs-varid'>defGlobalVar</span> <span class='hs-varid'>var</span> <span class='hs-varid'>var'</span> <a name="line-95"></a> <span class='hs-varid'>return</span> <span class='hs-varid'>var'</span> <a name="line-96"></a> <a name="line-97"></a><a name="vectTopRhs"></a><span class='hs-definition'>vectTopRhs</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'>CoreExpr</span> <a name="line-98"></a><span class='hs-definition'>vectTopRhs</span> <span class='hs-varid'>var</span> <span class='hs-varid'>expr</span> <a name="line-99"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-100"></a> <span class='hs-varid'>closedV</span> <span class='hs-varop'>.</span> <span class='hs-varid'>liftM</span> <span class='hs-varid'>vectorised</span> <a name="line-101"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>inBind</span> <span class='hs-varid'>var</span> <a name="line-102"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>vectPolyExpr</span> <span class='hs-layout'>(</span><span class='hs-varid'>freeVars</span> <span class='hs-varid'>expr</span><span class='hs-layout'>)</span> <a name="line-103"></a> <a name="line-104"></a><a name="tryConvert"></a><span class='hs-definition'>tryConvert</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</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'>CoreExpr</span> <a name="line-105"></a><span class='hs-definition'>tryConvert</span> <span class='hs-varid'>var</span> <span class='hs-varid'>vect_var</span> <span class='hs-varid'>rhs</span> <a name="line-106"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>fromVect</span> <span class='hs-layout'>(</span><span class='hs-varid'>idType</span> <span class='hs-varid'>var</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>vect_var</span><span class='hs-layout'>)</span> <span class='hs-varop'>`orElseV`</span> <span class='hs-varid'>return</span> <span class='hs-varid'>rhs</span> <a name="line-107"></a> <a name="line-108"></a><span class='hs-comment'>-- ----------------------------------------------------------------------------</span> <a name="line-109"></a><span class='hs-comment'>-- Bindings</span> <a name="line-110"></a> <a name="line-111"></a><a name="vectBndr"></a><span class='hs-definition'>vectBndr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VVar</span> <a name="line-112"></a><span class='hs-definition'>vectBndr</span> <span class='hs-varid'>v</span> <a name="line-113"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-114"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vty</span><span class='hs-layout'>,</span> <span class='hs-varid'>lty</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectAndLiftType</span> <span class='hs-layout'>(</span><span class='hs-varid'>idType</span> <span class='hs-varid'>v</span><span class='hs-layout'>)</span> <a name="line-115"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>vv</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>v</span> <span class='hs-varop'>`</span><span class='hs-conid'>Id</span><span class='hs-varop'>.</span><span class='hs-varid'>setIdType</span><span class='hs-varop'>`</span> <span class='hs-varid'>vty</span> <a name="line-116"></a> <span class='hs-varid'>lv</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>v</span> <span class='hs-varop'>`</span><span class='hs-conid'>Id</span><span class='hs-varop'>.</span><span class='hs-varid'>setIdType</span><span class='hs-varop'>`</span> <span class='hs-varid'>lty</span> <a name="line-117"></a> <span class='hs-varid'>updLEnv</span> <span class='hs-layout'>(</span><span class='hs-varid'>mapTo</span> <span class='hs-varid'>vv</span> <span class='hs-varid'>lv</span><span class='hs-layout'>)</span> <a name="line-118"></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-119"></a> <span class='hs-keyword'>where</span> <a name="line-120"></a> <span class='hs-varid'>mapTo</span> <span class='hs-varid'>vv</span> <span class='hs-varid'>lv</span> <span class='hs-varid'>env</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>env</span> <span class='hs-layout'>{</span> <span class='hs-varid'>local_vars</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>extendVarEnv</span> <span class='hs-layout'>(</span><span class='hs-varid'>local_vars</span> <span class='hs-varid'>env</span><span class='hs-layout'>)</span> <span class='hs-varid'>v</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> <span class='hs-layout'>}</span> <a name="line-121"></a> <a name="line-122"></a><a name="vectBndrNew"></a><span class='hs-definition'>vectBndrNew</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FastString</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VVar</span> <a name="line-123"></a><span class='hs-definition'>vectBndrNew</span> <span class='hs-varid'>v</span> <span class='hs-varid'>fs</span> <a name="line-124"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-125"></a> <span class='hs-varid'>vty</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectType</span> <span class='hs-layout'>(</span><span class='hs-varid'>idType</span> <span class='hs-varid'>v</span><span class='hs-layout'>)</span> <a name="line-126"></a> <span class='hs-varid'>vv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVVar</span> <span class='hs-varid'>fs</span> <span class='hs-varid'>vty</span> <a name="line-127"></a> <span class='hs-varid'>updLEnv</span> <span class='hs-layout'>(</span><span class='hs-varid'>upd</span> <span class='hs-varid'>vv</span><span class='hs-layout'>)</span> <a name="line-128"></a> <span class='hs-varid'>return</span> <span class='hs-varid'>vv</span> <a name="line-129"></a> <span class='hs-keyword'>where</span> <a name="line-130"></a> <span class='hs-varid'>upd</span> <span class='hs-varid'>vv</span> <span class='hs-varid'>env</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>env</span> <span class='hs-layout'>{</span> <span class='hs-varid'>local_vars</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>extendVarEnv</span> <span class='hs-layout'>(</span><span class='hs-varid'>local_vars</span> <span class='hs-varid'>env</span><span class='hs-layout'>)</span> <span class='hs-varid'>v</span> <span class='hs-varid'>vv</span> <span class='hs-layout'>}</span> <a name="line-131"></a> <a name="line-132"></a><a name="vectBndrIn"></a><span class='hs-definition'>vectBndrIn</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-layout'>(</span><span class='hs-conid'>VVar</span><span class='hs-layout'>,</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <a name="line-133"></a><span class='hs-definition'>vectBndrIn</span> <span class='hs-varid'>v</span> <span class='hs-varid'>p</span> <a name="line-134"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>localV</span> <a name="line-135"></a> <span class='hs-varop'>$</span> <span class='hs-keyword'>do</span> <a name="line-136"></a> <span class='hs-varid'>vv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectBndr</span> <span class='hs-varid'>v</span> <a name="line-137"></a> <span class='hs-varid'>x</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>p</span> <a name="line-138"></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'>x</span><span class='hs-layout'>)</span> <a name="line-139"></a> <a name="line-140"></a><a name="vectBndrNewIn"></a><span class='hs-definition'>vectBndrNewIn</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FastString</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-layout'>(</span><span class='hs-conid'>VVar</span><span class='hs-layout'>,</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <a name="line-141"></a><span class='hs-definition'>vectBndrNewIn</span> <span class='hs-varid'>v</span> <span class='hs-varid'>fs</span> <span class='hs-varid'>p</span> <a name="line-142"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>localV</span> <a name="line-143"></a> <span class='hs-varop'>$</span> <span class='hs-keyword'>do</span> <a name="line-144"></a> <span class='hs-varid'>vv</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectBndrNew</span> <span class='hs-varid'>v</span> <span class='hs-varid'>fs</span> <a name="line-145"></a> <span class='hs-varid'>x</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>p</span> <a name="line-146"></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'>x</span><span class='hs-layout'>)</span> <a name="line-147"></a> <a name="line-148"></a><a name="vectBndrsIn"></a><span class='hs-definition'>vectBndrsIn</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Var</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>[</span><span class='hs-conid'>VVar</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>,</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <a name="line-149"></a><span class='hs-definition'>vectBndrsIn</span> <span class='hs-varid'>vs</span> <span class='hs-varid'>p</span> <a name="line-150"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>localV</span> <a name="line-151"></a> <span class='hs-varop'>$</span> <span class='hs-keyword'>do</span> <a name="line-152"></a> <span class='hs-varid'>vvs</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>vectBndr</span> <span class='hs-varid'>vs</span> <a name="line-153"></a> <span class='hs-varid'>x</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>p</span> <a name="line-154"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>vvs</span><span class='hs-layout'>,</span> <span class='hs-varid'>x</span><span class='hs-layout'>)</span> <a name="line-155"></a> <a name="line-156"></a><span class='hs-comment'>-- ----------------------------------------------------------------------------</span> <a name="line-157"></a><span class='hs-comment'>-- Expressions</span> <a name="line-158"></a> <a name="line-159"></a><a name="vectVar"></a><span class='hs-definition'>vectVar</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-160"></a><span class='hs-definition'>vectVar</span> <span class='hs-varid'>v</span> <a name="line-161"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-162"></a> <span class='hs-varid'>r</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>lookupVar</span> <span class='hs-varid'>v</span> <a name="line-163"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>r</span> <span class='hs-keyword'>of</span> <a name="line-164"></a> <span class='hs-conid'>Local</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> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>vv</span><span class='hs-layout'>,</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>lv</span><span class='hs-layout'>)</span> <a name="line-165"></a> <span class='hs-conid'>Global</span> <span class='hs-varid'>vv</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-166"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>vexpr</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>vv</span> <a name="line-167"></a> <span class='hs-varid'>lexpr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftPD</span> <span class='hs-varid'>vexpr</span> <a name="line-168"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>vexpr</span><span class='hs-layout'>,</span> <span class='hs-varid'>lexpr</span><span class='hs-layout'>)</span> <a name="line-169"></a> <a name="line-170"></a><a name="vectPolyVar"></a><span class='hs-definition'>vectPolyVar</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Var</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-171"></a><span class='hs-definition'>vectPolyVar</span> <span class='hs-varid'>v</span> <span class='hs-varid'>tys</span> <a name="line-172"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-173"></a> <span class='hs-varid'>vtys</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>vectType</span> <span class='hs-varid'>tys</span> <a name="line-174"></a> <span class='hs-varid'>r</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>lookupVar</span> <span class='hs-varid'>v</span> <a name="line-175"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>r</span> <span class='hs-keyword'>of</span> <a name="line-176"></a> <span class='hs-conid'>Local</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> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>liftM2</span> <span class='hs-conid'>(,)</span> <span class='hs-layout'>(</span><span class='hs-varid'>polyApply</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>vv</span><span class='hs-layout'>)</span> <span class='hs-varid'>vtys</span><span class='hs-layout'>)</span> <a name="line-177"></a> <span class='hs-layout'>(</span><span class='hs-varid'>polyApply</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>lv</span><span class='hs-layout'>)</span> <span class='hs-varid'>vtys</span><span class='hs-layout'>)</span> <a name="line-178"></a> <span class='hs-conid'>Global</span> <span class='hs-varid'>poly</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-179"></a> <span class='hs-varid'>vexpr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>polyApply</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>poly</span><span class='hs-layout'>)</span> <span class='hs-varid'>vtys</span> <a name="line-180"></a> <span class='hs-varid'>lexpr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftPD</span> <span class='hs-varid'>vexpr</span> <a name="line-181"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>vexpr</span><span class='hs-layout'>,</span> <span class='hs-varid'>lexpr</span><span class='hs-layout'>)</span> <a name="line-182"></a> <a name="line-183"></a><a name="vectLiteral"></a><span class='hs-definition'>vectLiteral</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Literal</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-184"></a><span class='hs-definition'>vectLiteral</span> <span class='hs-varid'>lit</span> <a name="line-185"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-186"></a> <span class='hs-varid'>lexpr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftPD</span> <span class='hs-layout'>(</span><span class='hs-conid'>Lit</span> <span class='hs-varid'>lit</span><span class='hs-layout'>)</span> <a name="line-187"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>Lit</span> <span class='hs-varid'>lit</span><span class='hs-layout'>,</span> <span class='hs-varid'>lexpr</span><span class='hs-layout'>)</span> <a name="line-188"></a> <a name="line-189"></a><a name="vectPolyExpr"></a><span class='hs-definition'>vectPolyExpr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CoreExprWithFVs</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-190"></a><span class='hs-definition'>vectPolyExpr</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnNote</span> <span class='hs-varid'>note</span> <span class='hs-varid'>expr</span><span class='hs-layout'>)</span> <a name="line-191"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-layout'>(</span><span class='hs-varid'>vNote</span> <span class='hs-varid'>note</span><span class='hs-layout'>)</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vectPolyExpr</span> <span class='hs-varid'>expr</span> <a name="line-192"></a><span class='hs-definition'>vectPolyExpr</span> <span class='hs-varid'>expr</span> <a name="line-193"></a> <span class='hs-keyglyph'>=</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-194"></a> <span class='hs-keyword'>do</span> <a name="line-195"></a> <span class='hs-varid'>mono'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectFnExpr</span> <span class='hs-conid'>False</span> <span class='hs-varid'>mono</span> <a name="line-196"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mapVect</span> <span class='hs-varid'>abstract</span> <span class='hs-varid'>mono'</span> <a name="line-197"></a> <span class='hs-keyword'>where</span> <a name="line-198"></a> <span class='hs-layout'>(</span><span class='hs-varid'>tvs</span><span class='hs-layout'>,</span> <span class='hs-varid'>mono</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>collectAnnTypeBinders</span> <span class='hs-varid'>expr</span> <a name="line-199"></a> <a name="line-200"></a><a name="vectExpr"></a><span class='hs-definition'>vectExpr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CoreExprWithFVs</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-201"></a><span class='hs-definition'>vectExpr</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> <a name="line-202"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-varid'>vType</span> <span class='hs-layout'>(</span><span class='hs-varid'>vectType</span> <span class='hs-varid'>ty</span><span class='hs-layout'>)</span> <a name="line-203"></a> <a name="line-204"></a><span class='hs-definition'>vectExpr</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnVar</span> <span class='hs-varid'>v</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectVar</span> <span class='hs-varid'>v</span> <a name="line-205"></a> <a name="line-206"></a><span class='hs-definition'>vectExpr</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnLit</span> <span class='hs-varid'>lit</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectLiteral</span> <span class='hs-varid'>lit</span> <a name="line-207"></a> <a name="line-208"></a><span class='hs-definition'>vectExpr</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnNote</span> <span class='hs-varid'>note</span> <span class='hs-varid'>expr</span><span class='hs-layout'>)</span> <a name="line-209"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-layout'>(</span><span class='hs-varid'>vNote</span> <span class='hs-varid'>note</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>vectExpr</span> <span class='hs-varid'>expr</span><span class='hs-layout'>)</span> <a name="line-210"></a> <a name="line-211"></a><span class='hs-definition'>vectExpr</span> <span class='hs-varid'>e</span><span class='hs-keyglyph'>@</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-keyword'>_</span> <span class='hs-varid'>arg</span><span class='hs-layout'>)</span> <a name="line-212"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isAnnTypeArg</span> <span class='hs-varid'>arg</span> <a name="line-213"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectTyAppExpr</span> <span class='hs-varid'>fn</span> <span class='hs-varid'>tys</span> <a name="line-214"></a> <span class='hs-keyword'>where</span> <a name="line-215"></a> <span class='hs-layout'>(</span><span class='hs-varid'>fn</span><span class='hs-layout'>,</span> <span class='hs-varid'>tys</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>collectAnnTypeArgs</span> <span class='hs-varid'>e</span> <a name="line-216"></a> <a name="line-217"></a><span class='hs-definition'>vectExpr</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-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnVar</span> <span class='hs-varid'>v</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnLit</span> <span class='hs-varid'>lit</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-218"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>con</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>isDataConId_maybe</span> <span class='hs-varid'>v</span> <a name="line-219"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>is_special_con</span> <span class='hs-varid'>con</span> <a name="line-220"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-221"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>vexpr</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>App</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>v</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-conid'>Lit</span> <span class='hs-varid'>lit</span><span class='hs-layout'>)</span> <a name="line-222"></a> <span class='hs-varid'>lexpr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftPD</span> <span class='hs-varid'>vexpr</span> <a name="line-223"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>vexpr</span><span class='hs-layout'>,</span> <span class='hs-varid'>lexpr</span><span class='hs-layout'>)</span> <a name="line-224"></a> <span class='hs-keyword'>where</span> <a name="line-225"></a> <span class='hs-varid'>is_special_con</span> <span class='hs-varid'>con</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>con</span> <span class='hs-varop'>`elem`</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>intDataCon</span><span class='hs-layout'>,</span> <span class='hs-varid'>floatDataCon</span><span class='hs-layout'>,</span> <span class='hs-varid'>doubleDataCon</span><span class='hs-keyglyph'>]</span> <a name="line-226"></a> <a name="line-227"></a> <a name="line-228"></a><span class='hs-definition'>vectExpr</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'>fn</span> <span class='hs-varid'>arg</span><span class='hs-layout'>)</span> <a name="line-229"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-230"></a> <span class='hs-varid'>arg_ty'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectType</span> <span class='hs-varid'>arg_ty</span> <a name="line-231"></a> <span class='hs-varid'>res_ty'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectType</span> <span class='hs-varid'>res_ty</span> <a name="line-232"></a> <span class='hs-varid'>fn'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>fn</span> <a name="line-233"></a> <span class='hs-varid'>arg'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>arg</span> <a name="line-234"></a> <span class='hs-varid'>mkClosureApp</span> <span class='hs-varid'>arg_ty'</span> <span class='hs-varid'>res_ty'</span> <span class='hs-varid'>fn'</span> <span class='hs-varid'>arg'</span> <a name="line-235"></a> <span class='hs-keyword'>where</span> <a name="line-236"></a> <span class='hs-layout'>(</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-keyglyph'>=</span> <span class='hs-varid'>splitFunTy</span> <span class='hs-varop'>.</span> <span class='hs-varid'>exprType</span> <span class='hs-varop'>$</span> <span class='hs-varid'>deAnnotate</span> <span class='hs-varid'>fn</span> <a name="line-237"></a> <a name="line-238"></a><span class='hs-definition'>vectExpr</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnCase</span> <span class='hs-varid'>scrut</span> <span class='hs-varid'>bndr</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>alts</span><span class='hs-layout'>)</span> <a name="line-239"></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-varid'>ty_args</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>splitTyConApp_maybe</span> <span class='hs-varid'>scrut_ty</span> <a name="line-240"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>isAlgTyCon</span> <span class='hs-varid'>tycon</span> <a name="line-241"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectAlgCase</span> <span class='hs-varid'>tycon</span> <span class='hs-varid'>ty_args</span> <span class='hs-varid'>scrut</span> <span class='hs-varid'>bndr</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>alts</span> <a name="line-242"></a> <span class='hs-keyword'>where</span> <a name="line-243"></a> <span class='hs-varid'>scrut_ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>exprType</span> <span class='hs-layout'>(</span><span class='hs-varid'>deAnnotate</span> <span class='hs-varid'>scrut</span><span class='hs-layout'>)</span> <a name="line-244"></a> <a name="line-245"></a><span class='hs-definition'>vectExpr</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnLet</span> <span class='hs-layout'>(</span><span class='hs-conid'>AnnNonRec</span> <span class='hs-varid'>bndr</span> <span class='hs-varid'>rhs</span><span class='hs-layout'>)</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-246"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-247"></a> <span class='hs-varid'>vrhs</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>localV</span> <span class='hs-varop'>.</span> <span class='hs-varid'>inBind</span> <span class='hs-varid'>bndr</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vectPolyExpr</span> <span class='hs-varid'>rhs</span> <a name="line-248"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vbndr</span><span class='hs-layout'>,</span> <span class='hs-varid'>vbody</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectBndrIn</span> <span class='hs-varid'>bndr</span> <span class='hs-layout'>(</span><span class='hs-varid'>vectExpr</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-249"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vLet</span> <span class='hs-layout'>(</span><span class='hs-varid'>vNonRec</span> <span class='hs-varid'>vbndr</span> <span class='hs-varid'>vrhs</span><span class='hs-layout'>)</span> <span class='hs-varid'>vbody</span> <a name="line-250"></a> <a name="line-251"></a><span class='hs-definition'>vectExpr</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnLet</span> <span class='hs-layout'>(</span><span class='hs-conid'>AnnRec</span> <span class='hs-varid'>bs</span><span class='hs-layout'>)</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-252"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-253"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vbndrs</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varid'>vrhss</span><span class='hs-layout'>,</span> <span class='hs-varid'>vbody</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectBndrsIn</span> <span class='hs-varid'>bndrs</span> <a name="line-254"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>liftM2</span> <span class='hs-conid'>(,)</span> <a name="line-255"></a> <span class='hs-layout'>(</span><span class='hs-varid'>zipWithM</span> <span class='hs-varid'>vect_rhs</span> <span class='hs-varid'>bndrs</span> <span class='hs-varid'>rhss</span><span class='hs-layout'>)</span> <a name="line-256"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vectPolyExpr</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-257"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vLet</span> <span class='hs-layout'>(</span><span class='hs-varid'>vRec</span> <span class='hs-varid'>vbndrs</span> <span class='hs-varid'>vrhss</span><span class='hs-layout'>)</span> <span class='hs-varid'>vbody</span> <a name="line-258"></a> <span class='hs-keyword'>where</span> <a name="line-259"></a> <span class='hs-layout'>(</span><span class='hs-varid'>bndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>rhss</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>unzip</span> <span class='hs-varid'>bs</span> <a name="line-260"></a> <a name="line-261"></a> <span class='hs-varid'>vect_rhs</span> <span class='hs-varid'>bndr</span> <span class='hs-varid'>rhs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>localV</span> <a name="line-262"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>inBind</span> <span class='hs-varid'>bndr</span> <a name="line-263"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>rhs</span> <a name="line-264"></a> <a name="line-265"></a><span class='hs-definition'>vectExpr</span> <span class='hs-varid'>e</span><span class='hs-keyglyph'>@</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'>bndr</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <a name="line-266"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isId</span> <span class='hs-varid'>bndr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectFnExpr</span> <span class='hs-conid'>True</span> <span class='hs-varid'>e</span> <a name="line-267"></a><span class='hs-comment'>{- <a name="line-268"></a>onlyIfV (isEmptyVarSet fvs) (vectScalarLam bs $ deAnnotate body) <a name="line-269"></a> `orElseV` vectLam True fvs bs body <a name="line-270"></a> where <a name="line-271"></a> (bs,body) = collectAnnValBinders e <a name="line-272"></a>-}</span> <a name="line-273"></a> <a name="line-274"></a><span class='hs-definition'>vectExpr</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>cantVectorise</span> <span class='hs-str'>"Can't vectorise expression"</span> <span class='hs-layout'>(</span><span class='hs-varid'>ppr</span> <span class='hs-varop'>$</span> <span class='hs-varid'>deAnnotate</span> <span class='hs-varid'>e</span><span class='hs-layout'>)</span> <a name="line-275"></a> <a name="line-276"></a><a name="vectFnExpr"></a><span class='hs-definition'>vectFnExpr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Bool</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExprWithFVs</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-277"></a><span class='hs-definition'>vectFnExpr</span> <span class='hs-varid'>inline</span> <span class='hs-varid'>e</span><span class='hs-keyglyph'>@</span><span class='hs-layout'>(</span><span class='hs-varid'>fvs</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnLam</span> <span class='hs-varid'>bndr</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <a name="line-278"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isId</span> <span class='hs-varid'>bndr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>onlyIfV</span> <span class='hs-layout'>(</span><span class='hs-varid'>isEmptyVarSet</span> <span class='hs-varid'>fvs</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>vectScalarLam</span> <span class='hs-varid'>bs</span> <span class='hs-varop'>$</span> <span class='hs-varid'>deAnnotate</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-279"></a> <span class='hs-varop'>`orElseV`</span> <span class='hs-varid'>vectLam</span> <span class='hs-varid'>inline</span> <span class='hs-varid'>fvs</span> <span class='hs-varid'>bs</span> <span class='hs-varid'>body</span> <a name="line-280"></a> <span class='hs-keyword'>where</span> <a name="line-281"></a> <span class='hs-layout'>(</span><span class='hs-varid'>bs</span><span class='hs-layout'>,</span><span class='hs-varid'>body</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>collectAnnValBinders</span> <span class='hs-varid'>e</span> <a name="line-282"></a><span class='hs-definition'>vectFnExpr</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>e</span> <a name="line-283"></a> <a name="line-284"></a> <a name="line-285"></a><a name="vectScalarLam"></a><span class='hs-definition'>vectScalarLam</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Var</span><span class='hs-keyglyph'>]</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'>VExpr</span> <a name="line-286"></a><span class='hs-definition'>vectScalarLam</span> <span class='hs-varid'>args</span> <span class='hs-varid'>body</span> <a name="line-287"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-288"></a> <span class='hs-varid'>scalars</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>globalScalars</span> <a name="line-289"></a> <span class='hs-varid'>onlyIfV</span> <span class='hs-layout'>(</span><span class='hs-varid'>all</span> <span class='hs-varid'>is_scalar_ty</span> <span class='hs-varid'>arg_tys</span> <a name="line-290"></a> <span class='hs-varop'>&&</span> <span class='hs-varid'>is_scalar_ty</span> <span class='hs-varid'>res_ty</span> <a name="line-291"></a> <span class='hs-varop'>&&</span> <span class='hs-varid'>is_scalar</span> <span class='hs-layout'>(</span><span class='hs-varid'>extendVarSetList</span> <span class='hs-varid'>scalars</span> <span class='hs-varid'>args</span><span class='hs-layout'>)</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</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'>fn_var</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>hoistExpr</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"fn"</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkLams</span> <span class='hs-varid'>args</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-294"></a> <span class='hs-varid'>zipf</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>zipScalars</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varid'>res_ty</span> <a name="line-295"></a> <span class='hs-varid'>clo</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>scalarClosure</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varid'>res_ty</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>fn_var</span><span class='hs-layout'>)</span> <a name="line-296"></a> <span class='hs-layout'>(</span><span class='hs-varid'>zipf</span> <span class='hs-varop'>`App`</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>fn_var</span><span class='hs-layout'>)</span> <a name="line-297"></a> <span class='hs-varid'>clo_var</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>hoistExpr</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"clo"</span><span class='hs-layout'>)</span> <span class='hs-varid'>clo</span> <a name="line-298"></a> <span class='hs-varid'>lclo</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftPD</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>clo_var</span><span class='hs-layout'>)</span> <a name="line-299"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>clo_var</span><span class='hs-layout'>,</span> <span class='hs-varid'>lclo</span><span class='hs-layout'>)</span> <a name="line-300"></a> <span class='hs-keyword'>where</span> <a name="line-301"></a> <span class='hs-varid'>arg_tys</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>map</span> <span class='hs-varid'>idType</span> <span class='hs-varid'>args</span> <a name="line-302"></a> <span class='hs-varid'>res_ty</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>exprType</span> <span class='hs-varid'>body</span> <a name="line-303"></a> <a name="line-304"></a> <span class='hs-varid'>is_scalar_ty</span> <span class='hs-varid'>ty</span> <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-305"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tycon</span> <span class='hs-varop'>==</span> <span class='hs-varid'>intTyCon</span> <a name="line-306"></a> <span class='hs-varop'>||</span> <span class='hs-varid'>tycon</span> <span class='hs-varop'>==</span> <span class='hs-varid'>floatTyCon</span> <a name="line-307"></a> <span class='hs-varop'>||</span> <span class='hs-varid'>tycon</span> <span class='hs-varop'>==</span> <span class='hs-varid'>doubleTyCon</span> <a name="line-308"></a> <a name="line-309"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>False</span> <a name="line-310"></a> <a name="line-311"></a> <span class='hs-varid'>is_scalar</span> <span class='hs-varid'>vs</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>v</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>v</span> <span class='hs-varop'>`elemVarSet`</span> <span class='hs-varid'>vs</span> <a name="line-312"></a> <span class='hs-varid'>is_scalar</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>e</span><span class='hs-keyglyph'>@</span><span class='hs-layout'>(</span><span class='hs-conid'>Lit</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>is_scalar_ty</span> <span class='hs-varop'>$</span> <span class='hs-varid'>exprType</span> <span class='hs-varid'>e</span> <a name="line-313"></a> <span class='hs-varid'>is_scalar</span> <span class='hs-varid'>vs</span> <span class='hs-layout'>(</span><span class='hs-conid'>App</span> <span class='hs-varid'>e1</span> <span class='hs-varid'>e2</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>is_scalar</span> <span class='hs-varid'>vs</span> <span class='hs-varid'>e1</span> <span class='hs-varop'>&&</span> <span class='hs-varid'>is_scalar</span> <span class='hs-varid'>vs</span> <span class='hs-varid'>e2</span> <a name="line-314"></a> <span class='hs-varid'>is_scalar</span> <span class='hs-keyword'>_</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>False</span> <a name="line-315"></a> <a name="line-316"></a><a name="vectLam"></a><span class='hs-definition'>vectLam</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Bool</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VarSet</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Var</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CoreExprWithFVs</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-317"></a><span class='hs-definition'>vectLam</span> <span class='hs-varid'>inline</span> <span class='hs-varid'>fvs</span> <span class='hs-varid'>bs</span> <span class='hs-varid'>body</span> <a name="line-318"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-319"></a> <span class='hs-varid'>tyvars</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>localTyVars</span> <a name="line-320"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vs</span><span class='hs-layout'>,</span> <span class='hs-varid'>vvs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>readLEnv</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-321"></a> <span class='hs-varid'>unzip</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-varid'>var</span><span class='hs-layout'>,</span> <span class='hs-varid'>vv</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>var</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>varSetElems</span> <span class='hs-varid'>fvs</span> <a name="line-322"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>vv</span> <span class='hs-keyglyph'><-</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>lookupVarEnv</span> <span class='hs-layout'>(</span><span class='hs-varid'>local_vars</span> <span class='hs-varid'>env</span><span class='hs-layout'>)</span> <span class='hs-varid'>var</span><span class='hs-keyglyph'>]</span><span class='hs-keyglyph'>]</span> <a name="line-323"></a> <a name="line-324"></a> <span class='hs-varid'>arg_tys</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-layout'>(</span><span class='hs-varid'>vectType</span> <span class='hs-varop'>.</span> <span class='hs-varid'>idType</span><span class='hs-layout'>)</span> <span class='hs-varid'>bs</span> <a name="line-325"></a> <span class='hs-varid'>res_ty</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectType</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varop'>$</span> <span class='hs-varid'>deAnnotate</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-326"></a> <a name="line-327"></a> <span class='hs-varid'>buildClosures</span> <span class='hs-varid'>tyvars</span> <span class='hs-varid'>vvs</span> <span class='hs-varid'>arg_tys</span> <span class='hs-varid'>res_ty</span> <a name="line-328"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>hoistPolyVExpr</span> <span class='hs-varid'>tyvars</span> <a name="line-329"></a> <span class='hs-varop'>$</span> <span class='hs-keyword'>do</span> <a name="line-330"></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-331"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vbndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>vbody</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectBndrsIn</span> <span class='hs-layout'>(</span><span class='hs-varid'>vs</span> <span class='hs-varop'>++</span> <span class='hs-varid'>bs</span><span class='hs-layout'>)</span> <a name="line-332"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vectExpr</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-333"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>.</span> <span class='hs-varid'>maybe_inline</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vLams</span> <span class='hs-varid'>lc</span> <span class='hs-varid'>vbndrs</span> <span class='hs-varid'>vbody</span> <a name="line-334"></a> <span class='hs-keyword'>where</span> <a name="line-335"></a> <span class='hs-varid'>maybe_inline</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>if</span> <span class='hs-varid'>inline</span> <span class='hs-keyword'>then</span> <span class='hs-varid'>vInlineMe</span> <span class='hs-keyword'>else</span> <span class='hs-varid'>id</span> <a name="line-336"></a> <a name="line-337"></a><a name="vectTyAppExpr"></a><span class='hs-definition'>vectTyAppExpr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CoreExprWithFVs</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-338"></a><span class='hs-definition'>vectTyAppExpr</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>AnnVar</span> <span class='hs-varid'>v</span><span class='hs-layout'>)</span> <span class='hs-varid'>tys</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectPolyVar</span> <span class='hs-varid'>v</span> <span class='hs-varid'>tys</span> <a name="line-339"></a><span class='hs-definition'>vectTyAppExpr</span> <span class='hs-varid'>e</span> <span class='hs-varid'>tys</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>cantVectorise</span> <span class='hs-str'>"Can't vectorise expression"</span> <a name="line-340"></a> <span class='hs-layout'>(</span><span class='hs-varid'>ppr</span> <span class='hs-varop'>$</span> <span class='hs-varid'>deAnnotate</span> <span class='hs-varid'>e</span> <span class='hs-varop'>`mkTyApps`</span> <span class='hs-varid'>tys</span><span class='hs-layout'>)</span> <a name="line-341"></a> <a name="line-342"></a><span class='hs-comment'>-- We convert</span> <a name="line-343"></a><span class='hs-comment'>--</span> <a name="line-344"></a><span class='hs-comment'>-- case e :: t of v { ... }</span> <a name="line-345"></a><span class='hs-comment'>--</span> <a name="line-346"></a><span class='hs-comment'>-- to</span> <a name="line-347"></a><span class='hs-comment'>--</span> <a name="line-348"></a><span class='hs-comment'>-- V: let v' = e in case v' of _ { ... }</span> <a name="line-349"></a><span class='hs-comment'>-- L: let v' = e in case v' `cast` ... of _ { ... }</span> <a name="line-350"></a><span class='hs-comment'>--</span> <a name="line-351"></a><span class='hs-comment'>-- When lifting, we have to do it this way because v must have the type</span> <a name="line-352"></a><span class='hs-comment'>-- [:V(T):] but the scrutinee must be cast to the representation type. We also</span> <a name="line-353"></a><span class='hs-comment'>-- have to handle the case where v is a wild var correctly.</span> <a name="line-354"></a><span class='hs-comment'>--</span> <a name="line-355"></a> <a name="line-356"></a><a name="vectAlgCase"></a><span class='hs-comment'>-- FIXME: this is too lazy</span> <a name="line-357"></a><span class='hs-definition'>vectAlgCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>TyCon</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'>CoreExprWithFVs</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Var</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Type</span> <a name="line-358"></a> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>AltCon</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'>CoreExprWithFVs</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <a name="line-359"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>VM</span> <span class='hs-conid'>VExpr</span> <a name="line-360"></a><span class='hs-definition'>vectAlgCase</span> <span class='hs-sel'>_tycon</span> <span class='hs-sel'>_ty_args</span> <span class='hs-varid'>scrut</span> <span class='hs-varid'>bndr</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>DEFAULT</span><span class='hs-layout'>,</span> <span class='hs-conid'>[]</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-361"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-362"></a> <span class='hs-varid'>vscrut</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>scrut</span> <a name="line-363"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vty</span><span class='hs-layout'>,</span> <span class='hs-varid'>lty</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectAndLiftType</span> <span class='hs-varid'>ty</span> <a name="line-364"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vbndr</span><span class='hs-layout'>,</span> <span class='hs-varid'>vbody</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectBndrIn</span> <span class='hs-varid'>bndr</span> <span class='hs-layout'>(</span><span class='hs-varid'>vectExpr</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-365"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vCaseDEFAULT</span> <span class='hs-varid'>vscrut</span> <span class='hs-varid'>vbndr</span> <span class='hs-varid'>vty</span> <span class='hs-varid'>lty</span> <span class='hs-varid'>vbody</span> <a name="line-366"></a> <a name="line-367"></a><span class='hs-definition'>vectAlgCase</span> <span class='hs-sel'>_tycon</span> <span class='hs-sel'>_ty_args</span> <span class='hs-varid'>scrut</span> <span class='hs-varid'>bndr</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-conid'>[]</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-368"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-369"></a> <span class='hs-varid'>vscrut</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>scrut</span> <a name="line-370"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vty</span><span class='hs-layout'>,</span> <span class='hs-varid'>lty</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectAndLiftType</span> <span class='hs-varid'>ty</span> <a name="line-371"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vbndr</span><span class='hs-layout'>,</span> <span class='hs-varid'>vbody</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectBndrIn</span> <span class='hs-varid'>bndr</span> <span class='hs-layout'>(</span><span class='hs-varid'>vectExpr</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-372"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vCaseDEFAULT</span> <span class='hs-varid'>vscrut</span> <span class='hs-varid'>vbndr</span> <span class='hs-varid'>vty</span> <span class='hs-varid'>lty</span> <span class='hs-varid'>vbody</span> <a name="line-373"></a> <a name="line-374"></a><span class='hs-definition'>vectAlgCase</span> <span class='hs-sel'>_tycon</span> <span class='hs-sel'>_ty_args</span> <span class='hs-varid'>scrut</span> <span class='hs-varid'>bndr</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>dc</span><span class='hs-layout'>,</span> <span class='hs-varid'>bndrs</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-375"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-376"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vty</span><span class='hs-layout'>,</span> <span class='hs-varid'>lty</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectAndLiftType</span> <span class='hs-varid'>ty</span> <a name="line-377"></a> <span class='hs-varid'>vexpr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>scrut</span> <a name="line-378"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vbndr</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varid'>vbndrs</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varid'>vect_body</span><span class='hs-layout'>,</span> <span class='hs-varid'>lift_body</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-379"></a> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vect_scrut_bndr</span> <a name="line-380"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>vectBndrsIn</span> <span class='hs-varid'>bndrs</span> <a name="line-381"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>body</span> <a name="line-382"></a> <span class='hs-keyword'>let</span> <span class='hs-layout'>(</span><span class='hs-varid'>vect_bndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>lift_bndrs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>unzip</span> <span class='hs-varid'>vbndrs</span> <a name="line-383"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vscrut</span><span class='hs-layout'>,</span> <span class='hs-varid'>lscrut</span><span class='hs-layout'>,</span> <span class='hs-varid'>pdata_tc</span><span class='hs-layout'>,</span> <span class='hs-sel'>_arg_tys</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mkVScrut</span> <span class='hs-layout'>(</span><span class='hs-varid'>vVar</span> <span class='hs-varid'>vbndr</span><span class='hs-layout'>)</span> <a name="line-384"></a> <span class='hs-varid'>vect_dc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>maybeV</span> <span class='hs-layout'>(</span><span class='hs-varid'>lookupDataCon</span> <span class='hs-varid'>dc</span><span class='hs-layout'>)</span> <a name="line-385"></a> <span class='hs-keyword'>let</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>pdata_dc</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tyConDataCons</span> <span class='hs-varid'>pdata_tc</span> <a name="line-386"></a> <a name="line-387"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>vcase</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mk_wild_case</span> <span class='hs-varid'>vscrut</span> <span class='hs-varid'>vty</span> <span class='hs-varid'>vect_dc</span> <span class='hs-varid'>vect_bndrs</span> <span class='hs-varid'>vect_body</span> <a name="line-388"></a> <span class='hs-varid'>lcase</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mk_wild_case</span> <span class='hs-varid'>lscrut</span> <span class='hs-varid'>lty</span> <span class='hs-varid'>pdata_dc</span> <span class='hs-varid'>lift_bndrs</span> <span class='hs-varid'>lift_body</span> <a name="line-389"></a> <a name="line-390"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>vLet</span> <span class='hs-layout'>(</span><span class='hs-varid'>vNonRec</span> <span class='hs-varid'>vbndr</span> <span class='hs-varid'>vexpr</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>vcase</span><span class='hs-layout'>,</span> <span class='hs-varid'>lcase</span><span class='hs-layout'>)</span> <a name="line-391"></a> <span class='hs-keyword'>where</span> <a name="line-392"></a> <span class='hs-varid'>vect_scrut_bndr</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isDeadBinder</span> <span class='hs-varid'>bndr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectBndrNewIn</span> <span class='hs-varid'>bndr</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"scrut"</span><span class='hs-layout'>)</span> <a name="line-393"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectBndrIn</span> <span class='hs-varid'>bndr</span> <a name="line-394"></a> <a name="line-395"></a> <span class='hs-varid'>mk_wild_case</span> <span class='hs-varid'>expr</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>dc</span> <span class='hs-varid'>bndrs</span> <span class='hs-varid'>body</span> <a name="line-396"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkWildCase</span> <span class='hs-varid'>expr</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>expr</span><span class='hs-layout'>)</span> <span class='hs-varid'>ty</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>dc</span><span class='hs-layout'>,</span> <span class='hs-varid'>bndrs</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-397"></a> <a name="line-398"></a><span class='hs-definition'>vectAlgCase</span> <span class='hs-varid'>tycon</span> <span class='hs-sel'>_ty_args</span> <span class='hs-varid'>scrut</span> <span class='hs-varid'>bndr</span> <span class='hs-varid'>ty</span> <span class='hs-varid'>alts</span> <a name="line-399"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-400"></a> <span class='hs-varid'>vect_tc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>maybeV</span> <span class='hs-layout'>(</span><span class='hs-varid'>lookupTyCon</span> <span class='hs-varid'>tycon</span><span class='hs-layout'>)</span> <a name="line-401"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vty</span><span class='hs-layout'>,</span> <span class='hs-varid'>lty</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectAndLiftType</span> <span class='hs-varid'>ty</span> <a name="line-402"></a> <a name="line-403"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>arity</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>length</span> <span class='hs-layout'>(</span><span class='hs-varid'>tyConDataCons</span> <span class='hs-varid'>vect_tc</span><span class='hs-layout'>)</span> <a name="line-404"></a> <span class='hs-varid'>sel_ty</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-layout'>(</span><span class='hs-varid'>selTy</span> <span class='hs-varid'>arity</span><span class='hs-layout'>)</span> <a name="line-405"></a> <span class='hs-varid'>sel_bndr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"sel"</span><span class='hs-layout'>)</span> <span class='hs-varid'>sel_ty</span> <a name="line-406"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>sel</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>sel_bndr</span> <a name="line-407"></a> <a name="line-408"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vbndr</span><span class='hs-layout'>,</span> <span class='hs-varid'>valts</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vect_scrut_bndr</span> <a name="line-409"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>mapM</span> <span class='hs-layout'>(</span><span class='hs-varid'>proc_alt</span> <span class='hs-varid'>arity</span> <span class='hs-varid'>sel</span> <span class='hs-varid'>vty</span> <span class='hs-varid'>lty</span><span class='hs-layout'>)</span> <span class='hs-varid'>alts'</span> <a name="line-410"></a> <span class='hs-keyword'>let</span> <span class='hs-layout'>(</span><span class='hs-varid'>vect_dcs</span><span class='hs-layout'>,</span> <span class='hs-varid'>vect_bndrss</span><span class='hs-layout'>,</span> <span class='hs-varid'>lift_bndrss</span><span class='hs-layout'>,</span> <span class='hs-varid'>vbodies</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>unzip4</span> <span class='hs-varid'>valts</span> <a name="line-411"></a> <a name="line-412"></a> <span class='hs-varid'>vexpr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>scrut</span> <a name="line-413"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vect_scrut</span><span class='hs-layout'>,</span> <span class='hs-varid'>lift_scrut</span><span class='hs-layout'>,</span> <span class='hs-varid'>pdata_tc</span><span class='hs-layout'>,</span> <span class='hs-sel'>_arg_tys</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mkVScrut</span> <span class='hs-layout'>(</span><span class='hs-varid'>vVar</span> <span class='hs-varid'>vbndr</span><span class='hs-layout'>)</span> <a name="line-414"></a> <span class='hs-keyword'>let</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>pdata_dc</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tyConDataCons</span> <span class='hs-varid'>pdata_tc</span> <a name="line-415"></a> <a name="line-416"></a> <span class='hs-keyword'>let</span> <span class='hs-layout'>(</span><span class='hs-varid'>vect_bodies</span><span class='hs-layout'>,</span> <span class='hs-varid'>lift_bodies</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>unzip</span> <span class='hs-varid'>vbodies</span> <a name="line-417"></a> <a name="line-418"></a> <span class='hs-varid'>vdummy</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newDummyVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>vect_scrut</span><span class='hs-layout'>)</span> <a name="line-419"></a> <span class='hs-varid'>ldummy</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newDummyVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>lift_scrut</span><span class='hs-layout'>)</span> <a name="line-420"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>vect_case</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Case</span> <span class='hs-varid'>vect_scrut</span> <span class='hs-varid'>vdummy</span> <span class='hs-varid'>vty</span> <a name="line-421"></a> <span class='hs-layout'>(</span><span class='hs-varid'>zipWith3</span> <span class='hs-varid'>mk_vect_alt</span> <span class='hs-varid'>vect_dcs</span> <span class='hs-varid'>vect_bndrss</span> <span class='hs-varid'>vect_bodies</span><span class='hs-layout'>)</span> <a name="line-422"></a> <a name="line-423"></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-424"></a> <span class='hs-varid'>lbody</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>combinePD</span> <span class='hs-varid'>vty</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'>sel</span> <span class='hs-varid'>lift_bodies</span> <a name="line-425"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>lift_case</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Case</span> <span class='hs-varid'>lift_scrut</span> <span class='hs-varid'>ldummy</span> <span class='hs-varid'>lty</span> <a name="line-426"></a> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>pdata_dc</span><span class='hs-layout'>,</span> <span class='hs-varid'>sel_bndr</span> <span class='hs-conop'>:</span> <span class='hs-varid'>concat</span> <span class='hs-varid'>lift_bndrss</span><span class='hs-layout'>,</span> <a name="line-427"></a> <span class='hs-varid'>lbody</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <a name="line-428"></a> <a name="line-429"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>.</span> <span class='hs-varid'>vLet</span> <span class='hs-layout'>(</span><span class='hs-varid'>vNonRec</span> <span class='hs-varid'>vbndr</span> <span class='hs-varid'>vexpr</span><span class='hs-layout'>)</span> <a name="line-430"></a> <span class='hs-varop'>$</span> <span class='hs-layout'>(</span><span class='hs-varid'>vect_case</span><span class='hs-layout'>,</span> <span class='hs-varid'>lift_case</span><span class='hs-layout'>)</span> <a name="line-431"></a> <span class='hs-keyword'>where</span> <a name="line-432"></a> <span class='hs-varid'>vect_scrut_bndr</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isDeadBinder</span> <span class='hs-varid'>bndr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectBndrNewIn</span> <span class='hs-varid'>bndr</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"scrut"</span><span class='hs-layout'>)</span> <a name="line-433"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vectBndrIn</span> <span class='hs-varid'>bndr</span> <a name="line-434"></a> <a name="line-435"></a> <span class='hs-varid'>alts'</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>sortBy</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-layout'>(</span><span class='hs-varid'>alt1</span><span class='hs-layout'>,</span> <span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>alt2</span><span class='hs-layout'>,</span> <span class='hs-keyword'>_</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'>cmp</span> <span class='hs-varid'>alt1</span> <span class='hs-varid'>alt2</span><span class='hs-layout'>)</span> <span class='hs-varid'>alts</span> <a name="line-436"></a> <a name="line-437"></a> <span class='hs-varid'>cmp</span> <span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>dc1</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>dc2</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>dataConTag</span> <span class='hs-varid'>dc1</span> <span class='hs-varop'>`compare`</span> <span class='hs-varid'>dataConTag</span> <span class='hs-varid'>dc2</span> <a name="line-438"></a> <span class='hs-varid'>cmp</span> <span class='hs-conid'>DEFAULT</span> <span class='hs-conid'>DEFAULT</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>EQ</span> <a name="line-439"></a> <span class='hs-varid'>cmp</span> <span class='hs-conid'>DEFAULT</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>LT</span> <a name="line-440"></a> <span class='hs-varid'>cmp</span> <span class='hs-keyword'>_</span> <span class='hs-conid'>DEFAULT</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>GT</span> <a name="line-441"></a> <span class='hs-varid'>cmp</span> <span class='hs-keyword'>_</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>panic</span> <span class='hs-str'>"vectAlgCase/cmp"</span> <a name="line-442"></a> <a name="line-443"></a> <span class='hs-varid'>proc_alt</span> <span class='hs-varid'>arity</span> <span class='hs-varid'>sel</span> <span class='hs-varid'>vty</span> <span class='hs-varid'>lty</span> <span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>dc</span><span class='hs-layout'>,</span> <span class='hs-varid'>bndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-444"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-445"></a> <span class='hs-varid'>vect_dc</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>maybeV</span> <span class='hs-layout'>(</span><span class='hs-varid'>lookupDataCon</span> <span class='hs-varid'>dc</span><span class='hs-layout'>)</span> <a name="line-446"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>ntag</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>dataConTagZ</span> <span class='hs-varid'>vect_dc</span> <a name="line-447"></a> <span class='hs-varid'>tag</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkDataConTag</span> <span class='hs-varid'>vect_dc</span> <a name="line-448"></a> <span class='hs-varid'>fvs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>freeVarsOf</span> <span class='hs-varid'>body</span> <span class='hs-varop'>`delVarSetList`</span> <span class='hs-varid'>bndrs</span> <a name="line-449"></a> <a name="line-450"></a> <span class='hs-varid'>pick</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-layout'>(</span><span class='hs-varid'>selPick</span> <span class='hs-varid'>arity</span><span class='hs-layout'>)</span> <a name="line-451"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>flags_expr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkApps</span> <span class='hs-varid'>pick</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>sel</span><span class='hs-layout'>,</span> <span class='hs-varid'>tag</span><span class='hs-keyglyph'>]</span> <a name="line-452"></a> <span class='hs-varid'>flags_var</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>newLocalVar</span> <span class='hs-layout'>(</span><span class='hs-varid'>fsLit</span> <span class='hs-str'>"flags"</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>exprType</span> <span class='hs-varid'>flags_expr</span><span class='hs-layout'>)</span> <a name="line-453"></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-454"></a> <span class='hs-varid'>elems</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>builtin</span> <span class='hs-layout'>(</span><span class='hs-varid'>selElements</span> <span class='hs-varid'>arity</span> <span class='hs-varid'>ntag</span><span class='hs-layout'>)</span> <a name="line-455"></a> <a name="line-456"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vbndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>vbody</span><span class='hs-layout'>)</span> <a name="line-457"></a> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectBndrsIn</span> <span class='hs-varid'>bndrs</span> <a name="line-458"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>localV</span> <a name="line-459"></a> <span class='hs-varop'>$</span> <span class='hs-keyword'>do</span> <a name="line-460"></a> <span class='hs-varid'>binds</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mapM</span> <span class='hs-layout'>(</span><span class='hs-varid'>pack_var</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-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>flags_var</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-461"></a> <span class='hs-varop'>.</span> <span class='hs-varid'>filter</span> <span class='hs-varid'>isLocalId</span> <a name="line-462"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>varSetElems</span> <span class='hs-varid'>fvs</span> <a name="line-463"></a> <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> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>vectExpr</span> <span class='hs-varid'>body</span> <a name="line-464"></a> <span class='hs-varid'>empty</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>emptyPD</span> <span class='hs-varid'>vty</span> <a name="line-465"></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-conid'>Case</span> <span class='hs-layout'>(</span><span class='hs-varid'>elems</span> <span class='hs-varop'>`App`</span> <span class='hs-varid'>sel</span><span class='hs-layout'>)</span> <span class='hs-varid'>lc</span> <span class='hs-varid'>lty</span> <a name="line-466"></a> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>DEFAULT</span><span class='hs-layout'>,</span> <span class='hs-conid'>[]</span><span class='hs-layout'>,</span> <span class='hs-conid'>Let</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonRec</span> <span class='hs-varid'>flags_var</span> <span class='hs-varid'>flags_expr</span><span class='hs-layout'>)</span> <a name="line-467"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>mkLets</span> <span class='hs-layout'>(</span><span class='hs-varid'>concat</span> <span class='hs-varid'>binds</span><span class='hs-layout'>)</span> <span class='hs-varid'>le</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <a name="line-468"></a> <span class='hs-layout'>(</span><span class='hs-conid'>LitAlt</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkMachInt</span> <span class='hs-num'>0</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-conid'>[]</span><span class='hs-layout'>,</span> <span class='hs-varid'>empty</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-469"></a> <span class='hs-keyword'>let</span> <span class='hs-layout'>(</span><span class='hs-varid'>vect_bndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>lift_bndrs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>unzip</span> <span class='hs-varid'>vbndrs</span> <a name="line-470"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>vect_dc</span><span class='hs-layout'>,</span> <span class='hs-varid'>vect_bndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>lift_bndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>vbody</span><span class='hs-layout'>)</span> <a name="line-471"></a> <a name="line-472"></a> <span class='hs-varid'>proc_alt</span> <span class='hs-keyword'>_</span> <span class='hs-keyword'>_</span> <span class='hs-keyword'>_</span> <span class='hs-keyword'>_</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>panic</span> <span class='hs-str'>"vectAlgCase/proc_alt"</span> <a name="line-473"></a> <a name="line-474"></a> <span class='hs-varid'>mk_vect_alt</span> <span class='hs-varid'>vect_dc</span> <span class='hs-varid'>bndrs</span> <span class='hs-varid'>body</span> <span class='hs-keyglyph'>=</span> <span class='hs-layout'>(</span><span class='hs-conid'>DataAlt</span> <span class='hs-varid'>vect_dc</span><span class='hs-layout'>,</span> <span class='hs-varid'>bndrs</span><span class='hs-layout'>,</span> <span class='hs-varid'>body</span><span class='hs-layout'>)</span> <a name="line-475"></a> <a name="line-476"></a> <span class='hs-varid'>pack_var</span> <span class='hs-varid'>len</span> <span class='hs-varid'>flags</span> <span class='hs-varid'>v</span> <a name="line-477"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-478"></a> <span class='hs-varid'>r</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>lookupVar</span> <span class='hs-varid'>v</span> <a name="line-479"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>r</span> <span class='hs-keyword'>of</span> <a name="line-480"></a> <span class='hs-conid'>Local</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> <span class='hs-keyglyph'>-></span> <a name="line-481"></a> <span class='hs-keyword'>do</span> <a name="line-482"></a> <span class='hs-varid'>lv'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>cloneVar</span> <span class='hs-varid'>lv</span> <a name="line-483"></a> <span class='hs-varid'>expr</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>packPD</span> <span class='hs-layout'>(</span><span class='hs-varid'>idType</span> <span class='hs-varid'>vv</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-conid'>Var</span> <span class='hs-varid'>lv</span><span class='hs-layout'>)</span> <span class='hs-varid'>len</span> <span class='hs-varid'>flags</span> <a name="line-484"></a> <span class='hs-varid'>updLEnv</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>env</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>env</span> <span class='hs-layout'>{</span> <span class='hs-varid'>local_vars</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>extendVarEnv</span> <a name="line-485"></a> <span class='hs-layout'>(</span><span class='hs-varid'>local_vars</span> <span class='hs-varid'>env</span><span class='hs-layout'>)</span> <span class='hs-varid'>v</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> <span class='hs-layout'>}</span><span class='hs-layout'>)</span> <a name="line-486"></a> <span class='hs-varid'>return</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>NonRec</span> <span class='hs-varid'>lv'</span> <span class='hs-varid'>expr</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <a name="line-487"></a> <a name="line-488"></a> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-conid'>[]</span> <a name="line-489"></a> </pre></body> </html>