X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Fprofiling%2FSCCfinal.lhs;h=97aedf223f3ad001dd1c93d6204463637ef8c069;hb=ef5b4b146aa172d8ac10f39b5eb3d7a0f948d8f1;hp=24e0fb3a313496dd84f686fa5e7cadc5b13efd9d;hpb=7a3bd641457666e10d0a47be9f22762e03defbf0;p=ghc-hetmet.git diff --git a/ghc/compiler/profiling/SCCfinal.lhs b/ghc/compiler/profiling/SCCfinal.lhs index 24e0fb3..97aedf2 100644 --- a/ghc/compiler/profiling/SCCfinal.lhs +++ b/ghc/compiler/profiling/SCCfinal.lhs @@ -1,5 +1,5 @@ % -% (c) The GRASP/AQUA Project, Glasgow University, 1992-1996 +% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998 % \section[SCCfinal]{Modify and collect code generation for final STG program} @@ -23,73 +23,76 @@ This is now a sort-of-normal STG-to-STG pass (WDP 94/06), run by stg2stg. * "Distributes" given cost-centres to all as-yet-unmarked RHSs. \begin{code} -#include "HsVersions.h" - module SCCfinal ( stgMassageForProfiling ) where -IMP_Ubiq(){-uitous-} +#include "HsVersions.h" import StgSyn -import CmdLineOpts ( opt_AutoSccsOnIndividualCafs, - opt_CompilingGhcInternals - ) +import CmdLineOpts ( DynFlags, opt_AutoSccsOnIndividualCafs ) import CostCentre -- lots of things -import Id ( idType, mkSysLocal, emptyIdSet ) -import Maybes ( maybeToBool ) -import PprStyle -- ToDo: rm -import SrcLoc ( noSrcLoc ) -import Type ( splitSigmaTy, getFunTy_maybe ) -import UniqSupply ( getUnique, splitUniqSupply ) -import Util ( removeDups, assertPanic ) +import Id ( Id ) +import Module ( Module ) +import UniqSupply ( uniqFromSupply, splitUniqSupply, UniqSupply ) +import Unique ( Unique ) +import VarSet +import ListSetOps ( removeDups ) +import Outputable infixr 9 `thenMM`, `thenMM_` \end{code} \begin{code} -type CollectedCCs = ([CostCentre], -- locally defined ones - [CostCentre]) -- ones needing "extern" decls - stgMassageForProfiling - :: FAST_STRING -> FAST_STRING -- module name, group name + :: DynFlags + -> Module -- module name -> UniqSupply -- unique supply -> [StgBinding] -- input -> (CollectedCCs, [StgBinding]) -stgMassageForProfiling mod_name grp_name us stg_binds +stgMassageForProfiling dflags mod_name us stg_binds = let - ((local_ccs, extern_ccs), + ((local_ccs, extern_ccs, cc_stacks), stg_binds2) - = initMM mod_name us (mapMM do_top_binding stg_binds) + = initMM mod_name us (do_top_bindings stg_binds) - fixed_ccs - = if do_auto_sccs_on_cafs || doing_prelude - then [] -- don't need "all CAFs" CC (for Prelude, we use PreludeCC) - else [all_cafs_cc] + (fixed_ccs, fixed_cc_stacks) + = if opt_AutoSccsOnIndividualCafs + then ([],[]) -- don't need "all CAFs" CC + -- (for Prelude, we use PreludeCC) + else ([all_cafs_cc], [all_cafs_ccs]) local_ccs_no_dups = fst (removeDups cmpCostCentre local_ccs) extern_ccs_no_dups = fst (removeDups cmpCostCentre extern_ccs) in - ((fixed_ccs ++ local_ccs_no_dups, extern_ccs_no_dups), stg_binds2) + ((fixed_ccs ++ local_ccs_no_dups, + extern_ccs_no_dups, + fixed_cc_stacks ++ cc_stacks), stg_binds2) where - do_auto_sccs_on_cafs = opt_AutoSccsOnIndividualCafs -- only use! - doing_prelude = opt_CompilingGhcInternals - all_cafs_cc = if doing_prelude - then preludeCafsCostCentre - else mkAllCafsCC mod_name grp_name + all_cafs_cc = mkAllCafsCC mod_name + all_cafs_ccs = mkSingletonCCS all_cafs_cc ---------- - do_top_binding :: StgBinding -> MassageM StgBinding + do_top_bindings :: [StgBinding] -> MassageM [StgBinding] - do_top_binding (StgNonRec b rhs) - = do_top_rhs b rhs `thenMM` \ rhs' -> - returnMM (StgNonRec b rhs') + do_top_bindings [] = returnMM [] - do_top_binding (StgRec pairs) - = mapMM do_pair pairs `thenMM` \ pairs2 -> - returnMM (StgRec pairs2) + do_top_bindings (StgNonRec b rhs : bs) + = do_top_rhs b rhs `thenMM` \ rhs' -> + addTopLevelIshId b ( + do_top_bindings bs `thenMM` \bs' -> + returnMM (StgNonRec b rhs' : bs') + ) + + do_top_bindings (StgRec pairs : bs) + = addTopLevelIshIds binders ( + mapMM do_pair pairs `thenMM` \ pairs2 -> + do_top_bindings bs `thenMM` \ bs' -> + returnMM (StgRec pairs2 : bs') + ) where + binders = map fst pairs do_pair (b, rhs) = do_top_rhs b rhs `thenMM` \ rhs2 -> returnMM (b, rhs2) @@ -97,165 +100,148 @@ stgMassageForProfiling mod_name grp_name us stg_binds ---------- do_top_rhs :: Id -> StgRhs -> MassageM StgRhs - do_top_rhs binder (StgRhsClosure _ bi fv u [] (StgSCC ty cc (StgCon con args lvs))) - | not (isSccCountCostCentre cc) + do_top_rhs binder (StgRhsClosure _ bi fv u srt [] (StgSCC cc (StgConApp con args))) + | not (isSccCountCostCentre cc) && not (isDllConApp dflags con args) -- Trivial _scc_ around nothing but static data -- Eliminate _scc_ ... and turn into StgRhsCon - = returnMM (StgRhsCon dontCareCostCentre con args) + -- isDllConApp checks for LitLit args too + = returnMM (StgRhsCon dontCareCCS con args) + +{- Can't do this one with cost-centre stacks: --SDM do_top_rhs binder (StgRhsClosure no_cc bi fv u [] (StgSCC ty cc expr)) - | (noCostCentreAttached no_cc || currentOrSubsumedCosts no_cc) + | (noCCSAttached no_cc || currentOrSubsumedCCS no_cc) && not (isSccCountCostCentre cc) -- Top level CAF without a cost centre attached -- Attach and collect cc of trivial _scc_ in body = collectCC cc `thenMM_` set_prevailing_cc cc (do_expr expr) `thenMM` \ expr' -> returnMM (StgRhsClosure cc bi fv u [] expr') +-} - do_top_rhs binder (StgRhsClosure no_cc bi fv u [] body) - | noCostCentreAttached no_cc || currentOrSubsumedCosts no_cc + do_top_rhs binder (StgRhsClosure no_cc bi fv u srt [] body) + | noCCSAttached no_cc || currentOrSubsumedCCS no_cc -- Top level CAF without a cost centre attached -- Attach CAF cc (collect if individual CAF ccs) - = let - (collect, caf_cc) - = if do_auto_sccs_on_cafs then - (True, mkAutoCC binder mod_name grp_name IsCafCC) - else - (False, all_cafs_cc) - in - (if collect then collectCC caf_cc else nopMM) `thenMM_` - set_prevailing_cc caf_cc (do_expr body) `thenMM` \ body' -> - returnMM (StgRhsClosure caf_cc bi fv u [] body') - - do_top_rhs binder (StgRhsClosure cc bi fv u [] body) + = (if opt_AutoSccsOnIndividualCafs + then let cc = mkAutoCC binder mod_name CafCC + ccs = mkSingletonCCS cc + in + collectCC cc `thenMM_` + collectCCS ccs `thenMM_` + returnMM ccs + else + returnMM all_cafs_ccs) `thenMM` \ caf_ccs -> + set_prevailing_cc caf_ccs (do_expr body) `thenMM` \ body' -> + returnMM (StgRhsClosure caf_ccs bi fv u srt [] body') + + do_top_rhs binder (StgRhsClosure cc bi fv u srt [] body) -- Top level CAF with cost centre attached -- Should this be a CAF cc ??? Does this ever occur ??? - = trace ("SCCfinal: CAF with cc: " ++ showCostCentre PprDebug False cc) $ - collectCC cc `thenMM_` - set_prevailing_cc cc (do_expr body) `thenMM` \ body' -> - returnMM (StgRhsClosure cc bi fv u [] body') - - do_top_rhs binder (StgRhsClosure _ bi fv u args (StgSCC ty cc expr)) - | not (isSccCountCostCentre cc) - -- Top level function with trivial _scc_ in body - -- Attach and collect cc of trivial _scc_ - = collectCC cc `thenMM_` - set_prevailing_cc cc (do_expr expr) `thenMM` \ expr' -> - returnMM (StgRhsClosure cc bi fv u args expr') + = pprPanic "SCCfinal: CAF with cc:" (ppr cc) - do_top_rhs binder (StgRhsClosure cc bi fv u args body) + do_top_rhs binder (StgRhsClosure no_ccs bi fv u srt args body) -- Top level function, probably subsumed - = let - (cc_closure, cc_body) - = if noCostCentreAttached cc - then (subsumedCosts, useCurrentCostCentre) - else (cc, cc) - in - set_prevailing_cc cc_body (do_expr body) `thenMM` \ body' -> - returnMM (StgRhsClosure cc_closure bi fv u args body') + | noCCSAttached no_ccs + = set_lambda_cc (do_expr body) `thenMM` \ body' -> + returnMM (StgRhsClosure subsumedCCS bi fv u srt args body') - do_top_rhs binder (StgRhsCon cc con args) + | otherwise + = pprPanic "SCCfinal: CAF with cc:" (ppr no_ccs) + + do_top_rhs binder (StgRhsCon ccs con args) -- Top-level (static) data is not counted in heap - -- profiles; nor do we set CCC from it; so we + -- profiles; nor do we set CCCS from it; so we -- just slam in dontCareCostCentre - = returnMM (StgRhsCon dontCareCostCentre con args) + = returnMM (StgRhsCon dontCareCCS con args) ------ do_expr :: StgExpr -> MassageM StgExpr - do_expr (StgApp fn args lvs) - = boxHigherOrderArgs (StgApp fn) args lvs + do_expr (StgLit l) = returnMM (StgLit l) + + do_expr (StgApp fn args) + = boxHigherOrderArgs (StgApp fn) args - do_expr (StgCon con args lvs) - = boxHigherOrderArgs (StgCon con) args lvs + do_expr (StgConApp con args) + = boxHigherOrderArgs (\args -> StgConApp con args) args - do_expr (StgPrim op args lvs) - = boxHigherOrderArgs (StgPrim op) args lvs + do_expr (StgOpApp con args res_ty) + = boxHigherOrderArgs (\args -> StgOpApp con args res_ty) args - do_expr (StgSCC ty cc expr) -- Ha, we found a cost centre! - = collectCC cc `thenMM_` - set_prevailing_cc cc (do_expr expr) `thenMM` \ expr' -> - returnMM (StgSCC ty cc expr') + do_expr (StgSCC cc expr) -- Ha, we found a cost centre! + = collectCC cc `thenMM_` + do_expr expr `thenMM` \ expr' -> + returnMM (StgSCC cc expr') - do_expr (StgCase expr fv1 fv2 uniq alts) + do_expr (StgCase expr fv1 fv2 bndr srt alt_type alts) = do_expr expr `thenMM` \ expr' -> - do_alts alts `thenMM` \ alts' -> - returnMM (StgCase expr' fv1 fv2 uniq alts') + mapMM do_alt alts `thenMM` \ alts' -> + returnMM (StgCase expr' fv1 fv2 bndr srt alt_type alts') where - do_alts (StgAlgAlts ty alts def) - = mapMM do_alt alts `thenMM` \ alts' -> - do_deflt def `thenMM` \ def' -> - returnMM (StgAlgAlts ty alts' def') - where - do_alt (id, bs, use_mask, e) - = do_expr e `thenMM` \ e' -> - returnMM (id, bs, use_mask, e') - - do_alts (StgPrimAlts ty alts def) - = mapMM do_alt alts `thenMM` \ alts' -> - do_deflt def `thenMM` \ def' -> - returnMM (StgPrimAlts ty alts' def') - where - do_alt (l,e) - = do_expr e `thenMM` \ e' -> - returnMM (l,e') - - do_deflt StgNoDefault = returnMM StgNoDefault - do_deflt (StgBindDefault b is_used e) - = do_expr e `thenMM` \ e' -> - returnMM (StgBindDefault b is_used e') + do_alt (id, bs, use_mask, e) + = do_expr e `thenMM` \ e' -> + returnMM (id, bs, use_mask, e') do_expr (StgLet b e) - = do_binding b `thenMM` \ b' -> - do_expr e `thenMM` \ e' -> - returnMM (StgLet b' e') - - do_expr (StgLetNoEscape lvs1 lvs2 rhs body) - = do_binding rhs `thenMM` \ rhs' -> - do_expr body `thenMM` \ body' -> - returnMM (StgLetNoEscape lvs1 lvs2 rhs' body') - - ---------- - do_binding :: StgBinding -> MassageM StgBinding - - do_binding (StgNonRec b rhs) - = do_rhs rhs `thenMM` \ rhs' -> - returnMM (StgNonRec b rhs') - - do_binding (StgRec pairs) - = mapMM do_pair pairs `thenMM` \ new_pairs -> - returnMM (StgRec new_pairs) + = do_let b e `thenMM` \ (b,e) -> + returnMM (StgLet b e) + + do_expr (StgLetNoEscape lvs1 lvs2 b e) + = do_let b e `thenMM` \ (b,e) -> + returnMM (StgLetNoEscape lvs1 lvs2 b e) + +#ifdef DEBUG + do_expr other = pprPanic "SCCfinal.do_expr" (ppr other) +#endif + + ---------------------------------- + + do_let (StgNonRec b rhs) e + = do_rhs rhs `thenMM` \ rhs' -> + addTopLevelIshId b ( + do_expr e `thenMM` \ e' -> + returnMM (StgNonRec b rhs',e') + ) + + do_let (StgRec pairs) e + = addTopLevelIshIds binders ( + mapMM do_pair pairs `thenMM` \ pairs' -> + do_expr e `thenMM` \ e' -> + returnMM (StgRec pairs', e') + ) where - do_pair (b, rhs) - = do_rhs rhs `thenMM` \ rhs' -> - returnMM (b, rhs') + binders = map fst pairs + do_pair (b, rhs) + = do_rhs rhs `thenMM` \ rhs2 -> + returnMM (b, rhs2) + ---------------------------------- do_rhs :: StgRhs -> MassageM StgRhs -- We play much the same game as we did in do_top_rhs above; -- but we don't have to worry about cafs etc. - do_rhs (StgRhsClosure _ bi fv u [] (StgSCC ty cc (StgCon con args lvs))) +{- + do_rhs (StgRhsClosure closure_cc bi fv u [] (StgSCC ty cc (StgCon (DataCon con) args _))) | not (isSccCountCostCentre cc) = collectCC cc `thenMM_` returnMM (StgRhsCon cc con args) +-} - do_rhs (StgRhsClosure _ bi fv u args (StgSCC ty cc expr)) - | not (isSccCountCostCentre cc) - = collectCC cc `thenMM_` - set_prevailing_cc cc (do_expr expr) `thenMM` \ expr' -> - returnMM (StgRhsClosure cc bi fv u args expr') - - do_rhs (StgRhsClosure cc bi fv u args body) - = set_prevailing_cc_maybe cc $ \ cc' -> - set_lambda_cc (do_expr body) `thenMM` \ body' -> - returnMM (StgRhsClosure cc' bi fv u args body') + do_rhs (StgRhsClosure _ bi fv u srt args expr) + = slurpSCCs currentCCS expr `thenMM` \ (expr', ccs) -> + do_expr expr' `thenMM` \ expr'' -> + returnMM (StgRhsClosure ccs bi fv u srt args expr'') + where + slurpSCCs ccs (StgSCC cc e) + = collectCC cc `thenMM_` + slurpSCCs ccs e `thenMM` \ (e', ccs') -> + returnMM (e', pushCCOnCCS cc ccs') + slurpSCCs ccs e + = returnMM (e, ccs) do_rhs (StgRhsCon cc con args) - = set_prevailing_cc_maybe cc $ \ cc' -> - returnMM (StgRhsCon cc' con args) - - -- ToDo: Box args and sort out any let bindings ??? - -- Nope: maybe later? WDP 94/06 + = returnMM (StgRhsCon currentCCS con args) \end{code} %************************************************************************ @@ -264,61 +250,53 @@ stgMassageForProfiling mod_name grp_name us stg_binds %* * %************************************************************************ +Boxing is *turned off* at the moment, until we can figure out how to +do it properly in general. + \begin{code} boxHigherOrderArgs - :: ([StgArg] -> StgLiveVars -> StgExpr) - -- An application lacking its arguments and live-var info + :: ([StgArg] -> StgExpr) + -- An application lacking its arguments -> [StgArg] -- arguments which we might box - -> StgLiveVars -- live var info, which we do *not* try - -- to maintain/update (setStgVarInfo will - -- do that) -> MassageM StgExpr -boxHigherOrderArgs almost_expr args live_vars - = returnMM (almost_expr args live_vars) - -{- No boxing for now ... should be moved to desugarer and preserved ... - -boxHigherOrderArgs almost_expr args live_vars - = get_prevailing_cc `thenMM` \ cc -> - if (isCafCC cc || isDictCC cc) then - -- no boxing required inside CAF/DICT cc - -- since CAF/DICT functions are subsumed anyway - returnMM (almost_expr args live_vars) - else - mapAccumMM do_arg [] args `thenMM` \ (let_bindings, new_args) -> - returnMM (foldr (mk_stg_let cc) (almost_expr new_args live_vars) let_bindings) +#ifndef PROF_DO_BOXING +boxHigherOrderArgs almost_expr args + = returnMM (almost_expr args) +#else +boxHigherOrderArgs almost_expr args + = getTopLevelIshIds `thenMM` \ ids -> + mapAccumMM (do_arg ids) [] args `thenMM` \ (let_bindings, new_args) -> + returnMM (foldr (mk_stg_let currentCCS) (almost_expr new_args) let_bindings) where --------------- - do_arg bindings atom@(StgLitAtom _) = returnMM (bindings, atom) - do_arg bindings atom@(StgVarAtom old_var) - = let - var_type = getIdUniType old_var + do_arg ids bindings arg@(StgVarArg old_var) + | (not (isLocalVar old_var) || elemVarSet old_var ids) + && isFunTy (dropForAlls var_type) + = -- make a trivial let-binding for the top-level function + getUniqueMM `thenMM` \ uniq -> + let + new_var = mkSysLocal FSLIT("sf") uniq var_type in - if toplevelishId old_var && isFunType (getTauType var_type) - then - -- make a trivial let-binding for the top-level function - getUniqueMM `thenMM` \ uniq -> - let - new_var = mkSysLocal SLIT("ho") uniq var_type noSrcLoc - in - returnMM ( (new_var, old_var) : bindings, StgVarAtom new_var ) - else - returnMM (bindings, atom) + returnMM ( (new_var, old_var) : bindings, StgVarArg new_var ) + where + var_type = idType old_var + + do_arg ids bindings arg = returnMM (bindings, arg) --------------- - mk_stg_let :: CostCentre -> (Id, Id) -> StgExpr -> StgExpr + mk_stg_let :: CostCentreStack -> (Id, Id) -> StgExpr -> StgExpr mk_stg_let cc (new_var, old_var) body = let - rhs_body = StgApp (StgVarAtom old_var) [{-args-}] bOGUS_LVs - rhs_closure = StgRhsClosure cc stgArgOcc [{-fvs-}] ReEntrant [{-args-}] rhs_body + rhs_body = StgApp old_var [{-args-}] + rhs_closure = StgRhsClosure cc stgArgOcc [{-fvs-}] ReEntrant NoSRT{-eeek!!!-} [{-args-}] rhs_body in StgLet (StgNonRec new_var rhs_closure) body where bOGUS_LVs = emptyUniqSet -- easier to print than: panic "mk_stg_let: LVs" --} +#endif \end{code} %************************************************************************ @@ -329,44 +307,44 @@ boxHigherOrderArgs almost_expr args live_vars \begin{code} type MassageM result - = FAST_STRING -- module name - -> CostCentre -- prevailing CostCentre + = Module -- module name + -> CostCentreStack -- prevailing CostCentre -- if none, subsumedCosts at top-level - -- useCurrentCostCentre at nested levels + -- currentCostCentre at nested levels -> UniqSupply + -> VarSet -- toplevel-ish Ids for boxing -> CollectedCCs -> (CollectedCCs, result) --- the initUs function also returns the final UniqueSupply and CollectedCCs +-- the initMM function also returns the final CollectedCCs -initMM :: FAST_STRING -- module name, which we may consult +initMM :: Module -- module name, which we may consult -> UniqSupply -> MassageM a -> (CollectedCCs, a) -initMM mod_name init_us m = m mod_name noCostCentre init_us ([],[]) +initMM mod_name init_us m = m mod_name noCCS init_us emptyVarSet ([],[],[]) thenMM :: MassageM a -> (a -> MassageM b) -> MassageM b thenMM_ :: MassageM a -> (MassageM b) -> MassageM b -thenMM expr cont mod scope_cc us ccs +thenMM expr cont mod scope_cc us ids ccs = case splitUniqSupply us of { (s1, s2) -> - case (expr mod scope_cc s1 ccs) of { (ccs2, result) -> - cont result mod scope_cc s2 ccs2 }} + case (expr mod scope_cc s1 ids ccs) of { (ccs2, result) -> + cont result mod scope_cc s2 ids ccs2 }} -thenMM_ expr cont mod scope_cc us ccs +thenMM_ expr cont mod scope_cc us ids ccs = case splitUniqSupply us of { (s1, s2) -> - case (expr mod scope_cc s1 ccs) of { (ccs2, _) -> - cont mod scope_cc s2 ccs2 }} + case (expr mod scope_cc s1 ids ccs) of { (ccs2, _) -> + cont mod scope_cc s2 ids ccs2 }} returnMM :: a -> MassageM a -returnMM result mod scope_cc us ccs = (ccs, result) +returnMM result mod scope_cc us ids ccs = (ccs, result) nopMM :: MassageM () -nopMM mod scope_cc us ccs = (ccs, ()) +nopMM mod scope_cc us ids ccs = (ccs, ()) mapMM :: (a -> MassageM b) -> [a] -> MassageM [b] - mapMM f [] = returnMM [] mapMM f (m:ms) = f m `thenMM` \ r -> @@ -374,7 +352,6 @@ mapMM f (m:ms) returnMM (r:rs) mapAccumMM :: (acc -> x -> MassageM (acc, y)) -> acc -> [x] -> MassageM (acc, [y]) - mapAccumMM f b [] = returnMM (b, []) mapAccumMM f b (m:ms) = f b m `thenMM` \ (b2, r) -> @@ -382,52 +359,53 @@ mapAccumMM f b (m:ms) returnMM (b3, r:rs) getUniqueMM :: MassageM Unique -getUniqueMM mod scope_cc us ccs = (ccs, getUnique us) +getUniqueMM mod scope_cc us ids ccs = (ccs, uniqFromSupply us) + +addTopLevelIshId :: Id -> MassageM a -> MassageM a +addTopLevelIshId id scope mod scope_cc us ids ccs + | isCurrentCCS scope_cc = scope mod scope_cc us ids ccs + | otherwise = scope mod scope_cc us (extendVarSet ids id) ccs + +addTopLevelIshIds :: [Id] -> MassageM a -> MassageM a +addTopLevelIshIds [] cont = cont +addTopLevelIshIds (id:ids) cont + = addTopLevelIshId id (addTopLevelIshIds ids cont) + +getTopLevelIshIds :: MassageM VarSet +getTopLevelIshIds mod scope_cc us ids ccs = (ccs, ids) \end{code} -\begin{code} -set_prevailing_cc :: CostCentre -> MassageM a -> MassageM a -set_prevailing_cc cc_to_set_to action mod scope_cc us ccs - -- set unconditionally - = action mod cc_to_set_to us ccs - -set_prevailing_cc_maybe :: CostCentre -> (CostCentre -> MassageM a) -> MassageM a -set_prevailing_cc_maybe cc_to_try action mod scope_cc us ccs - -- set only if a real cost centre - = let - cc_to_use - = if noCostCentreAttached cc_to_try || currentOrSubsumedCosts cc_to_try - then scope_cc -- carry on as before - else cc_to_try -- use new cost centre - in - action cc_to_use mod cc_to_use us ccs +The prevailing CCS is used to tell whether we're in a top-levelish +position, where top-levelish is defined as "not inside a lambda". +Prevailing CCs used to be used for something much more complicated, +I'm sure --SDM +\begin{code} set_lambda_cc :: MassageM a -> MassageM a -set_lambda_cc action mod scope_cc us ccs - -- used when moving inside a lambda; - -- if we were chugging along as "caf/dict" we change to "ccc" - = let - cc_to_use - = if isCafCC scope_cc || isDictCC scope_cc - then useCurrentCostCentre - else scope_cc - in - action mod cc_to_use us ccs +set_lambda_cc action mod scope_cc us ids ccs + = action mod currentCCS us ids ccs +set_prevailing_cc :: CostCentreStack -> MassageM a -> MassageM a +set_prevailing_cc cc_to_set_to action mod scope_cc us ids ccs + = action mod cc_to_set_to us ids ccs -get_prevailing_cc :: MassageM CostCentre -get_prevailing_cc mod scope_cc us ccs = (ccs, scope_cc) - +get_prevailing_cc :: MassageM CostCentreStack +get_prevailing_cc mod scope_cc us ids ccs = (ccs, scope_cc) \end{code} \begin{code} collectCC :: CostCentre -> MassageM () -collectCC cc mod_name scope_cc us (local_ccs, extern_ccs) - = ASSERT(not (noCostCentreAttached cc)) - ASSERT(not (currentOrSubsumedCosts cc)) +collectCC cc mod_name scope_cc us ids (local_ccs, extern_ccs, ccss) + = ASSERT(not (noCCAttached cc)) if (cc `ccFromThisModule` mod_name) then - ((cc : local_ccs, extern_ccs), ()) + ((cc : local_ccs, extern_ccs, ccss), ()) else -- must declare it "extern" - ((local_ccs, cc : extern_ccs), ()) + ((local_ccs, cc : extern_ccs, ccss), ()) + +collectCCS :: CostCentreStack -> MassageM () + +collectCCS ccs mod_name scope_cc us ids (local_ccs, extern_ccs, ccss) + = ASSERT(not (noCCSAttached ccs)) + ((local_ccs, extern_ccs, ccs : ccss), ()) \end{code}