diff options
Diffstat (limited to 'Intermediate.hs')
| -rw-r--r-- | Intermediate.hs | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/Intermediate.hs b/Intermediate.hs index b0f12b9..2705431 100644 --- a/Intermediate.hs +++ b/Intermediate.hs @@ -5,6 +5,7 @@ import qualified Data.Map.Strict as Map import Data.Maybe import AST +import Util data IRProgram = @@ -60,20 +61,24 @@ termOf (BB _ _ t) = t class AllRefs a where allRefs :: a -> [Ref] +instance AllRefs BB where + allRefs (BB _ inss term) = + sortUniq $ concatMap (allRefs . snd) inss ++ allRefs term + instance AllRefs InsCode where allRefs (IAssign r) = [r] allRefs (IParam _) = [] allRefs (IClosure _) = [] allRefs (IData _) = [] - allRefs (ICallC r rs) = r : rs - allRefs (IAllocClo _ rs) = rs + allRefs (ICallC r rs) = sortUniq (r : rs) + allRefs (IAllocClo _ rs) = sortUniq rs allRefs (IDiscard r) = [r] instance AllRefs Terminator where allRefs (IBr r _ _) = [r] allRefs (IJmp _) = [] allRefs (IRet r) = [r] - allRefs (ITailC r rs) = r : rs + allRefs (ITailC r rs) = sortUniq (r : rs) allRefs IExit = [] allRefs IUnknown = [] |