summaryrefslogtreecommitdiff
path: root/compiler/GHC/Core/Subst.hs
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/GHC/Core/Subst.hs')
-rw-r--r--compiler/GHC/Core/Subst.hs18
1 files changed, 11 insertions, 7 deletions
diff --git a/compiler/GHC/Core/Subst.hs b/compiler/GHC/Core/Subst.hs
index ddb5b61f7b..f2b25e17e5 100644
--- a/compiler/GHC/Core/Subst.hs
+++ b/compiler/GHC/Core/Subst.hs
@@ -476,11 +476,12 @@ substIdBndr _doc rec_subst subst@(Subst in_scope env tvs cvs) old_id
where
id1 = uniqAway in_scope old_id -- id1 is cloned if necessary
id2 | no_type_change = id1
- | otherwise = setIdType id1 (substTy subst old_ty)
+ | otherwise = updateIdTypeAndMult (substTy subst) id1
old_ty = idType old_id
+ old_w = idMult old_id
no_type_change = (isEmptyVarEnv tvs && isEmptyVarEnv cvs) ||
- noFreeVarsOfType old_ty
+ (noFreeVarsOfType old_ty && noFreeVarsOfType old_w)
-- new_id has the right IdInfo
-- The lazy-set is because we're in a loop here, with
@@ -600,13 +601,16 @@ substCo subst co = Coercion.substCo (getTCvSubst subst) co
substIdType :: Subst -> Id -> Id
substIdType subst@(Subst _ _ tv_env cv_env) id
- | (isEmptyVarEnv tv_env && isEmptyVarEnv cv_env) || noFreeVarsOfType old_ty = id
- | otherwise = setIdType id (substTy subst old_ty)
- -- The tyCoVarsOfType is cheaper than it looks
- -- because we cache the free tyvars of the type
- -- in a Note in the id's type itself
+ | (isEmptyVarEnv tv_env && isEmptyVarEnv cv_env)
+ || (noFreeVarsOfType old_ty && noFreeVarsOfType old_w) = id
+ | otherwise =
+ updateIdTypeAndMult (substTy subst) id
+ -- The tyCoVarsOfType is cheaper than it looks
+ -- because we cache the free tyvars of the type
+ -- in a Note in the id's type itself
where
old_ty = idType id
+ old_w = varMult id
------------------
-- | Substitute into some 'IdInfo' with regard to the supplied new 'Id'.