{-# LANGUAGE BangPatterns, CPP, DeriveFunctor #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE FlexibleContexts #-}
{-# OPTIONS_GHC -Wno-incomplete-uni-patterns #-}
module GHC.CmmToAsm.Reg.Graph.Stats (
RegAllocStats (..),
pprStats,
pprStatsSpills,
pprStatsLifetimes,
pprStatsConflict,
pprStatsLifeConflict,
countSRMs, addSRM
) where
import GHC.Prelude
import qualified GHC.Data.Graph.Color as Color
import GHC.CmmToAsm.Reg.Liveness
import GHC.CmmToAsm.Reg.Graph.Spill
import GHC.CmmToAsm.Reg.Graph.SpillCost
import GHC.CmmToAsm.Reg.Graph.TrivColorable
import GHC.CmmToAsm.Reg.Target
import GHC.CmmToAsm.Instr
import GHC.CmmToAsm.Types
import GHC.Platform
import GHC.Platform.Reg
import GHC.Platform.Reg.Class
import GHC.Types.Unique.FM
import GHC.Types.Unique.Set
import GHC.Utils.Outputable
import GHC.Utils.Monad.State
data RegAllocStats statics instr
= RegAllocStatsStart
{
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raLiveCmm :: [LiveCmmDecl statics instr]
, forall statics instr.
RegAllocStats statics instr -> Graph VirtualReg RegClass RealReg
raGraph :: Color.Graph VirtualReg RegClass RealReg
, forall statics instr. RegAllocStats statics instr -> SpillCostInfo
raSpillCosts :: SpillCostInfo
, forall statics instr. RegAllocStats statics instr -> Platform
raPlatform :: !Platform
}
| RegAllocStatsSpill
{
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raCode :: [LiveCmmDecl statics instr]
, raGraph :: Color.Graph VirtualReg RegClass RealReg
, forall statics instr.
RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
raCoalesced :: UniqFM VirtualReg VirtualReg
, forall statics instr. RegAllocStats statics instr -> SpillStats
raSpillStats :: SpillStats
, raSpillCosts :: SpillCostInfo
, forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raSpilled :: [LiveCmmDecl statics instr]
, raPlatform :: !Platform
}
| RegAllocStatsColored
{
raCode :: [LiveCmmDecl statics instr]
, raGraph :: Color.Graph VirtualReg RegClass RealReg
, forall statics instr.
RegAllocStats statics instr -> Graph VirtualReg RegClass RealReg
raGraphColored :: Color.Graph VirtualReg RegClass RealReg
, raCoalesced :: UniqFM VirtualReg VirtualReg
, forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raCodeCoalesced :: [LiveCmmDecl statics instr]
, forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raPatched :: [LiveCmmDecl statics instr]
, forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raSpillClean :: [LiveCmmDecl statics instr]
, forall statics instr.
RegAllocStats statics instr -> [NatCmmDecl statics instr]
raFinal :: [NatCmmDecl statics instr]
, forall statics instr.
RegAllocStats statics instr -> (Int, Int, Int)
raSRMs :: (Int, Int, Int)
, raPlatform :: !Platform
}
deriving ((forall a b.
(a -> b) -> RegAllocStats statics a -> RegAllocStats statics b)
-> (forall a b.
a -> RegAllocStats statics b -> RegAllocStats statics a)
-> Functor (RegAllocStats statics)
forall a b. a -> RegAllocStats statics b -> RegAllocStats statics a
forall a b.
(a -> b) -> RegAllocStats statics a -> RegAllocStats statics b
forall statics a b.
a -> RegAllocStats statics b -> RegAllocStats statics a
forall statics a b.
(a -> b) -> RegAllocStats statics a -> RegAllocStats statics b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
<$ :: forall a b. a -> RegAllocStats statics b -> RegAllocStats statics a
$c<$ :: forall statics a b.
a -> RegAllocStats statics b -> RegAllocStats statics a
fmap :: forall a b.
(a -> b) -> RegAllocStats statics a -> RegAllocStats statics b
$cfmap :: forall statics a b.
(a -> b) -> RegAllocStats statics a -> RegAllocStats statics b
Functor)
instance (OutputableP Platform statics, OutputableP Platform instr)
=> Outputable (RegAllocStats statics instr) where
ppr :: RegAllocStats statics instr -> SDoc
ppr (s :: RegAllocStats statics instr
s@RegAllocStatsStart{})
= String -> SDoc
text String
"# Start"
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Native code with liveness information."
SDoc -> SDoc -> SDoc
$$ Platform -> [LiveCmmDecl statics instr] -> SDoc
forall env a. OutputableP env a => env -> a -> SDoc
pdoc (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s) (RegAllocStats statics instr -> [LiveCmmDecl statics instr]
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raLiveCmm RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Initial register conflict graph."
SDoc -> SDoc -> SDoc
$$ (RealReg -> SDoc)
-> Triv VirtualReg RegClass RealReg
-> Graph VirtualReg RegClass RealReg
-> SDoc
forall k cls color.
(Uniquable k, Outputable k, Outputable cls, Outputable color) =>
(color -> SDoc) -> Triv k cls color -> Graph k cls color -> SDoc
Color.dotGraph
(Platform -> RealReg -> SDoc
targetRegDotColor (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s))
(Platform
-> (RegClass -> VirtualReg -> Int)
-> (RegClass -> RealReg -> Int)
-> Triv VirtualReg RegClass RealReg
trivColorable (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s)
(Platform -> RegClass -> VirtualReg -> Int
targetVirtualRegSqueeze (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s))
(Platform -> RegClass -> RealReg -> Int
targetRealRegSqueeze (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s)))
(RegAllocStats statics instr -> Graph VirtualReg RegClass RealReg
forall statics instr.
RegAllocStats statics instr -> Graph VirtualReg RegClass RealReg
raGraph RegAllocStats statics instr
s)
ppr (s :: RegAllocStats statics instr
s@RegAllocStatsSpill{}) =
String -> SDoc
text String
"# Spill"
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Code with liveness information."
SDoc -> SDoc -> SDoc
$$ Platform -> [LiveCmmDecl statics instr] -> SDoc
forall env a. OutputableP env a => env -> a -> SDoc
pdoc (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s) (RegAllocStats statics instr -> [LiveCmmDecl statics instr]
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raCode RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ (if (Bool -> Bool
not (Bool -> Bool) -> Bool -> Bool
forall a b. (a -> b) -> a -> b
$ UniqFM VirtualReg VirtualReg -> Bool
forall key elt. UniqFM key elt -> Bool
isNullUFM (UniqFM VirtualReg VirtualReg -> Bool)
-> UniqFM VirtualReg VirtualReg -> Bool
forall a b. (a -> b) -> a -> b
$ RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
forall statics instr.
RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
raCoalesced RegAllocStats statics instr
s)
then String -> SDoc
text String
"# Registers coalesced."
SDoc -> SDoc -> SDoc
$$ UniqFM VirtualReg VirtualReg
-> ([(Unique, VirtualReg)] -> SDoc) -> SDoc
forall key a. UniqFM key a -> ([(Unique, a)] -> SDoc) -> SDoc
pprUFMWithKeys (RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
forall statics instr.
RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
raCoalesced RegAllocStats statics instr
s) ([SDoc] -> SDoc
vcat ([SDoc] -> SDoc)
-> ([(Unique, VirtualReg)] -> [SDoc])
-> [(Unique, VirtualReg)]
-> SDoc
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((Unique, VirtualReg) -> SDoc) -> [(Unique, VirtualReg)] -> [SDoc]
forall a b. (a -> b) -> [a] -> [b]
map (Unique, VirtualReg) -> SDoc
forall a. Outputable a => a -> SDoc
ppr)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
else SDoc
empty)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Spills inserted."
SDoc -> SDoc -> SDoc
$$ SpillStats -> SDoc
forall a. Outputable a => a -> SDoc
ppr (RegAllocStats statics instr -> SpillStats
forall statics instr. RegAllocStats statics instr -> SpillStats
raSpillStats RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Code with spills inserted."
SDoc -> SDoc -> SDoc
$$ Platform -> [LiveCmmDecl statics instr] -> SDoc
forall env a. OutputableP env a => env -> a -> SDoc
pdoc (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s) (RegAllocStats statics instr -> [LiveCmmDecl statics instr]
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raSpilled RegAllocStats statics instr
s)
ppr (s :: RegAllocStats statics instr
s@RegAllocStatsColored { raSRMs :: forall statics instr.
RegAllocStats statics instr -> (Int, Int, Int)
raSRMs = (Int
spills, Int
reloads, Int
moves) })
= String -> SDoc
text String
"# Colored"
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Code with liveness information."
SDoc -> SDoc -> SDoc
$$ Platform -> [LiveCmmDecl statics instr] -> SDoc
forall env a. OutputableP env a => env -> a -> SDoc
pdoc (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s) (RegAllocStats statics instr -> [LiveCmmDecl statics instr]
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raCode RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Register conflict graph (colored)."
SDoc -> SDoc -> SDoc
$$ (RealReg -> SDoc)
-> Triv VirtualReg RegClass RealReg
-> Graph VirtualReg RegClass RealReg
-> SDoc
forall k cls color.
(Uniquable k, Outputable k, Outputable cls, Outputable color) =>
(color -> SDoc) -> Triv k cls color -> Graph k cls color -> SDoc
Color.dotGraph
(Platform -> RealReg -> SDoc
targetRegDotColor (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s))
(Platform
-> (RegClass -> VirtualReg -> Int)
-> (RegClass -> RealReg -> Int)
-> Triv VirtualReg RegClass RealReg
trivColorable (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s)
(Platform -> RegClass -> VirtualReg -> Int
targetVirtualRegSqueeze (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s))
(Platform -> RegClass -> RealReg -> Int
targetRealRegSqueeze (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s)))
(RegAllocStats statics instr -> Graph VirtualReg RegClass RealReg
forall statics instr.
RegAllocStats statics instr -> Graph VirtualReg RegClass RealReg
raGraphColored RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ (if (Bool -> Bool
not (Bool -> Bool) -> Bool -> Bool
forall a b. (a -> b) -> a -> b
$ UniqFM VirtualReg VirtualReg -> Bool
forall key elt. UniqFM key elt -> Bool
isNullUFM (UniqFM VirtualReg VirtualReg -> Bool)
-> UniqFM VirtualReg VirtualReg -> Bool
forall a b. (a -> b) -> a -> b
$ RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
forall statics instr.
RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
raCoalesced RegAllocStats statics instr
s)
then String -> SDoc
text String
"# Registers coalesced."
SDoc -> SDoc -> SDoc
$$ UniqFM VirtualReg VirtualReg
-> ([(Unique, VirtualReg)] -> SDoc) -> SDoc
forall key a. UniqFM key a -> ([(Unique, a)] -> SDoc) -> SDoc
pprUFMWithKeys (RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
forall statics instr.
RegAllocStats statics instr -> UniqFM VirtualReg VirtualReg
raCoalesced RegAllocStats statics instr
s) ([SDoc] -> SDoc
vcat ([SDoc] -> SDoc)
-> ([(Unique, VirtualReg)] -> [SDoc])
-> [(Unique, VirtualReg)]
-> SDoc
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((Unique, VirtualReg) -> SDoc) -> [(Unique, VirtualReg)] -> [SDoc]
forall a b. (a -> b) -> [a] -> [b]
map (Unique, VirtualReg) -> SDoc
forall a. Outputable a => a -> SDoc
ppr)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
else SDoc
empty)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Native code after coalescings applied."
SDoc -> SDoc -> SDoc
$$ Platform -> [LiveCmmDecl statics instr] -> SDoc
forall env a. OutputableP env a => env -> a -> SDoc
pdoc (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s) (RegAllocStats statics instr -> [LiveCmmDecl statics instr]
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raCodeCoalesced RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Native code after register allocation."
SDoc -> SDoc -> SDoc
$$ Platform -> [LiveCmmDecl statics instr] -> SDoc
forall env a. OutputableP env a => env -> a -> SDoc
pdoc (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s) (RegAllocStats statics instr -> [LiveCmmDecl statics instr]
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raPatched RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Clean out unneeded spill/reloads."
SDoc -> SDoc -> SDoc
$$ Platform -> [LiveCmmDecl statics instr] -> SDoc
forall env a. OutputableP env a => env -> a -> SDoc
pdoc (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s) (RegAllocStats statics instr -> [LiveCmmDecl statics instr]
forall statics instr.
RegAllocStats statics instr -> [LiveCmmDecl statics instr]
raSpillClean RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Final code, after rewriting spill/rewrite pseudo instrs."
SDoc -> SDoc -> SDoc
$$ Platform -> [NatCmmDecl statics instr] -> SDoc
forall env a. OutputableP env a => env -> a -> SDoc
pdoc (RegAllocStats statics instr -> Platform
forall statics instr. RegAllocStats statics instr -> Platform
raPlatform RegAllocStats statics instr
s) (RegAllocStats statics instr -> [NatCmmDecl statics instr]
forall statics instr.
RegAllocStats statics instr -> [NatCmmDecl statics instr]
raFinal RegAllocStats statics instr
s)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"# Score:"
SDoc -> SDoc -> SDoc
$$ (String -> SDoc
text String
"# spills inserted: " SDoc -> SDoc -> SDoc
<> Int -> SDoc
int Int
spills)
SDoc -> SDoc -> SDoc
$$ (String -> SDoc
text String
"# reloads inserted: " SDoc -> SDoc -> SDoc
<> Int -> SDoc
int Int
reloads)
SDoc -> SDoc -> SDoc
$$ (String -> SDoc
text String
"# reg-reg moves remaining: " SDoc -> SDoc -> SDoc
<> Int -> SDoc
int Int
moves)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
""
pprStats
:: [RegAllocStats statics instr]
-> Color.Graph VirtualReg RegClass RealReg
-> SDoc
pprStats :: forall statics instr.
[RegAllocStats statics instr]
-> Graph VirtualReg RegClass RealReg -> SDoc
pprStats [RegAllocStats statics instr]
stats Graph VirtualReg RegClass RealReg
graph
= let outSpills :: SDoc
outSpills = [RegAllocStats statics instr] -> SDoc
forall statics instr. [RegAllocStats statics instr] -> SDoc
pprStatsSpills [RegAllocStats statics instr]
stats
outLife :: SDoc
outLife = [RegAllocStats statics instr] -> SDoc
forall statics instr. [RegAllocStats statics instr] -> SDoc
pprStatsLifetimes [RegAllocStats statics instr]
stats
outConflict :: SDoc
outConflict = [RegAllocStats statics instr] -> SDoc
forall statics instr. [RegAllocStats statics instr] -> SDoc
pprStatsConflict [RegAllocStats statics instr]
stats
outScatter :: SDoc
outScatter = [RegAllocStats statics instr]
-> Graph VirtualReg RegClass RealReg -> SDoc
forall statics instr.
[RegAllocStats statics instr]
-> Graph VirtualReg RegClass RealReg -> SDoc
pprStatsLifeConflict [RegAllocStats statics instr]
stats Graph VirtualReg RegClass RealReg
graph
in [SDoc] -> SDoc
vcat [SDoc
outSpills, SDoc
outLife, SDoc
outConflict, SDoc
outScatter]
pprStatsSpills
:: [RegAllocStats statics instr] -> SDoc
pprStatsSpills :: forall statics instr. [RegAllocStats statics instr] -> SDoc
pprStatsSpills [RegAllocStats statics instr]
stats
= let
finals :: [RegAllocStats statics instr]
finals = [ RegAllocStats statics instr
s | s :: RegAllocStats statics instr
s@RegAllocStatsColored{} <- [RegAllocStats statics instr]
stats]
total :: (Int, Int, Int)
total = ((Int, Int, Int) -> (Int, Int, Int) -> (Int, Int, Int))
-> (Int, Int, Int) -> [(Int, Int, Int)] -> (Int, Int, Int)
forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl' (Int, Int, Int) -> (Int, Int, Int) -> (Int, Int, Int)
addSRM (Int
0, Int
0, Int
0)
([(Int, Int, Int)] -> (Int, Int, Int))
-> [(Int, Int, Int)] -> (Int, Int, Int)
forall a b. (a -> b) -> a -> b
$ (RegAllocStats statics instr -> (Int, Int, Int))
-> [RegAllocStats statics instr] -> [(Int, Int, Int)]
forall a b. (a -> b) -> [a] -> [b]
map RegAllocStats statics instr -> (Int, Int, Int)
forall statics instr.
RegAllocStats statics instr -> (Int, Int, Int)
raSRMs [RegAllocStats statics instr]
finals
in ( String -> SDoc
text String
"-- spills-added-total"
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"-- (stores, loads, reg_reg_moves_remaining)"
SDoc -> SDoc -> SDoc
$$ (Int, Int, Int) -> SDoc
forall a. Outputable a => a -> SDoc
ppr (Int, Int, Int)
total
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"")
pprStatsLifetimes
:: [RegAllocStats statics instr] -> SDoc
pprStatsLifetimes :: forall statics instr. [RegAllocStats statics instr] -> SDoc
pprStatsLifetimes [RegAllocStats statics instr]
stats
= let info :: SpillCostInfo
info = (SpillCostInfo -> SpillCostInfo -> SpillCostInfo)
-> SpillCostInfo -> [SpillCostInfo] -> SpillCostInfo
forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl' SpillCostInfo -> SpillCostInfo -> SpillCostInfo
plusSpillCostInfo SpillCostInfo
zeroSpillCostInfo
[ RegAllocStats statics instr -> SpillCostInfo
forall statics instr. RegAllocStats statics instr -> SpillCostInfo
raSpillCosts RegAllocStats statics instr
s
| s :: RegAllocStats statics instr
s@RegAllocStatsStart{} <- [RegAllocStats statics instr]
stats ]
lifeBins :: UniqFM Int (Int, Int)
lifeBins = UniqFM VirtualReg (VirtualReg, Int) -> UniqFM Int (Int, Int)
binLifetimeCount (UniqFM VirtualReg (VirtualReg, Int) -> UniqFM Int (Int, Int))
-> UniqFM VirtualReg (VirtualReg, Int) -> UniqFM Int (Int, Int)
forall a b. (a -> b) -> a -> b
$ SpillCostInfo -> UniqFM VirtualReg (VirtualReg, Int)
lifeMapFromSpillCostInfo SpillCostInfo
info
in ( String -> SDoc
text String
"-- vreg-population-lifetimes"
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"-- (instruction_count, number_of_vregs_that_lived_that_long)"
SDoc -> SDoc -> SDoc
$$ UniqFM Int (Int, Int) -> ([(Int, Int)] -> SDoc) -> SDoc
forall key a. UniqFM key a -> ([a] -> SDoc) -> SDoc
pprUFM UniqFM Int (Int, Int)
lifeBins ([SDoc] -> SDoc
vcat ([SDoc] -> SDoc)
-> ([(Int, Int)] -> [SDoc]) -> [(Int, Int)] -> SDoc
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((Int, Int) -> SDoc) -> [(Int, Int)] -> [SDoc]
forall a b. (a -> b) -> [a] -> [b]
map (Int, Int) -> SDoc
forall a. Outputable a => a -> SDoc
ppr)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"\n")
binLifetimeCount :: UniqFM VirtualReg (VirtualReg, Int) -> UniqFM Int (Int, Int)
binLifetimeCount :: UniqFM VirtualReg (VirtualReg, Int) -> UniqFM Int (Int, Int)
binLifetimeCount UniqFM VirtualReg (VirtualReg, Int)
fm
= let lifes :: [(Int, (Int, Int))]
lifes = (Int -> (Int, (Int, Int))) -> [Int] -> [(Int, (Int, Int))]
forall a b. (a -> b) -> [a] -> [b]
map (\Int
l -> (Int
l, (Int
l, Int
1)))
([Int] -> [(Int, (Int, Int))]) -> [Int] -> [(Int, (Int, Int))]
forall a b. (a -> b) -> a -> b
$ ((VirtualReg, Int) -> Int) -> [(VirtualReg, Int)] -> [Int]
forall a b. (a -> b) -> [a] -> [b]
map (VirtualReg, Int) -> Int
forall a b. (a, b) -> b
snd
([(VirtualReg, Int)] -> [Int]) -> [(VirtualReg, Int)] -> [Int]
forall a b. (a -> b) -> a -> b
$ UniqFM VirtualReg (VirtualReg, Int) -> [(VirtualReg, Int)]
forall key elt. UniqFM key elt -> [elt]
nonDetEltsUFM UniqFM VirtualReg (VirtualReg, Int)
fm
in ((Int, Int) -> (Int, Int) -> (Int, Int))
-> UniqFM Int (Int, Int)
-> [(Int, (Int, Int))]
-> UniqFM Int (Int, Int)
forall key elt.
Uniquable key =>
(elt -> elt -> elt)
-> UniqFM key elt -> [(key, elt)] -> UniqFM key elt
addListToUFM_C
(\(Int
l1, Int
c1) (Int
_, Int
c2) -> (Int
l1, Int
c1 Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
c2))
UniqFM Int (Int, Int)
forall key elt. UniqFM key elt
emptyUFM
[(Int, (Int, Int))]
lifes
pprStatsConflict
:: [RegAllocStats statics instr] -> SDoc
pprStatsConflict :: forall statics instr. [RegAllocStats statics instr] -> SDoc
pprStatsConflict [RegAllocStats statics instr]
stats
= let confMap :: UniqFM Int (Int, Int)
confMap = (UniqFM Int (Int, Int)
-> UniqFM Int (Int, Int) -> UniqFM Int (Int, Int))
-> UniqFM Int (Int, Int)
-> [UniqFM Int (Int, Int)]
-> UniqFM Int (Int, Int)
forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl' (((Int, Int) -> (Int, Int) -> (Int, Int))
-> UniqFM Int (Int, Int)
-> UniqFM Int (Int, Int)
-> UniqFM Int (Int, Int)
forall elt key.
(elt -> elt -> elt)
-> UniqFM key elt -> UniqFM key elt -> UniqFM key elt
plusUFM_C (\(Int
c1, Int
n1) (Int
_, Int
n2) -> (Int
c1, Int
n1 Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
n2)))
UniqFM Int (Int, Int)
forall key elt. UniqFM key elt
emptyUFM
([UniqFM Int (Int, Int)] -> UniqFM Int (Int, Int))
-> [UniqFM Int (Int, Int)] -> UniqFM Int (Int, Int)
forall a b. (a -> b) -> a -> b
$ (Graph VirtualReg RegClass RealReg -> UniqFM Int (Int, Int))
-> [Graph VirtualReg RegClass RealReg] -> [UniqFM Int (Int, Int)]
forall a b. (a -> b) -> [a] -> [b]
map Graph VirtualReg RegClass RealReg -> UniqFM Int (Int, Int)
forall k cls color. Graph k cls color -> UniqFM Int (Int, Int)
Color.slurpNodeConflictCount
[ RegAllocStats statics instr -> Graph VirtualReg RegClass RealReg
forall statics instr.
RegAllocStats statics instr -> Graph VirtualReg RegClass RealReg
raGraph RegAllocStats statics instr
s | s :: RegAllocStats statics instr
s@RegAllocStatsStart{} <- [RegAllocStats statics instr]
stats ]
in ( String -> SDoc
text String
"-- vreg-conflicts"
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"-- (conflict_count, number_of_vregs_that_had_that_many_conflicts)"
SDoc -> SDoc -> SDoc
$$ UniqFM Int (Int, Int) -> ([(Int, Int)] -> SDoc) -> SDoc
forall key a. UniqFM key a -> ([a] -> SDoc) -> SDoc
pprUFM UniqFM Int (Int, Int)
confMap ([SDoc] -> SDoc
vcat ([SDoc] -> SDoc)
-> ([(Int, Int)] -> [SDoc]) -> [(Int, Int)] -> SDoc
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((Int, Int) -> SDoc) -> [(Int, Int)] -> [SDoc]
forall a b. (a -> b) -> [a] -> [b]
map (Int, Int) -> SDoc
forall a. Outputable a => a -> SDoc
ppr)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"\n")
pprStatsLifeConflict
:: [RegAllocStats statics instr]
-> Color.Graph VirtualReg RegClass RealReg
-> SDoc
pprStatsLifeConflict :: forall statics instr.
[RegAllocStats statics instr]
-> Graph VirtualReg RegClass RealReg -> SDoc
pprStatsLifeConflict [RegAllocStats statics instr]
stats Graph VirtualReg RegClass RealReg
graph
= let lifeMap :: UniqFM VirtualReg (VirtualReg, Int)
lifeMap = SpillCostInfo -> UniqFM VirtualReg (VirtualReg, Int)
lifeMapFromSpillCostInfo
(SpillCostInfo -> UniqFM VirtualReg (VirtualReg, Int))
-> SpillCostInfo -> UniqFM VirtualReg (VirtualReg, Int)
forall a b. (a -> b) -> a -> b
$ (SpillCostInfo -> SpillCostInfo -> SpillCostInfo)
-> SpillCostInfo -> [SpillCostInfo] -> SpillCostInfo
forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl' SpillCostInfo -> SpillCostInfo -> SpillCostInfo
plusSpillCostInfo SpillCostInfo
zeroSpillCostInfo
([SpillCostInfo] -> SpillCostInfo)
-> [SpillCostInfo] -> SpillCostInfo
forall a b. (a -> b) -> a -> b
$ [ RegAllocStats statics instr -> SpillCostInfo
forall statics instr. RegAllocStats statics instr -> SpillCostInfo
raSpillCosts RegAllocStats statics instr
s | s :: RegAllocStats statics instr
s@RegAllocStatsStart{} <- [RegAllocStats statics instr]
stats ]
scatter :: [SDoc]
scatter = (VirtualReg -> SDoc) -> [VirtualReg] -> [SDoc]
forall a b. (a -> b) -> [a] -> [b]
map (\VirtualReg
r -> let lifetime :: Int
lifetime = case UniqFM VirtualReg (VirtualReg, Int)
-> VirtualReg -> Maybe (VirtualReg, Int)
forall key elt. Uniquable key => UniqFM key elt -> key -> Maybe elt
lookupUFM UniqFM VirtualReg (VirtualReg, Int)
lifeMap VirtualReg
r of
Just (VirtualReg
_, Int
l) -> Int
l
Maybe (VirtualReg, Int)
Nothing -> Int
0
Just Node VirtualReg RegClass RealReg
node = Graph VirtualReg RegClass RealReg
-> VirtualReg -> Maybe (Node VirtualReg RegClass RealReg)
forall k cls color.
Uniquable k =>
Graph k cls color -> k -> Maybe (Node k cls color)
Color.lookupNode Graph VirtualReg RegClass RealReg
graph VirtualReg
r
in SDoc -> SDoc
parens (SDoc -> SDoc) -> SDoc -> SDoc
forall a b. (a -> b) -> a -> b
$ [SDoc] -> SDoc
hcat ([SDoc] -> SDoc) -> [SDoc] -> SDoc
forall a b. (a -> b) -> a -> b
$ SDoc -> [SDoc] -> [SDoc]
punctuate (String -> SDoc
text String
", ")
[ SDoc -> SDoc
doubleQuotes (SDoc -> SDoc) -> SDoc -> SDoc
forall a b. (a -> b) -> a -> b
$ VirtualReg -> SDoc
forall a. Outputable a => a -> SDoc
ppr (VirtualReg -> SDoc) -> VirtualReg -> SDoc
forall a b. (a -> b) -> a -> b
$ Node VirtualReg RegClass RealReg -> VirtualReg
forall k cls color. Node k cls color -> k
Color.nodeId Node VirtualReg RegClass RealReg
node
, Int -> SDoc
forall a. Outputable a => a -> SDoc
ppr (Int -> SDoc) -> Int -> SDoc
forall a b. (a -> b) -> a -> b
$ UniqSet VirtualReg -> Int
forall a. UniqSet a -> Int
sizeUniqSet (Node VirtualReg RegClass RealReg -> UniqSet VirtualReg
forall k cls color. Node k cls color -> UniqSet k
Color.nodeConflicts Node VirtualReg RegClass RealReg
node)
, Int -> SDoc
forall a. Outputable a => a -> SDoc
ppr (Int -> SDoc) -> Int -> SDoc
forall a b. (a -> b) -> a -> b
$ Int
lifetime ])
([VirtualReg] -> [SDoc]) -> [VirtualReg] -> [SDoc]
forall a b. (a -> b) -> a -> b
$ (Node VirtualReg RegClass RealReg -> VirtualReg)
-> [Node VirtualReg RegClass RealReg] -> [VirtualReg]
forall a b. (a -> b) -> [a] -> [b]
map Node VirtualReg RegClass RealReg -> VirtualReg
forall k cls color. Node k cls color -> k
Color.nodeId
([Node VirtualReg RegClass RealReg] -> [VirtualReg])
-> [Node VirtualReg RegClass RealReg] -> [VirtualReg]
forall a b. (a -> b) -> a -> b
$ UniqFM VirtualReg (Node VirtualReg RegClass RealReg)
-> [Node VirtualReg RegClass RealReg]
forall key elt. UniqFM key elt -> [elt]
nonDetEltsUFM
(UniqFM VirtualReg (Node VirtualReg RegClass RealReg)
-> [Node VirtualReg RegClass RealReg])
-> UniqFM VirtualReg (Node VirtualReg RegClass RealReg)
-> [Node VirtualReg RegClass RealReg]
forall a b. (a -> b) -> a -> b
$ Graph VirtualReg RegClass RealReg
-> UniqFM VirtualReg (Node VirtualReg RegClass RealReg)
forall k cls color.
Graph k cls color -> UniqFM k (Node k cls color)
Color.graphMap Graph VirtualReg RegClass RealReg
graph
in ( String -> SDoc
text String
"-- vreg-conflict-lifetime"
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"-- (vreg, vreg_conflicts, vreg_lifetime)"
SDoc -> SDoc -> SDoc
$$ ([SDoc] -> SDoc
vcat [SDoc]
scatter)
SDoc -> SDoc -> SDoc
$$ String -> SDoc
text String
"\n")
countSRMs
:: Instruction instr
=> LiveCmmDecl statics instr -> (Int, Int, Int)
countSRMs :: forall instr statics.
Instruction instr =>
LiveCmmDecl statics instr -> (Int, Int, Int)
countSRMs LiveCmmDecl statics instr
cmm
= State (Int, Int, Int) (LiveCmmDecl statics instr)
-> (Int, Int, Int) -> (Int, Int, Int)
forall s a. State s a -> s -> s
execState ((LiveBasicBlock instr
-> State (Int, Int, Int) (LiveBasicBlock instr))
-> LiveCmmDecl statics instr
-> State (Int, Int, Int) (LiveCmmDecl statics instr)
forall (m :: * -> *) instr statics.
Monad m =>
(LiveBasicBlock instr -> m (LiveBasicBlock instr))
-> LiveCmmDecl statics instr -> m (LiveCmmDecl statics instr)
mapBlockTopM LiveBasicBlock instr
-> State (Int, Int, Int) (LiveBasicBlock instr)
forall instr.
Instruction instr =>
GenBasicBlock (LiveInstr instr)
-> State (Int, Int, Int) (GenBasicBlock (LiveInstr instr))
countSRM_block LiveCmmDecl statics instr
cmm) (Int
0, Int
0, Int
0)
countSRM_block
:: Instruction instr
=> GenBasicBlock (LiveInstr instr)
-> State (Int, Int, Int) (GenBasicBlock (LiveInstr instr))
countSRM_block :: forall instr.
Instruction instr =>
GenBasicBlock (LiveInstr instr)
-> State (Int, Int, Int) (GenBasicBlock (LiveInstr instr))
countSRM_block (BasicBlock BlockId
i [LiveInstr instr]
instrs)
= do [LiveInstr instr]
instrs' <- (LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr))
-> [LiveInstr instr] -> State (Int, Int, Int) [LiveInstr instr]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr)
forall instr.
Instruction instr =>
LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr)
countSRM_instr [LiveInstr instr]
instrs
GenBasicBlock (LiveInstr instr)
-> State (Int, Int, Int) (GenBasicBlock (LiveInstr instr))
forall (m :: * -> *) a. Monad m => a -> m a
return (GenBasicBlock (LiveInstr instr)
-> State (Int, Int, Int) (GenBasicBlock (LiveInstr instr)))
-> GenBasicBlock (LiveInstr instr)
-> State (Int, Int, Int) (GenBasicBlock (LiveInstr instr))
forall a b. (a -> b) -> a -> b
$ BlockId -> [LiveInstr instr] -> GenBasicBlock (LiveInstr instr)
forall i. BlockId -> [i] -> GenBasicBlock i
BasicBlock BlockId
i [LiveInstr instr]
instrs'
countSRM_instr
:: Instruction instr
=> LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr)
countSRM_instr :: forall instr.
Instruction instr =>
LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr)
countSRM_instr LiveInstr instr
li
| LiveInstr SPILL{} Maybe Liveness
_ <- LiveInstr instr
li
= do ((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ()
forall s. (s -> s) -> State s ()
modify (((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ())
-> ((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ()
forall a b. (a -> b) -> a -> b
$ \(Int
s, Int
r, Int
m) -> (Int
s Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
1, Int
r, Int
m)
LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr)
forall (m :: * -> *) a. Monad m => a -> m a
return LiveInstr instr
li
| LiveInstr RELOAD{} Maybe Liveness
_ <- LiveInstr instr
li
= do ((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ()
forall s. (s -> s) -> State s ()
modify (((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ())
-> ((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ()
forall a b. (a -> b) -> a -> b
$ \(Int
s, Int
r, Int
m) -> (Int
s, Int
r Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
1, Int
m)
LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr)
forall (m :: * -> *) a. Monad m => a -> m a
return LiveInstr instr
li
| LiveInstr InstrSR instr
instr Maybe Liveness
_ <- LiveInstr instr
li
, Just (Reg, Reg)
_ <- InstrSR instr -> Maybe (Reg, Reg)
forall instr. Instruction instr => instr -> Maybe (Reg, Reg)
takeRegRegMoveInstr InstrSR instr
instr
= do ((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ()
forall s. (s -> s) -> State s ()
modify (((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ())
-> ((Int, Int, Int) -> (Int, Int, Int)) -> State (Int, Int, Int) ()
forall a b. (a -> b) -> a -> b
$ \(Int
s, Int
r, Int
m) -> (Int
s, Int
r, Int
m Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
1)
LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr)
forall (m :: * -> *) a. Monad m => a -> m a
return LiveInstr instr
li
| Bool
otherwise
= LiveInstr instr -> State (Int, Int, Int) (LiveInstr instr)
forall (m :: * -> *) a. Monad m => a -> m a
return LiveInstr instr
li
addSRM :: (Int, Int, Int) -> (Int, Int, Int) -> (Int, Int, Int)
addSRM :: (Int, Int, Int) -> (Int, Int, Int) -> (Int, Int, Int)
addSRM (Int
s1, Int
r1, Int
m1) (Int
s2, Int
r2, Int
m2)
= let !s :: Int
s = Int
s1 Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
s2
!r :: Int
r = Int
r1 Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
r2
!m :: Int
m = Int
m1 Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
m2
in (Int
s, Int
r, Int
m)