X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=compiler%2Fstranal%2FWorkWrap.lhs;h=71f9ef831756df690335096bcec0ad9a648b3272;hb=bd78c94a3b41f8d2097efc0415fa26e0cd1140ef;hp=5143eea08e4b5192801f65a7fc16e1970af62e26;hpb=a26e1e3310f4c92196fe6b4d407f72f3c6824132;p=ghc-hetmet.git

diff --git a/compiler/stranal/WorkWrap.lhs b/compiler/stranal/WorkWrap.lhs
index 5143eea..71f9ef8 100644
--- a/compiler/stranal/WorkWrap.lhs
+++ b/compiler/stranal/WorkWrap.lhs
@@ -10,7 +10,8 @@ module WorkWrap ( wwTopBinds, mkWrapper ) where
 
 import CoreSyn
 import CoreUnfold	( certainlyWillInline )
-import CoreUtils	( exprType, exprIsHNF, exprArity )
+import CoreUtils	( exprType, exprIsHNF )
+import CoreArity	( exprArity )
 import Var
 import Id		( Id, idType, isOneShotLambda, 
 			  setIdNewStrictness, mkWorkerId,
@@ -283,7 +284,7 @@ splitFun fn_id fn_info wrap_dmds res_info inline_prag rhs
 -- which is very annoying.
 get_one_shots :: Expr Var -> [Bool]
 get_one_shots (Lam b e)
-  | isIdVar b = isOneShotLambda b : get_one_shots e
+  | isId b    = isOneShotLambda b : get_one_shots e
   | otherwise = get_one_shots e
 get_one_shots (Note _ e) = get_one_shots e
 get_one_shots _    	 = noOneShotInfo