X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Fprelude%2FPrimOp.lhs;h=82a60e0911e0e09cc86576983610ec9e0cc4c625;hb=caac75c6a454396dadff0323162ed14adb4893cd;hp=f96617d87c713793393bff29914985257bb113a7;hpb=cd6d99c5f243b8c129fb2494bbd13b4bf98abda6;p=ghc-hetmet.git diff --git a/ghc/compiler/prelude/PrimOp.lhs b/ghc/compiler/prelude/PrimOp.lhs index f96617d..82a60e0 100644 --- a/ghc/compiler/prelude/PrimOp.lhs +++ b/ghc/compiler/prelude/PrimOp.lhs @@ -6,7 +6,7 @@ \begin{code} module PrimOp ( PrimOp(..), allThePrimOps, - primOpType, primOpSig, primOpUsg, primOpArity, + primOpType, primOpSig, primOpArity, mkPrimOpIdName, primOpRdrName, primOpTag, primOpOcc, commutableOp, @@ -17,8 +17,7 @@ module PrimOp ( getPrimOpResultInfo, PrimOpResultInfo(..), - CCall(..), CCallTarget(..), ccallMayGC, ccallIsCasm, pprCCallOp, - isDynamicTarget, dynamicTarget, setCCallUnique + eqCharName, eqIntName, eqFloatName, eqDoubleName, neqIntName, ) where #include "HsVersions.h" @@ -27,24 +26,18 @@ import PrimRep -- most of it import TysPrim import TysWiredIn -import Demand ( wwLazy, wwPrim, wwStrict, StrictnessInfo(..) ) +import NewDemand import Var ( TyVar ) -import CallConv ( CallConv, pprCallConv ) import Name ( Name, mkWiredInName ) import RdrName ( RdrName, mkRdrOrig ) import OccName ( OccName, pprOccName, mkVarOcc ) -import TyCon ( TyCon, tyConArity ) -import Type ( Type, mkForAllTys, mkFunTy, mkFunTys, mkTyVarTys, - mkTyConApp, typePrimRep, - splitFunTy_maybe, splitAlgTyConApp_maybe, splitTyConApp, - mkUTy, usOnce, usMany - ) -import Unique ( Unique, mkPrimOpIdUnique ) +import TyCon ( TyCon, isPrimTyCon, tyConPrimRep ) +import Type ( Type, mkForAllTys, mkFunTy, mkFunTys, typePrimRep, tyConAppTyCon ) +import PprType () -- get at Outputable Type instance. +import Unique ( mkPrimOpIdUnique ) import BasicTypes ( Arity, Boxity(..) ) -import CStrings ( CLabelString, pprCLabelString ) -import PrelNames ( pREL_GHC, pREL_GHC_Name ) +import PrelNames ( gHC_PRIM, gHC_PRIM_Name ) import Outputable -import Util ( zipWithEqual ) import FastTypes \end{code} @@ -61,7 +54,6 @@ These are in \tr{state-interface.verb} order. -- supplies: -- data PrimOp = ... #include "primop-data-decl.hs-incl" - | CCallOp CCall -- and don't forget to add CCall \end{code} Used for the Ord instance @@ -100,7 +92,6 @@ An @Enum@-derived list would be better; meanwhile... (ToDo) allThePrimOps :: [PrimOp] allThePrimOps = #include "primop-list.hs-incl" --- Doesn't include CCall, which is really a family of primops \end{code} %************************************************************************ @@ -147,7 +138,7 @@ mkGenPrimOp str tvs tys ty = GenPrimOp (mkVarOcc str) tvs tys ty Not all primops are strict! \begin{code} -primOpStrictness :: PrimOp -> Arity -> StrictnessInfo +primOpStrictness :: PrimOp -> Arity -> StrictSig -- See Demand.StrictnessInfo for discussion of what the results -- The arity should be the arity of the primop; that's why -- this function isn't exported. @@ -270,40 +261,6 @@ Invariants: stable name. -[Alastair Reid is to blame for this!] - -These days, (Glasgow) Haskell seems to have a bit of everything from -other languages: strict operations, mutable variables, sequencing, -pointers, etc. About the only thing left is LISP's ability to test -for pointer equality. So, let's add it in! - -\begin{verbatim} -reallyUnsafePtrEquality :: a -> a -> Int# -\end{verbatim} - -which tests any two closures (of the same type) to see if they're the -same. (Returns $0$ for @False@, $\neq 0$ for @True@ - to avoid -difficulties of trying to box up the result.) - -NB This is {\em really unsafe\/} because even something as trivial as -a garbage collection might change the answer by removing indirections. -Still, no-one's forcing you to use it. If you're worried about little -things like loss of referential transparency, you might like to wrap -it all up in a monad-like thing as John O'Donnell and John Hughes did -for non-determinism (1989 (Fraserburgh) Glasgow FP Workshop -Proceedings?) - -I'm thinking of using it to speed up a critical equality test in some -graphics stuff in a context where the possibility of saying that -denotationally equal things aren't isn't a problem (as long as it -doesn't happen too often.) ADR - -To Will: Jim said this was already in, but I can't see it so I'm -adding it. Up to you whether you add it. (Note that this could have -been readily implemented using a @veryDangerousCCall@ before they were -removed...) - - -- HWL: The first 4 Int# in all par... annotations denote: -- name, granularity info, size of result, degree of parallelism -- Same structure as _seq_ i.e. returns Int# @@ -333,8 +290,9 @@ primOpInfo op = pprPanic "primOpInfo:" (ppr op) Some PrimOps need to be called out-of-line because they either need to perform a heap check or they block. + \begin{code} -primOpOutOfLine (CCallOp c_call) = ccallMayGC c_call +primOpOutOfLine :: PrimOp -> Bool #include "primop-out-of-line.hs-incl" \end{code} @@ -404,7 +362,6 @@ duplicated. \begin{code} primOpHasSideEffects :: PrimOp -> Bool -primOpHasSideEffects (CCallOp _) = True #include "primop-has-side-effects.hs-incl" \end{code} @@ -413,7 +370,6 @@ any live variables that are stored in caller-saves registers. \begin{code} primOpNeedsWrapper :: PrimOp -> Bool -primOpNeedsWrapper (CCallOp _) = True #include "primop-needs-wrapper.hs-incl" \end{code} @@ -441,10 +397,10 @@ mkPrimOpIdName :: PrimOp -> Name -- We have to pass in the Id itself because it's a WiredInId -- and hence recursive mkPrimOpIdName op - = mkWiredInName pREL_GHC (primOpOcc op) (mkPrimOpIdUnique (primOpTag op)) + = mkWiredInName gHC_PRIM (primOpOcc op) (mkPrimOpIdUnique (primOpTag op)) primOpRdrName :: PrimOp -> RdrName -primOpRdrName op = mkRdrOrig pREL_GHC_Name (primOpOcc op) +primOpRdrName op = mkRdrOrig gHC_PRIM_Name (primOpOcc op) primOpOcc :: PrimOp -> OccName primOpOcc op = case (primOpInfo op) of @@ -457,7 +413,7 @@ primOpOcc op = case (primOpInfo op) of -- (type variables, argument types, result type) -- It also gives arity, strictness info -primOpSig :: PrimOp -> ([TyVar], [Type], Type, Arity, StrictnessInfo) +primOpSig :: PrimOp -> ([TyVar], [Type], Type, Arity, StrictSig) primOpSig op = (tyvars, arg_tys, res_ty, arity, primOpStrictness op arity) where @@ -469,50 +425,6 @@ primOpSig op Compare occ ty -> ([], [ty,ty], boolTy) GenPrimOp occ tyvars arg_tys res_ty -> (tyvars, arg_tys, res_ty) - --- primOpUsg is like primOpSig but the types it yields are the --- appropriate sigma (i.e., usage-annotated) types, --- as required by the UsageSP inference. - -primOpUsg :: PrimOp -> ([TyVar],[Type],Type) -primOpUsg p@(CCallOp _) = mangle p [] mkM -#include "primop-usage.hs-incl" - --- Things with no Haskell pointers inside: in actuality, usages are --- irrelevant here (hence it doesn't matter that some of these --- apparently permit duplication; since such arguments are never --- ENTERed anyway, the usage annotation they get is entirely irrelevant --- except insofar as it propagates to infect other values that *are* --- pointed. - - --- Helper bits & pieces for usage info. - -mkZ = mkUTy usOnce -- pointed argument used zero -mkO = mkUTy usOnce -- pointed argument used once -mkM = mkUTy usMany -- pointed argument used multiply -mkP = mkUTy usOnce -- unpointed argument -mkR = mkUTy usMany -- unpointed result - -nomangle op - = case primOpSig op of - (tyvars, arg_tys, res_ty, _, _) - -> (tyvars, map mkP arg_tys, mkR res_ty) - -mangle op fs g - = case primOpSig op of - (tyvars, arg_tys, res_ty, _, _) - -> (tyvars, zipWithEqual "primOpUsg" ($) fs arg_tys, g res_ty) - -inFun op f g ty - = case splitFunTy_maybe ty of - Just (a,b) -> mkFunTy (f a) (g b) - Nothing -> pprPanic "primOpUsg:inFun" (ppr op <+> ppr ty) - -inUB op fs ty - = case splitTyConApp ty of - (tc,tys) -> ASSERT( tc == tupleTyCon Unboxed (length fs) ) - mkTupleTy Unboxed (length fs) (zipWithEqual "primOpUsg" ($) fs tys) \end{code} \begin{code} @@ -525,21 +437,18 @@ data PrimOpResultInfo -- be out of line, or the code generator won't work. getPrimOpResultInfo :: PrimOp -> PrimOpResultInfo -getPrimOpResultInfo (CCallOp _) - = ReturnsAlg unboxedPairTyCon getPrimOpResultInfo op = case (primOpInfo op) of - Dyadic _ ty -> ReturnsPrim (typePrimRep ty) - Monadic _ ty -> ReturnsPrim (typePrimRep ty) - Compare _ ty -> ReturnsAlg boolTyCon - GenPrimOp _ _ _ ty -> - let rep = typePrimRep ty in - case rep of - PtrRep -> case splitAlgTyConApp_maybe ty of - Nothing -> pprPanic "getPrimOpResultInfo" - (ppr ty <+> ppr op) - Just (tc,_,_) -> ReturnsAlg tc - other -> ReturnsPrim other + Dyadic _ ty -> ReturnsPrim (typePrimRep ty) + Monadic _ ty -> ReturnsPrim (typePrimRep ty) + Compare _ ty -> ReturnsAlg boolTyCon + GenPrimOp _ _ _ ty | isPrimTyCon tc -> ReturnsPrim (tyConPrimRep tc) + | otherwise -> ReturnsAlg tc + where + tc = tyConAppTyCon ty + -- All primops return a tycon-app result + -- The tycon can be an unboxed tuple, though, which + -- gives rise to a ReturnAlg \end{code} The commutable ops are those for which we will try to move constants @@ -552,15 +461,6 @@ commutableOp :: PrimOp -> Bool Utils: \begin{code} -mkPrimTyApp :: [TyVar] -> PrimRep -> ([TyVar], Type) - -- CharRep --> ([], Char#) - -- StablePtrRep --> ([a], StablePtr# a) -mkPrimTyApp tvs kind - = (forall_tvs, mkTyConApp tycon (mkTyVarTys forall_tvs)) - where - tycon = primRepTyCon kind - forall_tvs = take (tyConArity tycon) tvs - dyadic_fun_ty ty = mkFunTys [ty, ty] ty monadic_fun_ty ty = mkFunTy ty ty compare_fun_ty ty = mkFunTys [ty, ty] boolTy @@ -569,93 +469,15 @@ compare_fun_ty ty = mkFunTys [ty, ty] boolTy Output stuff: \begin{code} pprPrimOp :: PrimOp -> SDoc - -pprPrimOp (CCallOp c_call) = pprCCallOp c_call -pprPrimOp other_op - = getPprStyle $ \ sty -> - if ifaceStyle sty then -- For interfaces Print it qualified with PrelGHC. - ptext SLIT("PrelGHC.") <> pprOccName occ - else - pprOccName occ - where - occ = primOpOcc other_op +pprPrimOp other_op = pprOccName (primOpOcc other_op) \end{code} - -%************************************************************************ -%* * -\subsubsection{CCalls} -%* * -%************************************************************************ - -A special ``trap-door'' to use in making calls direct to C functions: -\begin{code} -data CCall - = CCall CCallTarget - Bool -- True <=> really a "casm" - Bool -- True <=> might invoke Haskell GC - CallConv -- calling convention to use. - deriving( Eq ) - -data CCallTarget - = StaticTarget CLabelString -- An "unboxed" ccall# to `fn'. - | DynamicTarget Unique -- First argument (an Addr#) is the function pointer - -- (unique is used to generate a 'typedef' to cast - -- the function pointer if compiling the ccall# down to - -- .hc code - can't do this inline for tedious reasons.) - -instance Eq CCallTarget where - (StaticTarget l1) == (StaticTarget l2) = l1 == l2 - (DynamicTarget _) == (DynamicTarget _) = True - -- Ignore the arbitrary unique; this is important when comparing - -- a dynamic ccall read from an interface file A.hi with the - -- one constructed from A.hs, when deciding whether the interface - -- has changed - t1 == t2 = False - -ccallMayGC :: CCall -> Bool -ccallMayGC (CCall _ _ may_gc _) = may_gc - -ccallIsCasm :: CCall -> Bool -ccallIsCasm (CCall _ c_asm _ _) = c_asm - -isDynamicTarget (DynamicTarget _) = True -isDynamicTarget (StaticTarget _) = False - -dynamicTarget :: CCallTarget -dynamicTarget = DynamicTarget (panic "Unique in DynamicTarget not yet set") - -- The unique is really only to do with code generation, so it - -- is only set in CoreToStg; before then it's just an error message - -setCCallUnique :: CCall -> Unique -> CCall -setCCallUnique (CCall (DynamicTarget _) is_asm may_gc cconv) uniq - = CCall (DynamicTarget uniq) is_asm may_gc cconv -setCCallUnique ccall uniq = ccall -\end{code} +Names for some primops (for ndpFlatten/FlattenMonad.lhs) \begin{code} -pprCCallOp (CCall fun is_casm may_gc cconv) - = hcat [ ifPprDebug callconv - , text "__", ppr_dyn - , text before , ppr_fun , after] - where - callconv = text "{-" <> pprCallConv cconv <> text "-}" - - before - | is_casm && may_gc = "casm_GC ``" - | is_casm = "casm ``" - | may_gc = "ccall_GC " - | otherwise = "ccall " - - after - | is_casm = text "''" - | otherwise = empty - - ppr_dyn = case fun of - DynamicTarget _ -> text "dyn_" - _ -> empty - - ppr_fun = case fun of - DynamicTarget _ -> text "\"\"" - StaticTarget fn -> pprCLabelString fn +eqCharName = mkPrimOpIdName CharEqOp +eqIntName = mkPrimOpIdName IntEqOp +eqFloatName = mkPrimOpIdName FloatEqOp +eqDoubleName = mkPrimOpIdName DoubleEqOp +neqIntName = mkPrimOpIdName IntNeOp \end{code}