1 -----------------------------------------------------------------------------
3 -- Code generation for PrimOps.
5 -- (c) The University of Glasgow 2004
7 -----------------------------------------------------------------------------
13 #include "HsVersions.h"
15 import ForeignCall ( CCallConv(CCallConv) )
16 import StgSyn ( StgLiveVars, StgArg )
17 import CgBindery ( getVolatileRegs, getArgAmodes )
19 import CgInfoTbls ( getConstrTag )
20 import CgUtils ( cmmOffsetW, cmmOffsetB, cmmLoadIndexW )
23 import CLabel ( mkMAP_FROZEN_infoLabel, mkMAP_DIRTY_infoLabel,
24 mkDirty_MUT_VAR_Label, mkRtsCodeLabel )
28 import PrimOp ( PrimOp(..) )
29 import SMRep ( tablesNextToCode )
30 import Constants ( wORD_SIZE, wORD_SIZE_IN_BITS )
31 import StaticFlags ( opt_Parallel )
34 -- ---------------------------------------------------------------------------
35 -- Code generation for PrimOps
37 cgPrimOp :: [CmmReg] -- where to put the results
39 -> [StgArg] -- arguments
40 -> StgLiveVars -- live vars, in case we need to save them
43 cgPrimOp results op args live
44 = do arg_exprs <- getArgAmodes args
45 let non_void_args = [ e | (r,e) <- arg_exprs, nonVoidArg r ]
46 emitPrimOp results op non_void_args live
49 emitPrimOp :: [CmmReg] -- where to put the results
51 -> [CmmExpr] -- arguments
52 -> StgLiveVars -- live vars, in case we need to save them
55 -- First we handle various awkward cases specially. The remaining
56 -- easy cases are then handled by translateOp, defined below.
58 emitPrimOp [res_r,res_c] IntAddCOp [aa,bb] live
60 With some bit-twiddling, we can define int{Add,Sub}Czh portably in
61 C, and without needing any comparisons. This may not be the
62 fastest way to do it - if you have better code, please send it! --SDM
64 Return : r = a + b, c = 0 if no overflow, 1 on overflow.
66 We currently don't make use of the r value if c is != 0 (i.e.
67 overflow), we just convert to big integers and try again. This
68 could be improved by making r and c the correct values for
69 plugging into a new J#.
71 { r = ((I_)(a)) + ((I_)(b)); \
72 c = ((StgWord)(~(((I_)(a))^((I_)(b))) & (((I_)(a))^r))) \
73 >> (BITS_IN (I_) - 1); \
75 Wading through the mass of bracketry, it seems to reduce to:
76 c = ( (~(a^b)) & (a^r) ) >>unsigned (BITS_IN(I_)-1)
80 CmmAssign res_r (CmmMachOp mo_wordAdd [aa,bb]),
82 CmmMachOp mo_wordUShr [
83 CmmMachOp mo_wordAnd [
84 CmmMachOp mo_wordNot [CmmMachOp mo_wordXor [aa,bb]],
85 CmmMachOp mo_wordXor [aa, CmmReg res_r]
87 CmmLit (mkIntCLit (wORD_SIZE_IN_BITS - 1))
92 emitPrimOp [res_r,res_c] IntSubCOp [aa,bb] live
94 #define subIntCzh(r,c,a,b) \
95 { r = ((I_)(a)) - ((I_)(b)); \
96 c = ((StgWord)((((I_)(a))^((I_)(b))) & (((I_)(a))^r))) \
97 >> (BITS_IN (I_) - 1); \
100 c = ((a^b) & (a^r)) >>unsigned (BITS_IN(I_)-1)
103 CmmAssign res_r (CmmMachOp mo_wordSub [aa,bb]),
105 CmmMachOp mo_wordUShr [
106 CmmMachOp mo_wordAnd [
107 CmmMachOp mo_wordXor [aa,bb],
108 CmmMachOp mo_wordXor [aa, CmmReg res_r]
110 CmmLit (mkIntCLit (wORD_SIZE_IN_BITS - 1))
115 emitPrimOp [res] ParOp [arg] live
117 -- for now, just implement this in a C function
118 -- later, we might want to inline it.
119 vols <- getVolatileRegs live
120 stmtC (CmmCall (CmmForeignCall newspark CCallConv) [(res,NoHint)]
121 [(CmmReg (CmmGlobal BaseReg), PtrHint),
125 newspark = CmmLit (CmmLabel (mkRtsCodeLabel SLIT("newSpark")))
127 emitPrimOp [res] ReadMutVarOp [mutv] live
128 = stmtC (CmmAssign res (cmmLoadIndexW mutv fixedHdrSize))
130 emitPrimOp [] WriteMutVarOp [mutv,var] live
132 stmtC (CmmStore (cmmOffsetW mutv fixedHdrSize) var)
133 vols <- getVolatileRegs live
134 stmtC (CmmCall (CmmForeignCall (CmmLit (CmmLabel mkDirty_MUT_VAR_Label))
140 -- #define sizzeofByteArrayzh(r,a) \
141 -- r = (((StgArrWords *)(a))->words * sizeof(W_))
142 emitPrimOp [res] SizeofByteArrayOp [arg] live
144 CmmAssign res (CmmMachOp mo_wordMul [
145 cmmLoadIndexW arg fixedHdrSize,
146 CmmLit (mkIntCLit wORD_SIZE)
149 -- #define sizzeofMutableByteArrayzh(r,a) \
150 -- r = (((StgArrWords *)(a))->words * sizeof(W_))
151 emitPrimOp [res] SizeofMutableByteArrayOp [arg] live
152 = emitPrimOp [res] SizeofByteArrayOp [arg] live
155 -- #define touchzh(o) /* nothing */
156 emitPrimOp [] TouchOp [arg] live
159 -- #define byteArrayContentszh(r,a) r = BYTE_ARR_CTS(a)
160 emitPrimOp [res] ByteArrayContents_Char [arg] live
161 = stmtC (CmmAssign res (cmmOffsetB arg arrWordsHdrSize))
163 -- #define stableNameToIntzh(r,s) (r = ((StgStableName *)s)->sn)
164 emitPrimOp [res] StableNameToIntOp [arg] live
165 = stmtC (CmmAssign res (cmmLoadIndexW arg fixedHdrSize))
167 -- #define eqStableNamezh(r,sn1,sn2) \
168 -- (r = (((StgStableName *)sn1)->sn == ((StgStableName *)sn2)->sn))
169 emitPrimOp [res] EqStableNameOp [arg1,arg2] live
170 = stmtC (CmmAssign res (CmmMachOp mo_wordEq [
171 cmmLoadIndexW arg1 fixedHdrSize,
172 cmmLoadIndexW arg2 fixedHdrSize
176 emitPrimOp [res] ReallyUnsafePtrEqualityOp [arg1,arg2] live
177 = stmtC (CmmAssign res (CmmMachOp mo_wordEq [arg1,arg2]))
179 -- #define addrToHValuezh(r,a) r=(P_)a
180 emitPrimOp [res] AddrToHValueOp [arg] live
181 = stmtC (CmmAssign res arg)
183 -- #define dataToTagzh(r,a) r=(GET_TAG(((StgClosure *)a)->header.info))
184 emitPrimOp [res] DataToTagOp [arg] live
185 = stmtC (CmmAssign res (getConstrTag arg))
187 {- Freezing arrays-of-ptrs requires changing an info table, for the
188 benefit of the generational collector. It needs to scavenge mutable
189 objects, even if they are in old space. When they become immutable,
190 they can be removed from this scavenge list. -}
192 -- #define unsafeFreezzeArrayzh(r,a)
194 -- SET_INFO((StgClosure *)a,&stg_MUT_ARR_PTRS_FROZEN0_info);
197 emitPrimOp [res] UnsafeFreezeArrayOp [arg] live
198 = stmtsC [ setInfo arg (CmmLit (CmmLabel mkMAP_FROZEN_infoLabel)),
201 -- #define unsafeFreezzeByteArrayzh(r,a) r=(a)
202 emitPrimOp [res] UnsafeFreezeByteArrayOp [arg] live
203 = stmtC (CmmAssign res arg)
205 -- Reading/writing pointer arrays
207 emitPrimOp [r] ReadArrayOp [obj,ix] live = doReadPtrArrayOp r obj ix
208 emitPrimOp [r] IndexArrayOp [obj,ix] live = doReadPtrArrayOp r obj ix
209 emitPrimOp [] WriteArrayOp [obj,ix,v] live = doWritePtrArrayOp obj ix v
213 emitPrimOp res IndexOffAddrOp_Char args live = doIndexOffAddrOp (Just mo_u_8ToWord) I8 res args
214 emitPrimOp res IndexOffAddrOp_WideChar args live = doIndexOffAddrOp (Just mo_u_32ToWord) I32 res args
215 emitPrimOp res IndexOffAddrOp_Int args live = doIndexOffAddrOp Nothing wordRep res args
216 emitPrimOp res IndexOffAddrOp_Word args live = doIndexOffAddrOp Nothing wordRep res args
217 emitPrimOp res IndexOffAddrOp_Addr args live = doIndexOffAddrOp Nothing wordRep res args
218 emitPrimOp res IndexOffAddrOp_Float args live = doIndexOffAddrOp Nothing F32 res args
219 emitPrimOp res IndexOffAddrOp_Double args live = doIndexOffAddrOp Nothing F64 res args
220 emitPrimOp res IndexOffAddrOp_StablePtr args live = doIndexOffAddrOp Nothing wordRep res args
221 emitPrimOp res IndexOffAddrOp_Int8 args live = doIndexOffAddrOp (Just mo_s_8ToWord) I8 res args
222 emitPrimOp res IndexOffAddrOp_Int16 args live = doIndexOffAddrOp (Just mo_s_16ToWord) I16 res args
223 emitPrimOp res IndexOffAddrOp_Int32 args live = doIndexOffAddrOp (Just mo_s_32ToWord) I32 res args
224 emitPrimOp res IndexOffAddrOp_Int64 args live = doIndexOffAddrOp Nothing I64 res args
225 emitPrimOp res IndexOffAddrOp_Word8 args live = doIndexOffAddrOp (Just mo_u_8ToWord) I8 res args
226 emitPrimOp res IndexOffAddrOp_Word16 args live = doIndexOffAddrOp (Just mo_u_16ToWord) I16 res args
227 emitPrimOp res IndexOffAddrOp_Word32 args live = doIndexOffAddrOp (Just mo_u_32ToWord) I32 res args
228 emitPrimOp res IndexOffAddrOp_Word64 args live = doIndexOffAddrOp Nothing I64 res args
230 -- ReadXXXoffAddr, which are identical, for our purposes, to IndexXXXoffAddr.
232 emitPrimOp res ReadOffAddrOp_Char args live = doIndexOffAddrOp (Just mo_u_8ToWord) I8 res args
233 emitPrimOp res ReadOffAddrOp_WideChar args live = doIndexOffAddrOp (Just mo_u_32ToWord) I32 res args
234 emitPrimOp res ReadOffAddrOp_Int args live = doIndexOffAddrOp Nothing wordRep res args
235 emitPrimOp res ReadOffAddrOp_Word args live = doIndexOffAddrOp Nothing wordRep res args
236 emitPrimOp res ReadOffAddrOp_Addr args live = doIndexOffAddrOp Nothing wordRep res args
237 emitPrimOp res ReadOffAddrOp_Float args live = doIndexOffAddrOp Nothing F32 res args
238 emitPrimOp res ReadOffAddrOp_Double args live = doIndexOffAddrOp Nothing F64 res args
239 emitPrimOp res ReadOffAddrOp_StablePtr args live = doIndexOffAddrOp Nothing wordRep res args
240 emitPrimOp res ReadOffAddrOp_Int8 args live = doIndexOffAddrOp (Just mo_s_8ToWord) I8 res args
241 emitPrimOp res ReadOffAddrOp_Int16 args live = doIndexOffAddrOp (Just mo_s_16ToWord) I16 res args
242 emitPrimOp res ReadOffAddrOp_Int32 args live = doIndexOffAddrOp (Just mo_s_32ToWord) I32 res args
243 emitPrimOp res ReadOffAddrOp_Int64 args live = doIndexOffAddrOp Nothing I64 res args
244 emitPrimOp res ReadOffAddrOp_Word8 args live = doIndexOffAddrOp (Just mo_u_8ToWord) I8 res args
245 emitPrimOp res ReadOffAddrOp_Word16 args live = doIndexOffAddrOp (Just mo_u_16ToWord) I16 res args
246 emitPrimOp res ReadOffAddrOp_Word32 args live = doIndexOffAddrOp (Just mo_u_32ToWord) I32 res args
247 emitPrimOp res ReadOffAddrOp_Word64 args live = doIndexOffAddrOp Nothing I64 res args
251 emitPrimOp res IndexByteArrayOp_Char args live = doIndexByteArrayOp (Just mo_u_8ToWord) I8 res args
252 emitPrimOp res IndexByteArrayOp_WideChar args live = doIndexByteArrayOp (Just mo_u_32ToWord) I32 res args
253 emitPrimOp res IndexByteArrayOp_Int args live = doIndexByteArrayOp Nothing wordRep res args
254 emitPrimOp res IndexByteArrayOp_Word args live = doIndexByteArrayOp Nothing wordRep res args
255 emitPrimOp res IndexByteArrayOp_Addr args live = doIndexByteArrayOp Nothing wordRep res args
256 emitPrimOp res IndexByteArrayOp_Float args live = doIndexByteArrayOp Nothing F32 res args
257 emitPrimOp res IndexByteArrayOp_Double args live = doIndexByteArrayOp Nothing F64 res args
258 emitPrimOp res IndexByteArrayOp_StablePtr args live = doIndexByteArrayOp Nothing wordRep res args
259 emitPrimOp res IndexByteArrayOp_Int8 args live = doIndexByteArrayOp (Just mo_s_8ToWord) I8 res args
260 emitPrimOp res IndexByteArrayOp_Int16 args live = doIndexByteArrayOp (Just mo_s_16ToWord) I16 res args
261 emitPrimOp res IndexByteArrayOp_Int32 args live = doIndexByteArrayOp (Just mo_s_32ToWord) I32 res args
262 emitPrimOp res IndexByteArrayOp_Int64 args live = doIndexByteArrayOp Nothing I64 res args
263 emitPrimOp res IndexByteArrayOp_Word8 args live = doIndexByteArrayOp (Just mo_u_8ToWord) I8 res args
264 emitPrimOp res IndexByteArrayOp_Word16 args live = doIndexByteArrayOp (Just mo_u_16ToWord) I16 res args
265 emitPrimOp res IndexByteArrayOp_Word32 args live = doIndexByteArrayOp (Just mo_u_32ToWord) I32 res args
266 emitPrimOp res IndexByteArrayOp_Word64 args live = doIndexByteArrayOp Nothing I64 res args
268 -- ReadXXXArray, identical to IndexXXXArray.
270 emitPrimOp res ReadByteArrayOp_Char args live = doIndexByteArrayOp (Just mo_u_8ToWord) I8 res args
271 emitPrimOp res ReadByteArrayOp_WideChar args live = doIndexByteArrayOp (Just mo_u_32ToWord) I32 res args
272 emitPrimOp res ReadByteArrayOp_Int args live = doIndexByteArrayOp Nothing wordRep res args
273 emitPrimOp res ReadByteArrayOp_Word args live = doIndexByteArrayOp Nothing wordRep res args
274 emitPrimOp res ReadByteArrayOp_Addr args live = doIndexByteArrayOp Nothing wordRep res args
275 emitPrimOp res ReadByteArrayOp_Float args live = doIndexByteArrayOp Nothing F32 res args
276 emitPrimOp res ReadByteArrayOp_Double args live = doIndexByteArrayOp Nothing F64 res args
277 emitPrimOp res ReadByteArrayOp_StablePtr args live = doIndexByteArrayOp Nothing wordRep res args
278 emitPrimOp res ReadByteArrayOp_Int8 args live = doIndexByteArrayOp (Just mo_s_8ToWord) I8 res args
279 emitPrimOp res ReadByteArrayOp_Int16 args live = doIndexByteArrayOp (Just mo_s_16ToWord) I16 res args
280 emitPrimOp res ReadByteArrayOp_Int32 args live = doIndexByteArrayOp (Just mo_s_32ToWord) I32 res args
281 emitPrimOp res ReadByteArrayOp_Int64 args live = doIndexByteArrayOp Nothing I64 res args
282 emitPrimOp res ReadByteArrayOp_Word8 args live = doIndexByteArrayOp (Just mo_u_8ToWord) I8 res args
283 emitPrimOp res ReadByteArrayOp_Word16 args live = doIndexByteArrayOp (Just mo_u_16ToWord) I16 res args
284 emitPrimOp res ReadByteArrayOp_Word32 args live = doIndexByteArrayOp (Just mo_u_32ToWord) I32 res args
285 emitPrimOp res ReadByteArrayOp_Word64 args live = doIndexByteArrayOp Nothing I64 res args
289 emitPrimOp res WriteOffAddrOp_Char args live = doWriteOffAddrOp (Just mo_WordTo8) I8 res args
290 emitPrimOp res WriteOffAddrOp_WideChar args live = doWriteOffAddrOp (Just mo_WordTo32) I32 res args
291 emitPrimOp res WriteOffAddrOp_Int args live = doWriteOffAddrOp Nothing wordRep res args
292 emitPrimOp res WriteOffAddrOp_Word args live = doWriteOffAddrOp Nothing wordRep res args
293 emitPrimOp res WriteOffAddrOp_Addr args live = doWriteOffAddrOp Nothing wordRep res args
294 emitPrimOp res WriteOffAddrOp_Float args live = doWriteOffAddrOp Nothing F32 res args
295 emitPrimOp res WriteOffAddrOp_Double args live = doWriteOffAddrOp Nothing F64 res args
296 emitPrimOp res WriteOffAddrOp_StablePtr args live = doWriteOffAddrOp Nothing wordRep res args
297 emitPrimOp res WriteOffAddrOp_Int8 args live = doWriteOffAddrOp (Just mo_WordTo8) I8 res args
298 emitPrimOp res WriteOffAddrOp_Int16 args live = doWriteOffAddrOp (Just mo_WordTo16) I16 res args
299 emitPrimOp res WriteOffAddrOp_Int32 args live = doWriteOffAddrOp (Just mo_WordTo32) I32 res args
300 emitPrimOp res WriteOffAddrOp_Int64 args live = doWriteOffAddrOp Nothing I64 res args
301 emitPrimOp res WriteOffAddrOp_Word8 args live = doWriteOffAddrOp (Just mo_WordTo8) I8 res args
302 emitPrimOp res WriteOffAddrOp_Word16 args live = doWriteOffAddrOp (Just mo_WordTo16) I16 res args
303 emitPrimOp res WriteOffAddrOp_Word32 args live = doWriteOffAddrOp (Just mo_WordTo32) I32 res args
304 emitPrimOp res WriteOffAddrOp_Word64 args live = doWriteOffAddrOp Nothing I64 res args
308 emitPrimOp res WriteByteArrayOp_Char args live = doWriteByteArrayOp (Just mo_WordTo8) I8 res args
309 emitPrimOp res WriteByteArrayOp_WideChar args live = doWriteByteArrayOp (Just mo_WordTo32) I32 res args
310 emitPrimOp res WriteByteArrayOp_Int args live = doWriteByteArrayOp Nothing wordRep res args
311 emitPrimOp res WriteByteArrayOp_Word args live = doWriteByteArrayOp Nothing wordRep res args
312 emitPrimOp res WriteByteArrayOp_Addr args live = doWriteByteArrayOp Nothing wordRep res args
313 emitPrimOp res WriteByteArrayOp_Float args live = doWriteByteArrayOp Nothing F32 res args
314 emitPrimOp res WriteByteArrayOp_Double args live = doWriteByteArrayOp Nothing F64 res args
315 emitPrimOp res WriteByteArrayOp_StablePtr args live = doWriteByteArrayOp Nothing wordRep res args
316 emitPrimOp res WriteByteArrayOp_Int8 args live = doWriteByteArrayOp (Just mo_WordTo8) I8 res args
317 emitPrimOp res WriteByteArrayOp_Int16 args live = doWriteByteArrayOp (Just mo_WordTo16) I16 res args
318 emitPrimOp res WriteByteArrayOp_Int32 args live = doWriteByteArrayOp (Just mo_WordTo32) I32 res args
319 emitPrimOp res WriteByteArrayOp_Int64 args live = doWriteByteArrayOp Nothing I64 res args
320 emitPrimOp res WriteByteArrayOp_Word8 args live = doWriteByteArrayOp (Just mo_WordTo8) I8 res args
321 emitPrimOp res WriteByteArrayOp_Word16 args live = doWriteByteArrayOp (Just mo_WordTo16) I16 res args
322 emitPrimOp res WriteByteArrayOp_Word32 args live = doWriteByteArrayOp (Just mo_WordTo32) I32 res args
323 emitPrimOp res WriteByteArrayOp_Word64 args live = doWriteByteArrayOp Nothing I64 res args
326 -- The rest just translate straightforwardly
327 emitPrimOp [res] op [arg] live
329 = stmtC (CmmAssign res arg)
331 | Just (mop,rep) <- narrowOp op
332 = stmtC (CmmAssign res (CmmMachOp (mop rep wordRep) [
333 CmmMachOp (mop wordRep rep) [arg]]))
335 emitPrimOp [res] op args live
336 | Just prim <- callishOp op
337 = do vols <- getVolatileRegs live
338 stmtC (CmmCall (CmmPrim prim) [(res,NoHint)]
339 [(a,NoHint) | a<-args] (Just vols)) -- ToDo: hints?
341 | Just mop <- translateOp op
342 = let stmt = CmmAssign res (CmmMachOp mop args) in
346 = pprPanic "emitPrimOp: can't translate PrimOp" (ppr op)
349 -- These PrimOps are NOPs in Cmm
351 nopOp Int2WordOp = True
352 nopOp Word2IntOp = True
353 nopOp Int2AddrOp = True
354 nopOp Addr2IntOp = True
355 nopOp ChrOp = True -- Int# and Char# are rep'd the same
359 -- These PrimOps turn into double casts
361 narrowOp Narrow8IntOp = Just (MO_S_Conv, I8)
362 narrowOp Narrow16IntOp = Just (MO_S_Conv, I16)
363 narrowOp Narrow32IntOp = Just (MO_S_Conv, I32)
364 narrowOp Narrow8WordOp = Just (MO_U_Conv, I8)
365 narrowOp Narrow16WordOp = Just (MO_U_Conv, I16)
366 narrowOp Narrow32WordOp = Just (MO_U_Conv, I32)
369 -- Native word signless ops
371 translateOp IntAddOp = Just mo_wordAdd
372 translateOp IntSubOp = Just mo_wordSub
373 translateOp WordAddOp = Just mo_wordAdd
374 translateOp WordSubOp = Just mo_wordSub
375 translateOp AddrAddOp = Just mo_wordAdd
376 translateOp AddrSubOp = Just mo_wordSub
378 translateOp IntEqOp = Just mo_wordEq
379 translateOp IntNeOp = Just mo_wordNe
380 translateOp WordEqOp = Just mo_wordEq
381 translateOp WordNeOp = Just mo_wordNe
382 translateOp AddrEqOp = Just mo_wordEq
383 translateOp AddrNeOp = Just mo_wordNe
385 translateOp AndOp = Just mo_wordAnd
386 translateOp OrOp = Just mo_wordOr
387 translateOp XorOp = Just mo_wordXor
388 translateOp NotOp = Just mo_wordNot
389 translateOp SllOp = Just mo_wordShl
390 translateOp SrlOp = Just mo_wordUShr
392 translateOp AddrRemOp = Just mo_wordURem
394 -- Native word signed ops
396 translateOp IntMulOp = Just mo_wordMul
397 translateOp IntMulMayOfloOp = Just (MO_S_MulMayOflo wordRep)
398 translateOp IntQuotOp = Just mo_wordSQuot
399 translateOp IntRemOp = Just mo_wordSRem
400 translateOp IntNegOp = Just mo_wordSNeg
403 translateOp IntGeOp = Just mo_wordSGe
404 translateOp IntLeOp = Just mo_wordSLe
405 translateOp IntGtOp = Just mo_wordSGt
406 translateOp IntLtOp = Just mo_wordSLt
408 translateOp ISllOp = Just mo_wordShl
409 translateOp ISraOp = Just mo_wordSShr
410 translateOp ISrlOp = Just mo_wordUShr
412 -- Native word unsigned ops
414 translateOp WordGeOp = Just mo_wordUGe
415 translateOp WordLeOp = Just mo_wordULe
416 translateOp WordGtOp = Just mo_wordUGt
417 translateOp WordLtOp = Just mo_wordULt
419 translateOp WordMulOp = Just mo_wordMul
420 translateOp WordQuotOp = Just mo_wordUQuot
421 translateOp WordRemOp = Just mo_wordURem
423 translateOp AddrGeOp = Just mo_wordUGe
424 translateOp AddrLeOp = Just mo_wordULe
425 translateOp AddrGtOp = Just mo_wordUGt
426 translateOp AddrLtOp = Just mo_wordULt
430 translateOp CharEqOp = Just (MO_Eq wordRep)
431 translateOp CharNeOp = Just (MO_Ne wordRep)
432 translateOp CharGeOp = Just (MO_U_Ge wordRep)
433 translateOp CharLeOp = Just (MO_U_Le wordRep)
434 translateOp CharGtOp = Just (MO_U_Gt wordRep)
435 translateOp CharLtOp = Just (MO_U_Lt wordRep)
439 translateOp DoubleEqOp = Just (MO_Eq F64)
440 translateOp DoubleNeOp = Just (MO_Ne F64)
441 translateOp DoubleGeOp = Just (MO_S_Ge F64)
442 translateOp DoubleLeOp = Just (MO_S_Le F64)
443 translateOp DoubleGtOp = Just (MO_S_Gt F64)
444 translateOp DoubleLtOp = Just (MO_S_Lt F64)
446 translateOp DoubleAddOp = Just (MO_Add F64)
447 translateOp DoubleSubOp = Just (MO_Sub F64)
448 translateOp DoubleMulOp = Just (MO_Mul F64)
449 translateOp DoubleDivOp = Just (MO_S_Quot F64)
450 translateOp DoubleNegOp = Just (MO_S_Neg F64)
454 translateOp FloatEqOp = Just (MO_Eq F32)
455 translateOp FloatNeOp = Just (MO_Ne F32)
456 translateOp FloatGeOp = Just (MO_S_Ge F32)
457 translateOp FloatLeOp = Just (MO_S_Le F32)
458 translateOp FloatGtOp = Just (MO_S_Gt F32)
459 translateOp FloatLtOp = Just (MO_S_Lt F32)
461 translateOp FloatAddOp = Just (MO_Add F32)
462 translateOp FloatSubOp = Just (MO_Sub F32)
463 translateOp FloatMulOp = Just (MO_Mul F32)
464 translateOp FloatDivOp = Just (MO_S_Quot F32)
465 translateOp FloatNegOp = Just (MO_S_Neg F32)
469 translateOp Int2DoubleOp = Just (MO_S_Conv wordRep F64)
470 translateOp Double2IntOp = Just (MO_S_Conv F64 wordRep)
472 translateOp Int2FloatOp = Just (MO_S_Conv wordRep F32)
473 translateOp Float2IntOp = Just (MO_S_Conv F32 wordRep)
475 translateOp Float2DoubleOp = Just (MO_S_Conv F32 F64)
476 translateOp Double2FloatOp = Just (MO_S_Conv F64 F32)
478 -- Word comparisons masquerading as more exotic things.
480 translateOp SameMutVarOp = Just mo_wordEq
481 translateOp SameMVarOp = Just mo_wordEq
482 translateOp SameMutableArrayOp = Just mo_wordEq
483 translateOp SameMutableByteArrayOp = Just mo_wordEq
484 translateOp SameTVarOp = Just mo_wordEq
485 translateOp EqStablePtrOp = Just mo_wordEq
487 translateOp _ = Nothing
489 -- These primops are implemented by CallishMachOps, because they sometimes
490 -- turn into foreign calls depending on the backend.
492 callishOp DoublePowerOp = Just MO_F64_Pwr
493 callishOp DoubleSinOp = Just MO_F64_Sin
494 callishOp DoubleCosOp = Just MO_F64_Cos
495 callishOp DoubleTanOp = Just MO_F64_Tan
496 callishOp DoubleSinhOp = Just MO_F64_Sinh
497 callishOp DoubleCoshOp = Just MO_F64_Cosh
498 callishOp DoubleTanhOp = Just MO_F64_Tanh
499 callishOp DoubleAsinOp = Just MO_F64_Asin
500 callishOp DoubleAcosOp = Just MO_F64_Acos
501 callishOp DoubleAtanOp = Just MO_F64_Atan
502 callishOp DoubleLogOp = Just MO_F64_Log
503 callishOp DoubleExpOp = Just MO_F64_Exp
504 callishOp DoubleSqrtOp = Just MO_F64_Sqrt
506 callishOp FloatPowerOp = Just MO_F32_Pwr
507 callishOp FloatSinOp = Just MO_F32_Sin
508 callishOp FloatCosOp = Just MO_F32_Cos
509 callishOp FloatTanOp = Just MO_F32_Tan
510 callishOp FloatSinhOp = Just MO_F32_Sinh
511 callishOp FloatCoshOp = Just MO_F32_Cosh
512 callishOp FloatTanhOp = Just MO_F32_Tanh
513 callishOp FloatAsinOp = Just MO_F32_Asin
514 callishOp FloatAcosOp = Just MO_F32_Acos
515 callishOp FloatAtanOp = Just MO_F32_Atan
516 callishOp FloatLogOp = Just MO_F32_Log
517 callishOp FloatExpOp = Just MO_F32_Exp
518 callishOp FloatSqrtOp = Just MO_F32_Sqrt
520 callishOp _ = Nothing
522 ------------------------------------------------------------------------------
523 -- Helpers for translating various minor variants of array indexing.
525 doIndexOffAddrOp maybe_post_read_cast rep [res] [addr,idx]
526 = mkBasicIndexedRead 0 maybe_post_read_cast rep res addr idx
527 doIndexOffAddrOp _ _ _ _
528 = panic "CgPrimOp: doIndexOffAddrOp"
530 doIndexByteArrayOp maybe_post_read_cast rep [res] [addr,idx]
531 = mkBasicIndexedRead arrWordsHdrSize maybe_post_read_cast rep res addr idx
532 doIndexByteArrayOp _ _ _ _
533 = panic "CgPrimOp: doIndexByteArrayOp"
535 doReadPtrArrayOp res addr idx
536 = mkBasicIndexedRead arrPtrsHdrSize Nothing wordRep res addr idx
539 doWriteOffAddrOp maybe_pre_write_cast rep [] [addr,idx,val]
540 = mkBasicIndexedWrite 0 maybe_pre_write_cast rep addr idx val
541 doWriteOffAddrOp _ _ _ _
542 = panic "CgPrimOp: doWriteOffAddrOp"
544 doWriteByteArrayOp maybe_pre_write_cast rep [] [addr,idx,val]
545 = mkBasicIndexedWrite arrWordsHdrSize maybe_pre_write_cast rep addr idx val
546 doWriteByteArrayOp _ _ _ _
547 = panic "CgPrimOp: doWriteByteArrayOp"
549 doWritePtrArrayOp addr idx val
550 = do stmtC (setInfo addr (CmmLit (CmmLabel mkMAP_DIRTY_infoLabel)))
551 mkBasicIndexedWrite arrPtrsHdrSize Nothing wordRep addr idx val
554 mkBasicIndexedRead off Nothing read_rep res base idx
555 = stmtC (CmmAssign res (cmmLoadIndexOffExpr off read_rep base idx))
556 mkBasicIndexedRead off (Just cast) read_rep res base idx
557 = stmtC (CmmAssign res (CmmMachOp cast [
558 cmmLoadIndexOffExpr off read_rep base idx]))
560 mkBasicIndexedWrite off Nothing write_rep base idx val
561 = stmtC (CmmStore (cmmIndexOffExpr off write_rep base idx) val)
562 mkBasicIndexedWrite off (Just cast) write_rep base idx val
563 = stmtC (CmmStore (cmmIndexOffExpr off write_rep base idx) (CmmMachOp cast [val]))
565 -- ----------------------------------------------------------------------------
568 cmmIndexOffExpr :: ByteOff -> MachRep -> CmmExpr -> CmmExpr -> CmmExpr
569 cmmIndexOffExpr off rep base idx
570 = cmmIndexExpr rep (cmmOffsetB base off) idx
572 cmmLoadIndexOffExpr :: ByteOff -> MachRep -> CmmExpr -> CmmExpr -> CmmExpr
573 cmmLoadIndexOffExpr off rep base idx
574 = CmmLoad (cmmIndexOffExpr off rep base idx) rep
576 setInfo :: CmmExpr -> CmmExpr -> CmmStmt
577 setInfo closure_ptr info_ptr = CmmStore closure_ptr info_ptr