diff options
author | Adam Sandberg Eriksson <adam@sandbergericsson.se> | 2019-11-01 22:54:04 +0000 |
---|---|---|
committer | Marge Bot <ben+marge-bot@smart-cactus.org> | 2020-02-08 10:15:38 -0500 |
commit | 588acb99bc3cb377ceb76447dd60656b4a11de5a (patch) | |
tree | 598a83c392cca6ed977046685469038d661f831c /compiler | |
parent | 7c1228511f2dd4d262c04edb8539174a7de810b2 (diff) | |
download | haskell-588acb99bc3cb377ceb76447dd60656b4a11de5a.tar.gz |
slightly better named cost-centres for simple pattern bindings #17006
```
main = do
print $ g [1..100] a
where g xs x = map (`mod` x) xs
a :: Int = 324
```
The above program previously attributed the cost of computing 324 to a cost
centre named `(...)`, with this change the cost is attributed to `a` instead.
This change only affects simple pattern bindings (decorated variables: type
signatures, parens, ~ annotations and ! annotations).
Diffstat (limited to 'compiler')
-rw-r--r-- | compiler/GHC/Hs/Pat.hs | 19 | ||||
-rw-r--r-- | compiler/deSugar/Coverage.hs | 30 |
2 files changed, 40 insertions, 9 deletions
diff --git a/compiler/GHC/Hs/Pat.hs b/compiler/GHC/Hs/Pat.hs index 945c2c195f..9812fe2c44 100644 --- a/compiler/GHC/Hs/Pat.hs +++ b/compiler/GHC/Hs/Pat.hs @@ -31,6 +31,7 @@ module GHC.Hs.Pat ( mkPrefixConPat, mkCharLitPat, mkNilPat, + isSimplePat, looksLazyPatBind, isBangedLPat, patNeedsParens, parenthesizePat, @@ -274,6 +275,7 @@ data Pat p | XPat (XXPat p) + -- --------------------------------------------------------------------- data ListPatTc @@ -730,6 +732,23 @@ isIrrefutableHsPat go (XPat {}) = False +-- | Is the pattern any of combination of: +-- +-- - (pat) +-- - pat :: Type +-- - ~pat +-- - !pat +-- - x (variable) +isSimplePat :: LPat (GhcPass x) -> Maybe (IdP (GhcPass x)) +isSimplePat p = case unLoc p of + ParPat _ x -> isSimplePat x + SigPat _ x _ -> isSimplePat x + LazyPat _ x -> isSimplePat x + BangPat _ x -> isSimplePat x + VarPat _ x -> Just (unLoc x) + _ -> Nothing + + {- Note [Unboxed sum patterns aren't irrefutable] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Unlike unboxed tuples, unboxed sums are *not* irrefutable when used as diff --git a/compiler/deSugar/Coverage.hs b/compiler/deSugar/Coverage.hs index 178372a567..6f9fe36141 100644 --- a/compiler/deSugar/Coverage.hs +++ b/compiler/deSugar/Coverage.hs @@ -336,7 +336,12 @@ addTickLHsBind (L pos (funBind@(FunBind { fun_id = L _ id }))) = do -- TODO: Revisit this addTickLHsBind (L pos (pat@(PatBind { pat_lhs = lhs , pat_rhs = rhs }))) = do - let name = "(...)" + + let simplePatId = isSimplePat lhs + + -- TODO: better name for rhs's for non-simple patterns? + let name = maybe "(...)" getOccString simplePatId + (fvs, rhs') <- getFreeVars $ addPathEntry name $ addTickGRHSs False False rhs let pat' = pat { pat_rhs = rhs'} @@ -348,16 +353,24 @@ addTickLHsBind (L pos (pat@(PatBind { pat_lhs = lhs then return (L pos pat') else do + let mbCons = maybe id (:) + + let (initial_rhs_ticks, initial_patvar_tickss) = pat_ticks pat' + -- Allocate the ticks + rhs_tick <- bindTick density name pos fvs - let patvars = map getOccString (collectPatBinders lhs) - patvar_ticks <- mapM (\v -> bindTick density v pos fvs) patvars + let rhs_ticks = rhs_tick `mbCons` initial_rhs_ticks + + patvar_tickss <- case simplePatId of + Just{} -> return initial_patvar_tickss + Nothing -> do + let patvars = map getOccString (collectPatBinders lhs) + patvar_ticks <- mapM (\v -> bindTick density v pos fvs) patvars + return + (zipWith mbCons patvar_ticks + (initial_patvar_tickss ++ repeat [])) - -- Add to pattern - let mbCons = maybe id (:) - rhs_ticks = rhs_tick `mbCons` fst (pat_ticks pat') - patvar_tickss = zipWith mbCons patvar_ticks - (snd (pat_ticks pat') ++ repeat []) return $ L pos $ pat' { pat_ticks = (rhs_ticks, patvar_tickss) } -- Only internal stuff, not from source, uses VarBind, so we ignore it. @@ -365,7 +378,6 @@ addTickLHsBind var_bind@(L _ (VarBind {})) = return var_bind addTickLHsBind patsyn_bind@(L _ (PatSynBind {})) = return patsyn_bind addTickLHsBind bind@(L _ (XHsBindsLR {})) = return bind - bindTick :: TickDensity -> String -> SrcSpan -> FreeVars -> TM (Maybe (Tickish Id)) bindTick density name pos fvs = do |