X-Git-Url: http://git.megacz.com/?p=ghc-hetmet.git;a=blobdiff_plain;f=compiler%2Fvectorise%2FVectorise%2FExp.hs;h=862a760a43a5cd7322acac10c364da2f5a2dbacf;hp=d792fd681973a330709bc9dd741e12c79a9337f9;hb=21703cf93de9e93f6b278b4d46f8511a813cbeda;hpb=1c4593fd4e45a1ea5a32551408866e73840251f1 diff --git a/compiler/vectorise/Vectorise/Exp.hs b/compiler/vectorise/Vectorise/Exp.hs index d792fd6..862a760 100644 --- a/compiler/vectorise/Vectorise/Exp.hs +++ b/compiler/vectorise/Vectorise/Exp.hs @@ -176,8 +176,8 @@ vectScalarLam vectScalarLam args body = do scalars <- globalScalars - onlyIfV (all is_scalar_ty arg_tys - && is_scalar_ty res_ty + onlyIfV (all is_prim_ty arg_tys + && is_prim_ty res_ty && is_scalar (extendVarSetList scalars args) body && uses scalars body) $ do