Commit 9c9ff443 authored by rl@cse.unsw.edu.au's avatar rl@cse.unsw.edu.au
Browse files

Add utility function for vectorisation

parent 49012a1f
......@@ -2,7 +2,8 @@ module VectUtils (
collectAnnTypeBinders, collectAnnTypeArgs, isAnnTypeArg,
splitClosureTy,
mkPADictType, mkPArrayType,
paDictArgType, paDictOfType
paDictArgType, paDictOfType,
lookupPArrayFamInst
) where
#include "HsVersions.h"
......@@ -104,3 +105,6 @@ paDFunApply dfun tys
dicts <- mapM paDictOfType tys
return $ mkApps (mkTyApps dfun tys) dicts
lookupPArrayFamInst :: Type -> VM (TyCon, [Type])
lookupPArrayFamInst ty = builtin parrayTyCon >>= (`lookupFamInst` [ty])
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment