{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TypeFamilies #-}
module Futhark.Analysis.MemAlias
( analyzeSeqMem,
analyzeGPUMem,
aliasesOf,
MemAliases,
)
where
import Control.Monad.Reader
import Data.Bifunctor
import Data.Function ((&))
import Data.Functor ((<&>))
import qualified Data.Map as M
import Data.Maybe (fromMaybe, mapMaybe)
import qualified Data.Set as S
import Futhark.IR.GPUMem
import Futhark.IR.SeqMem
import Futhark.Util
import Futhark.Util.Pretty
newtype MemAliases = MemAliases (M.Map VName Names)
deriving (Int -> MemAliases -> ShowS
[MemAliases] -> ShowS
MemAliases -> String
(Int -> MemAliases -> ShowS)
-> (MemAliases -> String)
-> ([MemAliases] -> ShowS)
-> Show MemAliases
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [MemAliases] -> ShowS
$cshowList :: [MemAliases] -> ShowS
show :: MemAliases -> String
$cshow :: MemAliases -> String
showsPrec :: Int -> MemAliases -> ShowS
$cshowsPrec :: Int -> MemAliases -> ShowS
Show, MemAliases -> MemAliases -> Bool
(MemAliases -> MemAliases -> Bool)
-> (MemAliases -> MemAliases -> Bool) -> Eq MemAliases
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: MemAliases -> MemAliases -> Bool
$c/= :: MemAliases -> MemAliases -> Bool
== :: MemAliases -> MemAliases -> Bool
$c== :: MemAliases -> MemAliases -> Bool
Eq)
instance Semigroup MemAliases where
(MemAliases Map VName Names
m1) <> :: MemAliases -> MemAliases -> MemAliases
<> (MemAliases Map VName Names
m2) = Map VName Names -> MemAliases
MemAliases (Map VName Names -> MemAliases) -> Map VName Names -> MemAliases
forall a b. (a -> b) -> a -> b
$ (Names -> Names -> Names)
-> Map VName Names -> Map VName Names -> Map VName Names
forall k a. Ord k => (a -> a -> a) -> Map k a -> Map k a -> Map k a
M.unionWith Names -> Names -> Names
forall a. Semigroup a => a -> a -> a
(<>) Map VName Names
m1 Map VName Names
m2
instance Monoid MemAliases where
mempty :: MemAliases
mempty = Map VName Names -> MemAliases
MemAliases Map VName Names
forall a. Monoid a => a
mempty
instance Pretty MemAliases where
ppr :: MemAliases -> Doc
ppr (MemAliases Map VName Names
m) = [Doc] -> Doc
stack ([Doc] -> Doc) -> [Doc] -> Doc
forall a b. (a -> b) -> a -> b
$ ((VName, Names) -> Doc) -> [(VName, Names)] -> [Doc]
forall a b. (a -> b) -> [a] -> [b]
map (VName, Names) -> Doc
forall {a} {a}. (Pretty a, Pretty a) => (a, a) -> Doc
f ([(VName, Names)] -> [Doc]) -> [(VName, Names)] -> [Doc]
forall a b. (a -> b) -> a -> b
$ Map VName Names -> [(VName, Names)]
forall k a. Map k a -> [(k, a)]
M.toList Map VName Names
m
where
f :: (a, a) -> Doc
f (a
v, a
vs) = a -> Doc
forall a. Pretty a => a -> Doc
ppr a
v Doc -> Doc -> Doc
<+> Doc
"aliases:" Doc -> Doc -> Doc
</> Int -> Doc -> Doc
indent Int
2 (Doc -> Doc
oneLine (Doc -> Doc) -> Doc -> Doc
forall a b. (a -> b) -> a -> b
$ a -> Doc
forall a. Pretty a => a -> Doc
ppr a
vs)
addAlias :: VName -> VName -> MemAliases -> MemAliases
addAlias :: VName -> VName -> MemAliases -> MemAliases
addAlias VName
v1 VName
v2 MemAliases
m =
MemAliases
m MemAliases -> MemAliases -> MemAliases
forall a. Semigroup a => a -> a -> a
<> VName -> Names -> MemAliases
singleton VName
v1 (VName -> Names
oneName VName
v2) MemAliases -> MemAliases -> MemAliases
forall a. Semigroup a => a -> a -> a
<> VName -> Names -> MemAliases
singleton VName
v2 Names
forall a. Monoid a => a
mempty
singleton :: VName -> Names -> MemAliases
singleton :: VName -> Names -> MemAliases
singleton VName
v Names
ns = Map VName Names -> MemAliases
MemAliases (Map VName Names -> MemAliases) -> Map VName Names -> MemAliases
forall a b. (a -> b) -> a -> b
$ VName -> Names -> Map VName Names
forall k a. k -> a -> Map k a
M.singleton VName
v Names
ns
aliasesOf :: MemAliases -> VName -> Names
aliasesOf :: MemAliases -> VName -> Names
aliasesOf (MemAliases Map VName Names
m) VName
v = Names -> Maybe Names -> Names
forall a. a -> Maybe a -> a
fromMaybe Names
forall a. Monoid a => a
mempty (Maybe Names -> Names) -> Maybe Names -> Names
forall a b. (a -> b) -> a -> b
$ VName -> Map VName Names -> Maybe Names
forall k a. Ord k => k -> Map k a -> Maybe a
M.lookup VName
v Map VName Names
m
isIn :: VName -> MemAliases -> Bool
isIn :: VName -> MemAliases -> Bool
isIn VName
v (MemAliases Map VName Names
m) = VName
v VName -> Set VName -> Bool
forall a. Ord a => a -> Set a -> Bool
`S.member` Map VName Names -> Set VName
forall k a. Map k a -> Set k
M.keysSet Map VName Names
m
newtype Env inner = Env {forall inner.
Env inner -> MemAliases -> inner -> MemAliasesM inner MemAliases
onInner :: MemAliases -> inner -> MemAliasesM inner MemAliases}
type MemAliasesM inner a = Reader (Env inner) a
analyzeHostOp :: MemAliases -> HostOp GPUMem () -> MemAliasesM (HostOp GPUMem ()) MemAliases
analyzeHostOp :: MemAliases
-> HostOp GPUMem () -> MemAliasesM (HostOp GPUMem ()) MemAliases
analyzeHostOp MemAliases
m (SegOp (SegMap SegLevel
_ SegSpace
_ [Type]
_ KernelBody GPUMem
kbody)) =
Stms GPUMem
-> MemAliases -> MemAliasesM (HostOp GPUMem ()) MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms (KernelBody GPUMem -> Stms GPUMem
forall rep. KernelBody rep -> Stms rep
kernelBodyStms KernelBody GPUMem
kbody) MemAliases
m
analyzeHostOp MemAliases
m (SegOp (SegRed SegLevel
_ SegSpace
_ [SegBinOp GPUMem]
_ [Type]
_ KernelBody GPUMem
kbody)) =
Stms GPUMem
-> MemAliases -> MemAliasesM (HostOp GPUMem ()) MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms (KernelBody GPUMem -> Stms GPUMem
forall rep. KernelBody rep -> Stms rep
kernelBodyStms KernelBody GPUMem
kbody) MemAliases
m
analyzeHostOp MemAliases
m (SegOp (SegScan SegLevel
_ SegSpace
_ [SegBinOp GPUMem]
_ [Type]
_ KernelBody GPUMem
kbody)) =
Stms GPUMem
-> MemAliases -> MemAliasesM (HostOp GPUMem ()) MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms (KernelBody GPUMem -> Stms GPUMem
forall rep. KernelBody rep -> Stms rep
kernelBodyStms KernelBody GPUMem
kbody) MemAliases
m
analyzeHostOp MemAliases
m (SegOp (SegHist SegLevel
_ SegSpace
_ [HistOp GPUMem]
_ [Type]
_ KernelBody GPUMem
kbody)) =
Stms GPUMem
-> MemAliases -> MemAliasesM (HostOp GPUMem ()) MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms (KernelBody GPUMem -> Stms GPUMem
forall rep. KernelBody rep -> Stms rep
kernelBodyStms KernelBody GPUMem
kbody) MemAliases
m
analyzeHostOp MemAliases
m SizeOp {} = MemAliases -> MemAliasesM (HostOp GPUMem ()) MemAliases
forall (f :: * -> *) a. Applicative f => a -> f a
pure MemAliases
m
analyzeHostOp MemAliases
m GPUBody {} = MemAliases -> MemAliasesM (HostOp GPUMem ()) MemAliases
forall (f :: * -> *) a. Applicative f => a -> f a
pure MemAliases
m
analyzeHostOp MemAliases
m (OtherOp ()) = MemAliases -> MemAliasesM (HostOp GPUMem ()) MemAliases
forall (f :: * -> *) a. Applicative f => a -> f a
pure MemAliases
m
analyzeStm :: (Mem rep inner, LetDec rep ~ LetDecMem) => MemAliases -> Stm rep -> MemAliasesM inner MemAliases
analyzeStm :: forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
MemAliases -> Stm rep -> MemAliasesM inner MemAliases
analyzeStm MemAliases
m (Let (Pat [PatElem VName
vname LetDec rep
_]) StmAux (ExpDec rep)
_ (Op (Alloc SubExp
_ Space
_))) =
MemAliases -> ReaderT (Env inner) Identity MemAliases
forall (f :: * -> *) a. Applicative f => a -> f a
pure (MemAliases -> ReaderT (Env inner) Identity MemAliases)
-> MemAliases -> ReaderT (Env inner) Identity MemAliases
forall a b. (a -> b) -> a -> b
$ MemAliases
m MemAliases -> MemAliases -> MemAliases
forall a. Semigroup a => a -> a -> a
<> VName -> Names -> MemAliases
singleton VName
vname Names
forall a. Monoid a => a
mempty
analyzeStm MemAliases
m (Let Pat (LetDec rep)
_ StmAux (ExpDec rep)
_ (Op (Inner inner
inner))) = do
MemAliases -> inner -> ReaderT (Env inner) Identity MemAliases
on_inner <- (Env inner
-> MemAliases -> inner -> ReaderT (Env inner) Identity MemAliases)
-> ReaderT
(Env inner)
Identity
(MemAliases -> inner -> ReaderT (Env inner) Identity MemAliases)
forall r (m :: * -> *) a. MonadReader r m => (r -> a) -> m a
asks Env inner
-> MemAliases -> inner -> ReaderT (Env inner) Identity MemAliases
forall inner.
Env inner -> MemAliases -> inner -> MemAliasesM inner MemAliases
onInner
MemAliases -> inner -> ReaderT (Env inner) Identity MemAliases
on_inner MemAliases
m inner
inner
analyzeStm MemAliases
m (Let Pat (LetDec rep)
pat StmAux (ExpDec rep)
_ (Match [SubExp]
_ [Case (Body rep)]
cases Body rep
defbody MatchDec (BranchType rep)
_)) = do
let bodies :: [Body rep]
bodies = Body rep
defbody Body rep -> [Body rep] -> [Body rep]
forall a. a -> [a] -> [a]
: (Case (Body rep) -> Body rep) -> [Case (Body rep)] -> [Body rep]
forall a b. (a -> b) -> [a] -> [b]
map Case (Body rep) -> Body rep
forall body. Case body -> body
caseBody [Case (Body rep)]
cases
MemAliases
m' <- (MemAliases -> Stms rep -> ReaderT (Env inner) Identity MemAliases)
-> MemAliases
-> [Stms rep]
-> ReaderT (Env inner) Identity MemAliases
forall (t :: * -> *) (m :: * -> *) b a.
(Foldable t, Monad m) =>
(b -> a -> m b) -> b -> t a -> m b
foldM ((Stms rep -> MemAliases -> ReaderT (Env inner) Identity MemAliases)
-> MemAliases
-> Stms rep
-> ReaderT (Env inner) Identity MemAliases
forall a b c. (a -> b -> c) -> b -> a -> c
flip Stms rep -> MemAliases -> ReaderT (Env inner) Identity MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms) MemAliases
m ([Stms rep] -> ReaderT (Env inner) Identity MemAliases)
-> [Stms rep] -> ReaderT (Env inner) Identity MemAliases
forall a b. (a -> b) -> a -> b
$ (Body rep -> Stms rep) -> [Body rep] -> [Stms rep]
forall a b. (a -> b) -> [a] -> [b]
map Body rep -> Stms rep
forall rep. Body rep -> Stms rep
bodyStms [Body rep]
bodies
(Body rep -> [(VName, SubExp)]) -> [Body rep] -> [(VName, SubExp)]
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap ([VName] -> [SubExp] -> [(VName, SubExp)]
forall a b. [a] -> [b] -> [(a, b)]
zip (Pat LetDecMem -> [VName]
forall dec. Pat dec -> [VName]
patNames Pat (LetDec rep)
Pat LetDecMem
pat) ([SubExp] -> [(VName, SubExp)])
-> (Body rep -> [SubExp]) -> Body rep -> [(VName, SubExp)]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SubExpRes -> SubExp) -> [SubExpRes] -> [SubExp]
forall a b. (a -> b) -> [a] -> [b]
map SubExpRes -> SubExp
resSubExp ([SubExpRes] -> [SubExp])
-> (Body rep -> [SubExpRes]) -> Body rep -> [SubExp]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Body rep -> [SubExpRes]
forall rep. Body rep -> [SubExpRes]
bodyResult) [Body rep]
bodies
[(VName, SubExp)]
-> ([(VName, SubExp)] -> [(VName, VName)]) -> [(VName, VName)]
forall a b. a -> (a -> b) -> b
& ((VName, SubExp) -> Maybe (VName, VName))
-> [(VName, SubExp)] -> [(VName, VName)]
forall a b. (a -> Maybe b) -> [a] -> [b]
mapMaybe (MemAliases -> (VName, SubExp) -> Maybe (VName, VName)
filterFun MemAliases
m')
[(VName, VName)] -> ([(VName, VName)] -> MemAliases) -> MemAliases
forall a b. a -> (a -> b) -> b
& ((VName, VName) -> MemAliases -> MemAliases)
-> MemAliases -> [(VName, VName)] -> MemAliases
forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr ((VName -> VName -> MemAliases -> MemAliases)
-> (VName, VName) -> MemAliases -> MemAliases
forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry VName -> VName -> MemAliases -> MemAliases
addAlias) MemAliases
m'
MemAliases
-> (MemAliases -> ReaderT (Env inner) Identity MemAliases)
-> ReaderT (Env inner) Identity MemAliases
forall a b. a -> (a -> b) -> b
& MemAliases -> ReaderT (Env inner) Identity MemAliases
forall (f :: * -> *) a. Applicative f => a -> f a
pure
analyzeStm MemAliases
m (Let Pat (LetDec rep)
pat StmAux (ExpDec rep)
_ (DoLoop [(FParam rep, SubExp)]
params LoopForm rep
_ Body rep
body)) = do
let m_init :: MemAliases
m_init =
((Param FParamMem, SubExp) -> SubExp)
-> [(Param FParamMem, SubExp)] -> [SubExp]
forall a b. (a -> b) -> [a] -> [b]
map (Param FParamMem, SubExp) -> SubExp
forall a b. (a, b) -> b
snd [(FParam rep, SubExp)]
[(Param FParamMem, SubExp)]
params
[SubExp] -> ([SubExp] -> [(VName, SubExp)]) -> [(VName, SubExp)]
forall a b. a -> (a -> b) -> b
& [VName] -> [SubExp] -> [(VName, SubExp)]
forall a b. [a] -> [b] -> [(a, b)]
zip (Pat LetDecMem -> [VName]
forall dec. Pat dec -> [VName]
patNames Pat (LetDec rep)
Pat LetDecMem
pat)
[(VName, SubExp)]
-> ([(VName, SubExp)] -> [(VName, VName)]) -> [(VName, VName)]
forall a b. a -> (a -> b) -> b
& ((VName, SubExp) -> Maybe (VName, VName))
-> [(VName, SubExp)] -> [(VName, VName)]
forall a b. (a -> Maybe b) -> [a] -> [b]
mapMaybe (MemAliases -> (VName, SubExp) -> Maybe (VName, VName)
filterFun MemAliases
m)
[(VName, VName)] -> ([(VName, VName)] -> MemAliases) -> MemAliases
forall a b. a -> (a -> b) -> b
& ((VName, VName) -> MemAliases -> MemAliases)
-> MemAliases -> [(VName, VName)] -> MemAliases
forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr ((VName -> VName -> MemAliases -> MemAliases)
-> (VName, VName) -> MemAliases -> MemAliases
forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry VName -> VName -> MemAliases -> MemAliases
addAlias) MemAliases
m
m_params :: MemAliases
m_params =
((Param FParamMem, SubExp) -> Maybe (VName, VName))
-> [(Param FParamMem, SubExp)] -> [(VName, VName)]
forall a b. (a -> Maybe b) -> [a] -> [b]
mapMaybe (MemAliases -> (VName, SubExp) -> Maybe (VName, VName)
filterFun MemAliases
m_init ((VName, SubExp) -> Maybe (VName, VName))
-> ((Param FParamMem, SubExp) -> (VName, SubExp))
-> (Param FParamMem, SubExp)
-> Maybe (VName, VName)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Param FParamMem -> VName)
-> (Param FParamMem, SubExp) -> (VName, SubExp)
forall (p :: * -> * -> *) a b c.
Bifunctor p =>
(a -> b) -> p a c -> p b c
first Param FParamMem -> VName
forall dec. Param dec -> VName
paramName) [(FParam rep, SubExp)]
[(Param FParamMem, SubExp)]
params
[(VName, VName)] -> ([(VName, VName)] -> MemAliases) -> MemAliases
forall a b. a -> (a -> b) -> b
& ((VName, VName) -> MemAliases -> MemAliases)
-> MemAliases -> [(VName, VName)] -> MemAliases
forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr ((VName -> VName -> MemAliases -> MemAliases)
-> (VName, VName) -> MemAliases -> MemAliases
forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry VName -> VName -> MemAliases -> MemAliases
addAlias) MemAliases
m_init
MemAliases
m_body <- Stms rep -> MemAliases -> ReaderT (Env inner) Identity MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms (Body rep -> Stms rep
forall rep. Body rep -> Stms rep
bodyStms Body rep
body) MemAliases
m_params
[VName] -> [SubExp] -> [(VName, SubExp)]
forall a b. [a] -> [b] -> [(a, b)]
zip (Pat LetDecMem -> [VName]
forall dec. Pat dec -> [VName]
patNames Pat (LetDec rep)
Pat LetDecMem
pat) ((SubExpRes -> SubExp) -> [SubExpRes] -> [SubExp]
forall a b. (a -> b) -> [a] -> [b]
map SubExpRes -> SubExp
resSubExp ([SubExpRes] -> [SubExp]) -> [SubExpRes] -> [SubExp]
forall a b. (a -> b) -> a -> b
$ Body rep -> [SubExpRes]
forall rep. Body rep -> [SubExpRes]
bodyResult Body rep
body)
[(VName, SubExp)]
-> ([(VName, SubExp)] -> [(VName, VName)]) -> [(VName, VName)]
forall a b. a -> (a -> b) -> b
& ((VName, SubExp) -> Maybe (VName, VName))
-> [(VName, SubExp)] -> [(VName, VName)]
forall a b. (a -> Maybe b) -> [a] -> [b]
mapMaybe (MemAliases -> (VName, SubExp) -> Maybe (VName, VName)
filterFun MemAliases
m_body)
[(VName, VName)] -> ([(VName, VName)] -> MemAliases) -> MemAliases
forall a b. a -> (a -> b) -> b
& ((VName, VName) -> MemAliases -> MemAliases)
-> MemAliases -> [(VName, VName)] -> MemAliases
forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr ((VName -> VName -> MemAliases -> MemAliases)
-> (VName, VName) -> MemAliases -> MemAliases
forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry VName -> VName -> MemAliases -> MemAliases
addAlias) MemAliases
m_body
MemAliases
-> (MemAliases -> ReaderT (Env inner) Identity MemAliases)
-> ReaderT (Env inner) Identity MemAliases
forall a b. a -> (a -> b) -> b
& MemAliases -> ReaderT (Env inner) Identity MemAliases
forall (f :: * -> *) a. Applicative f => a -> f a
pure
analyzeStm MemAliases
m Stm rep
_ = MemAliases -> ReaderT (Env inner) Identity MemAliases
forall (f :: * -> *) a. Applicative f => a -> f a
pure MemAliases
m
filterFun :: MemAliases -> (VName, SubExp) -> Maybe (VName, VName)
filterFun :: MemAliases -> (VName, SubExp) -> Maybe (VName, VName)
filterFun MemAliases
m' (VName
v, Var VName
v') | VName
v' VName -> MemAliases -> Bool
`isIn` MemAliases
m' = (VName, VName) -> Maybe (VName, VName)
forall a. a -> Maybe a
Just (VName
v, VName
v')
filterFun MemAliases
_ (VName, SubExp)
_ = Maybe (VName, VName)
forall a. Maybe a
Nothing
analyzeStms :: (Mem rep inner, LetDec rep ~ LetDecMem) => Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms :: forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms =
(MemAliases
-> Seq (Stm rep) -> ReaderT (Env inner) Identity MemAliases)
-> Seq (Stm rep)
-> MemAliases
-> ReaderT (Env inner) Identity MemAliases
forall a b c. (a -> b -> c) -> b -> a -> c
flip ((MemAliases
-> Seq (Stm rep) -> ReaderT (Env inner) Identity MemAliases)
-> Seq (Stm rep)
-> MemAliases
-> ReaderT (Env inner) Identity MemAliases)
-> (MemAliases
-> Seq (Stm rep) -> ReaderT (Env inner) Identity MemAliases)
-> Seq (Stm rep)
-> MemAliases
-> ReaderT (Env inner) Identity MemAliases
forall a b. (a -> b) -> a -> b
$ (MemAliases -> Stm rep -> ReaderT (Env inner) Identity MemAliases)
-> MemAliases
-> Seq (Stm rep)
-> ReaderT (Env inner) Identity MemAliases
forall (t :: * -> *) (m :: * -> *) b a.
(Foldable t, Monad m) =>
(b -> a -> m b) -> b -> t a -> m b
foldM MemAliases -> Stm rep -> ReaderT (Env inner) Identity MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
MemAliases -> Stm rep -> MemAliasesM inner MemAliases
analyzeStm
analyzeFun :: (Mem rep inner, LetDec rep ~ LetDecMem) => FunDef rep -> MemAliasesM inner MemAliases
analyzeFun :: forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
FunDef rep -> MemAliasesM inner MemAliases
analyzeFun FunDef rep
f =
FunDef rep -> [FParam rep]
forall rep. FunDef rep -> [FParam rep]
funDefParams FunDef rep
f
[Param FParamMem]
-> ([Param FParamMem] -> [MemAliases]) -> [MemAliases]
forall a b. a -> (a -> b) -> b
& (Param FParamMem -> Maybe MemAliases)
-> [Param FParamMem] -> [MemAliases]
forall a b. (a -> Maybe b) -> [a] -> [b]
mapMaybe Param FParamMem -> Maybe MemAliases
forall {d} {u} {ret}. Param (MemInfo d u ret) -> Maybe MemAliases
justMem
[MemAliases] -> ([MemAliases] -> MemAliases) -> MemAliases
forall a b. a -> (a -> b) -> b
& [MemAliases] -> MemAliases
forall a. Monoid a => [a] -> a
mconcat
MemAliases
-> (MemAliases -> MemAliasesM inner MemAliases)
-> MemAliasesM inner MemAliases
forall a b. a -> (a -> b) -> b
& Stms rep -> MemAliases -> MemAliasesM inner MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Stms rep -> MemAliases -> MemAliasesM inner MemAliases
analyzeStms (Body rep -> Stms rep
forall rep. Body rep -> Stms rep
bodyStms (Body rep -> Stms rep) -> Body rep -> Stms rep
forall a b. (a -> b) -> a -> b
$ FunDef rep -> Body rep
forall rep. FunDef rep -> Body rep
funDefBody FunDef rep
f)
where
justMem :: Param (MemInfo d u ret) -> Maybe MemAliases
justMem (Param Attrs
_ VName
v (MemMem Space
_)) = MemAliases -> Maybe MemAliases
forall a. a -> Maybe a
Just (MemAliases -> Maybe MemAliases) -> MemAliases -> Maybe MemAliases
forall a b. (a -> b) -> a -> b
$ VName -> Names -> MemAliases
singleton VName
v Names
forall a. Monoid a => a
mempty
justMem Param (MemInfo d u ret)
_ = Maybe MemAliases
forall a. Maybe a
Nothing
transitiveClosure :: MemAliases -> MemAliases
transitiveClosure :: MemAliases -> MemAliases
transitiveClosure ma :: MemAliases
ma@(MemAliases Map VName Names
m) =
(VName -> Names -> MemAliases) -> Map VName Names -> MemAliases
forall m k a. Monoid m => (k -> a -> m) -> Map k a -> m
M.foldMapWithKey
( \VName
k Names
ns ->
Names -> [VName]
namesToList Names
ns
[VName] -> ([VName] -> Names) -> Names
forall a b. a -> (a -> b) -> b
& (VName -> Names) -> [VName] -> Names
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap (MemAliases -> VName -> Names
aliasesOf MemAliases
ma)
Names -> (Names -> MemAliases) -> MemAliases
forall a b. a -> (a -> b) -> b
& VName -> Names -> MemAliases
singleton VName
k
)
Map VName Names
m
MemAliases -> MemAliases -> MemAliases
forall a. Semigroup a => a -> a -> a
<> MemAliases
ma
analyzeSeqMem :: Prog SeqMem -> MemAliases
analyzeSeqMem :: Prog SeqMem -> MemAliases
analyzeSeqMem Prog SeqMem
prog = MemAliases -> MemAliases
completeBijection (MemAliases -> MemAliases) -> MemAliases -> MemAliases
forall a b. (a -> b) -> a -> b
$ Reader (Env ()) MemAliases -> Env () -> MemAliases
forall r a. Reader r a -> r -> a
runReader (Prog SeqMem -> Reader (Env ()) MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Prog rep -> MemAliasesM inner MemAliases
analyze Prog SeqMem
prog) (Env () -> MemAliases) -> Env () -> MemAliases
forall a b. (a -> b) -> a -> b
$ (MemAliases -> () -> Reader (Env ()) MemAliases) -> Env ()
forall inner.
(MemAliases -> inner -> MemAliasesM inner MemAliases) -> Env inner
Env ((MemAliases -> () -> Reader (Env ()) MemAliases) -> Env ())
-> (MemAliases -> () -> Reader (Env ()) MemAliases) -> Env ()
forall a b. (a -> b) -> a -> b
$ \MemAliases
x ()
_ -> MemAliases -> Reader (Env ()) MemAliases
forall (f :: * -> *) a. Applicative f => a -> f a
pure MemAliases
x
analyzeGPUMem :: Prog GPUMem -> MemAliases
analyzeGPUMem :: Prog GPUMem -> MemAliases
analyzeGPUMem Prog GPUMem
prog = MemAliases -> MemAliases
completeBijection (MemAliases -> MemAliases) -> MemAliases -> MemAliases
forall a b. (a -> b) -> a -> b
$ MemAliasesM (HostOp GPUMem ()) MemAliases
-> Env (HostOp GPUMem ()) -> MemAliases
forall r a. Reader r a -> r -> a
runReader (Prog GPUMem -> MemAliasesM (HostOp GPUMem ()) MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Prog rep -> MemAliasesM inner MemAliases
analyze Prog GPUMem
prog) (Env (HostOp GPUMem ()) -> MemAliases)
-> Env (HostOp GPUMem ()) -> MemAliases
forall a b. (a -> b) -> a -> b
$ (MemAliases
-> HostOp GPUMem () -> MemAliasesM (HostOp GPUMem ()) MemAliases)
-> Env (HostOp GPUMem ())
forall inner.
(MemAliases -> inner -> MemAliasesM inner MemAliases) -> Env inner
Env MemAliases
-> HostOp GPUMem () -> MemAliasesM (HostOp GPUMem ()) MemAliases
analyzeHostOp
analyze :: (Mem rep inner, LetDec rep ~ LetDecMem) => Prog rep -> MemAliasesM inner MemAliases
analyze :: forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
Prog rep -> MemAliasesM inner MemAliases
analyze Prog rep
prog =
Prog rep -> [FunDef rep]
forall rep. Prog rep -> [FunDef rep]
progFuns Prog rep
prog
[FunDef rep]
-> ([FunDef rep] -> ReaderT (Env inner) Identity MemAliases)
-> ReaderT (Env inner) Identity MemAliases
forall a b. a -> (a -> b) -> b
& (MemAliases
-> FunDef rep -> ReaderT (Env inner) Identity MemAliases)
-> MemAliases
-> [FunDef rep]
-> ReaderT (Env inner) Identity MemAliases
forall (t :: * -> *) (m :: * -> *) b a.
(Foldable t, Monad m) =>
(b -> a -> m b) -> b -> t a -> m b
foldM (\MemAliases
m FunDef rep
f -> MemAliases -> MemAliases -> MemAliases
forall a. Semigroup a => a -> a -> a
(<>) MemAliases
m (MemAliases -> MemAliases)
-> ReaderT (Env inner) Identity MemAliases
-> ReaderT (Env inner) Identity MemAliases
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> FunDef rep -> ReaderT (Env inner) Identity MemAliases
forall rep inner.
(Mem rep inner, LetDec rep ~ LetDecMem) =>
FunDef rep -> MemAliasesM inner MemAliases
analyzeFun FunDef rep
f) (Map VName Names -> MemAliases
MemAliases Map VName Names
forall a. Monoid a => a
mempty)
ReaderT (Env inner) Identity MemAliases
-> (MemAliases -> MemAliases)
-> ReaderT (Env inner) Identity MemAliases
forall (f :: * -> *) a b. Functor f => f a -> (a -> b) -> f b
<&> (MemAliases -> MemAliases) -> MemAliases -> MemAliases
forall a. Eq a => (a -> a) -> a -> a
fixPoint MemAliases -> MemAliases
transitiveClosure
completeBijection :: MemAliases -> MemAliases
completeBijection :: MemAliases -> MemAliases
completeBijection ma :: MemAliases
ma@(MemAliases Map VName Names
m) =
(VName -> Names -> MemAliases) -> Map VName Names -> MemAliases
forall m k a. Monoid m => (k -> a -> m) -> Map k a -> m
M.foldMapWithKey (\VName
k Names
ns -> (VName -> MemAliases) -> [VName] -> MemAliases
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap (VName -> Names -> MemAliases
`singleton` VName -> Names
oneName VName
k) (Names -> [VName]
namesToList Names
ns)) Map VName Names
m MemAliases -> MemAliases -> MemAliases
forall a. Semigroup a => a -> a -> a
<> MemAliases
ma