X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=compiler%2Fvectorise%2FVectorise.hs;h=d85ef6ab4f751682c714cda5aed6ac8df390acc8;hb=1c2f1ceb04aa84f0e014ec6750d8a08355200145;hp=63575b90f16f742c6379cc4150268bc351351ec0;hpb=8b3ebc412fc61eb1f2a6129190d85fcdd851235e;p=ghc-hetmet.git diff --git a/compiler/vectorise/Vectorise.hs b/compiler/vectorise/Vectorise.hs index 63575b9..d85ef6a 100644 --- a/compiler/vectorise/Vectorise.hs +++ b/compiler/vectorise/Vectorise.hs @@ -216,6 +216,8 @@ vectLiteral lit return (Lit lit, lexpr) vectPolyExpr :: CoreExprWithFVs -> VM VExpr +vectPolyExpr (_, AnnNote note expr) + = liftM (vNote note) $ vectPolyExpr expr vectPolyExpr expr = polyAbstract tvs $ \abstract -> do @@ -446,6 +448,7 @@ packLiftingContext len shape tag fvs res_ty p $ varSetElems fvs (vexpr, lexpr) <- p return (vexpr, Let (NonRec sel_var sel_expr) + . mkLets (concat bnds) $ Case len lc_var res_ty [(DEFAULT, [], lexpr)]) packFreeVar :: CoreExpr -> CoreExpr -> Var -> VM [CoreBind]