1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
-- | See https://gitlab.haskell.org/ghc/ghc/-/issues/22933
-- and https://gitlab.haskell.org/ghc/ghc/-/issues/22966
--
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE CApiFFI #-}
{-# LANGUAGE CPP #-}
{-# LANGUAGE MagicHash #-}
{-# LANGUAGE UnboxedTuples #-}
{-# LANGUAGE UnliftedFFITypes #-}
module Data.Mod.Compat
( timesWord2#
, remWord2#
) where
#ifdef aarch64_HOST_ARCH
import GHC.Exts (Word(..), Word#, timesWord#)
timesWord2# :: Word# -> Word# -> (# Word#, Word# #)
timesWord2# x y = (# z, timesWord# x y #)
where
!(W# z) = c_umulh (W# x) (W# y)
{-# INLINE timesWord2# #-}
foreign import capi unsafe "aarch64.h umulh" c_umulh :: Word -> Word -> Word
remWord2# :: Word# -> Word# -> Word# -> Word#
remWord2# lo hi m = r
where
!(W# r) = c_umodh (W# lo) (W# hi) (W# m)
{-# INLINE remWord2# #-}
foreign import capi unsafe "aarch64.h umodh" c_umodh :: Word -> Word -> Word -> Word
#else
import GHC.Exts (Word#, timesWord2#, quotRemWord2#)
remWord2# :: Word# -> Word# -> Word# -> Word#
remWord2# lo hi m = r
where
!(# _, r #) = quotRemWord2# hi lo m
{-# INLINE remWord2# #-}
#endif
|
