module GHC.Cmm.CallConv (
ParamLocation(..),
assignArgumentsPos,
assignStack,
realArgRegsCover,
allArgRegsCover
) where
import GHC.Prelude
import GHC.Cmm.Expr
import GHC.Cmm.Reg (GlobalArgRegs(..))
import GHC.Runtime.Heap.Layout
import GHC.Cmm (Convention(..))
import GHC.Platform
import GHC.Platform.Reg.Class
import GHC.Platform.Profile
import GHC.Utils.Outputable
import GHC.Utils.Panic
import Data.Maybe ( maybeToList )
import Data.List (nub)
data ParamLocation
= RegisterParam GlobalReg
| StackParam ByteOff
instance Outputable ParamLocation where
ppr :: ParamLocation -> SDoc
ppr (RegisterParam GlobalReg
g) = GlobalReg -> SDoc
forall a. Outputable a => a -> SDoc
ppr GlobalReg
g
ppr (StackParam Int
p) = Int -> SDoc
forall a. Outputable a => a -> SDoc
ppr Int
p
assignArgumentsPos :: Profile
-> ByteOff
-> Convention
-> (a -> CmmType)
-> [a]
-> (
ByteOff
, [(a, ParamLocation)]
)
assignArgumentsPos :: forall a.
Profile
-> Int
-> Convention
-> (a -> CmmType)
-> [a]
-> (Int, [(a, ParamLocation)])
assignArgumentsPos Profile
profile Int
off Convention
conv a -> CmmType
arg_ty [a]
reps = (Int
stk_off, [(a, ParamLocation)]
assignments)
where
platform :: Platform
platform = Profile -> Platform
profilePlatform Profile
profile
regs :: AvailRegs
regs = case ([a]
reps, Convention
conv) of
([a]
_, Convention
NativeNodeCall) -> Platform -> AvailRegs
getRegsWithNode Platform
platform
([a]
_, Convention
NativeDirectCall) -> Platform -> AvailRegs
getRegsWithoutNode Platform
platform
([a
_], Convention
NativeReturn) -> Platform -> AvailRegs
allRegs Platform
platform
([a]
_, Convention
NativeReturn) -> Platform -> AvailRegs
getRegsWithNode Platform
platform
([a]
_, Convention
GC) -> Platform -> AvailRegs
allRegs Platform
platform
([a]
_, Convention
Slow) -> AvailRegs
nodeOnly
([(a, ParamLocation)]
reg_assts, [a]
stk_args) = [(a, ParamLocation)]
-> [a] -> AvailRegs -> ([(a, ParamLocation)], [a])
assign_regs [] [a]
reps AvailRegs
regs
(Int
stk_off, [(a, ParamLocation)]
stk_assts) = Platform
-> Int -> (a -> CmmType) -> [a] -> (Int, [(a, ParamLocation)])
forall a.
Platform
-> Int -> (a -> CmmType) -> [a] -> (Int, [(a, ParamLocation)])
assignStack Platform
platform Int
off a -> CmmType
arg_ty [a]
stk_args
assignments :: [(a, ParamLocation)]
assignments = [(a, ParamLocation)]
reg_assts [(a, ParamLocation)]
-> [(a, ParamLocation)] -> [(a, ParamLocation)]
forall a. [a] -> [a] -> [a]
++ [(a, ParamLocation)]
stk_assts
assign_regs :: [(a, ParamLocation)]
-> [a] -> AvailRegs -> ([(a, ParamLocation)], [a])
assign_regs [(a, ParamLocation)]
assts [] AvailRegs
_ = ([(a, ParamLocation)]
assts, [])
assign_regs [(a, ParamLocation)]
assts (a
r:[a]
rs) AvailRegs
regs | CmmType -> Bool
isVecType CmmType
ty = ([(a, ParamLocation)], [a])
vec
| CmmType -> Bool
isFloatType CmmType
ty = ([(a, ParamLocation)], [a])
float
| Bool
otherwise = ([(a, ParamLocation)], [a])
int
where vec :: ([(a, ParamLocation)], [a])
vec = case AvailRegs
regs of
AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls (Int
s:[Int]
ss)
| Width -> Profile -> Bool
passVectorInReg Width
w Profile
profile
-> let reg_class :: Int -> GlobalReg
reg_class = case Width
w of
Width
W128 -> Int -> GlobalReg
XmmReg
Width
W256 -> Int -> GlobalReg
YmmReg
Width
W512 -> Int -> GlobalReg
ZmmReg
Width
_ -> String -> Int -> GlobalReg
forall a. HasCallStack => String -> a
panic String
"CmmCallConv.assignArgumentsPos: Invalid vector width"
in (ParamLocation, AvailRegs) -> ([(a, ParamLocation)], [a])
k (GlobalReg -> ParamLocation
RegisterParam (Int -> GlobalReg
reg_class Int
s), [GlobalReg]
-> [GlobalReg] -> [GlobalReg] -> [GlobalReg] -> [Int] -> AvailRegs
AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls [Int]
ss)
AvailRegs
_ -> ([(a, ParamLocation)]
assts, a
ra -> [a] -> [a]
forall a. a -> [a] -> [a]
:[a]
rs)
float :: ([(a, ParamLocation)], [a])
float = case (Width
w, AvailRegs
regs) of
(Width
W32, AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls (Int
s:[Int]
ss))
| Bool
passFloatInXmm -> (ParamLocation, AvailRegs) -> ([(a, ParamLocation)], [a])
k (GlobalReg -> ParamLocation
RegisterParam (Int -> GlobalReg
FloatReg Int
s), [GlobalReg]
-> [GlobalReg] -> [GlobalReg] -> [GlobalReg] -> [Int] -> AvailRegs
AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls [Int]
ss)
(Width
W32, AvailRegs [GlobalReg]
vs (GlobalReg
f:[GlobalReg]
fs) [GlobalReg]
ds [GlobalReg]
ls [Int]
ss)
| Bool -> Bool
not Bool
passFloatInXmm -> (ParamLocation, AvailRegs) -> ([(a, ParamLocation)], [a])
k (GlobalReg -> ParamLocation
RegisterParam GlobalReg
f, [GlobalReg]
-> [GlobalReg] -> [GlobalReg] -> [GlobalReg] -> [Int] -> AvailRegs
AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls [Int]
ss)
(Width
W64, AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls (Int
s:[Int]
ss))
| Bool
passFloatInXmm -> (ParamLocation, AvailRegs) -> ([(a, ParamLocation)], [a])
k (GlobalReg -> ParamLocation
RegisterParam (Int -> GlobalReg
DoubleReg Int
s), [GlobalReg]
-> [GlobalReg] -> [GlobalReg] -> [GlobalReg] -> [Int] -> AvailRegs
AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls [Int]
ss)
(Width
W64, AvailRegs [GlobalReg]
vs [GlobalReg]
fs (GlobalReg
d:[GlobalReg]
ds) [GlobalReg]
ls [Int]
ss)
| Bool -> Bool
not Bool
passFloatInXmm -> (ParamLocation, AvailRegs) -> ([(a, ParamLocation)], [a])
k (GlobalReg -> ParamLocation
RegisterParam GlobalReg
d, [GlobalReg]
-> [GlobalReg] -> [GlobalReg] -> [GlobalReg] -> [Int] -> AvailRegs
AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls [Int]
ss)
(Width, AvailRegs)
_ -> ([(a, ParamLocation)]
assts, (a
ra -> [a] -> [a]
forall a. a -> [a] -> [a]
:[a]
rs))
int :: ([(a, ParamLocation)], [a])
int = case (Width
w, AvailRegs
regs) of
(Width
W128, AvailRegs
_) -> String -> ([(a, ParamLocation)], [a])
forall a. HasCallStack => String -> a
panic String
"W128 unsupported register type"
(Width
_, AvailRegs (GlobalReg
v:[GlobalReg]
vs) [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls [Int]
ss) | Width -> Int
widthInBits Width
w Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
<= Width -> Int
widthInBits (Platform -> Width
wordWidth Platform
platform)
-> (ParamLocation, AvailRegs) -> ([(a, ParamLocation)], [a])
k (GlobalReg -> ParamLocation
RegisterParam GlobalReg
v, [GlobalReg]
-> [GlobalReg] -> [GlobalReg] -> [GlobalReg] -> [Int] -> AvailRegs
AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls [Int]
ss)
(Width
_, AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds (GlobalReg
l:[GlobalReg]
ls) [Int]
ss) | Width -> Int
widthInBits Width
w Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> Width -> Int
widthInBits (Platform -> Width
wordWidth Platform
platform)
-> (ParamLocation, AvailRegs) -> ([(a, ParamLocation)], [a])
k (GlobalReg -> ParamLocation
RegisterParam GlobalReg
l, [GlobalReg]
-> [GlobalReg] -> [GlobalReg] -> [GlobalReg] -> [Int] -> AvailRegs
AvailRegs [GlobalReg]
vs [GlobalReg]
fs [GlobalReg]
ds [GlobalReg]
ls [Int]
ss)
(Width, AvailRegs)
_ -> ([(a, ParamLocation)]
assts, (a
ra -> [a] -> [a]
forall a. a -> [a] -> [a]
:[a]
rs))
k :: (ParamLocation, AvailRegs) -> ([(a, ParamLocation)], [a])
k (ParamLocation
asst, AvailRegs
regs') = [(a, ParamLocation)]
-> [a] -> AvailRegs -> ([(a, ParamLocation)], [a])
assign_regs ((a
r, ParamLocation
asst) (a, ParamLocation) -> [(a, ParamLocation)] -> [(a, ParamLocation)]
forall a. a -> [a] -> [a]
: [(a, ParamLocation)]
assts) [a]
rs AvailRegs
regs'
ty :: CmmType
ty = a -> CmmType
arg_ty a
r
w :: Width
w = CmmType -> Width
typeWidth CmmType
ty
passFloatInXmm :: Bool
passFloatInXmm = Platform -> Bool
passFloatArgsInXmm Platform
platform
passFloatArgsInXmm :: Platform -> Bool
passFloatArgsInXmm :: Platform -> Bool
passFloatArgsInXmm Platform
platform =
case Platform -> Arch
platformArch Platform
platform of
Arch
ArchX86_64 -> Bool
True
Arch
ArchX86 -> Bool
False
Arch
_ -> Bool
False
passVectorInReg :: Width -> Profile -> Bool
passVectorInReg :: Width -> Profile -> Bool
passVectorInReg Width
_ Profile
_ = Bool
True
assignStack :: Platform -> ByteOff -> (a -> CmmType) -> [a]
-> (
ByteOff
, [(a, ParamLocation)]
)
assignStack :: forall a.
Platform
-> Int -> (a -> CmmType) -> [a] -> (Int, [(a, ParamLocation)])
assignStack Platform
platform Int
offset a -> CmmType
arg_ty [a]
args = Int -> [(a, ParamLocation)] -> [a] -> (Int, [(a, ParamLocation)])
assign_stk Int
offset [] ([a] -> [a]
forall a. [a] -> [a]
reverse [a]
args)
where
assign_stk :: Int -> [(a, ParamLocation)] -> [a] -> (Int, [(a, ParamLocation)])
assign_stk Int
offset [(a, ParamLocation)]
assts [] = (Int
offset, [(a, ParamLocation)]
assts)
assign_stk Int
offset [(a, ParamLocation)]
assts (a
r:[a]
rs)
= Int -> [(a, ParamLocation)] -> [a] -> (Int, [(a, ParamLocation)])
assign_stk Int
off' ((a
r, Int -> ParamLocation
StackParam Int
off') (a, ParamLocation) -> [(a, ParamLocation)] -> [(a, ParamLocation)]
forall a. a -> [a] -> [a]
: [(a, ParamLocation)]
assts) [a]
rs
where w :: Width
w = CmmType -> Width
typeWidth (a -> CmmType
arg_ty a
r)
off' :: Int
off' = Int
offset Int -> Int -> Int
forall a. Num a => a -> a -> a
+ Int
size
size :: Int
size = Platform -> Int -> Int
roundUpToWords Platform
platform (Width -> Int
widthInBytes Width
w)
data AvailRegs
= AvailRegs
{ AvailRegs -> [GlobalReg]
availVanillaRegs :: [GlobalReg]
, AvailRegs -> [GlobalReg]
availFloatRegs :: [GlobalReg]
, AvailRegs -> [GlobalReg]
availDoubleRegs :: [GlobalReg]
, AvailRegs -> [GlobalReg]
availLongRegs :: [GlobalReg]
, AvailRegs -> [Int]
availXMMRegs :: [Int]
}
noAvailRegs :: AvailRegs
noAvailRegs :: AvailRegs
noAvailRegs = [GlobalReg]
-> [GlobalReg] -> [GlobalReg] -> [GlobalReg] -> [Int] -> AvailRegs
AvailRegs [] [] [] [] []
getRegsWithoutNode, getRegsWithNode :: Platform -> AvailRegs
getRegsWithoutNode :: Platform -> AvailRegs
getRegsWithoutNode Platform
platform =
AvailRegs
{ availVanillaRegs :: [GlobalReg]
availVanillaRegs = (GlobalReg -> Bool) -> [GlobalReg] -> [GlobalReg]
forall a. (a -> Bool) -> [a] -> [a]
filter (\GlobalReg
r -> GlobalReg
r GlobalReg -> GlobalReg -> Bool
forall a. Eq a => a -> a -> Bool
/= GlobalReg
node) (Platform -> [GlobalReg]
realVanillaRegs Platform
platform)
, availFloatRegs :: [GlobalReg]
availFloatRegs = Platform -> [GlobalReg]
realFloatRegs Platform
platform
, availDoubleRegs :: [GlobalReg]
availDoubleRegs = Platform -> [GlobalReg]
realDoubleRegs Platform
platform
, availLongRegs :: [GlobalReg]
availLongRegs = Platform -> [GlobalReg]
realLongRegs Platform
platform
, availXMMRegs :: [Int]
availXMMRegs = Platform -> [Int]
realXmmRegNos Platform
platform }
getRegsWithNode :: Platform -> AvailRegs
getRegsWithNode Platform
platform =
AvailRegs
{ availVanillaRegs :: [GlobalReg]
availVanillaRegs = if [GlobalReg] -> Bool
forall a. [a] -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null (Platform -> [GlobalReg]
realVanillaRegs Platform
platform)
then [Int -> GlobalReg
VanillaReg Int
1]
else Platform -> [GlobalReg]
realVanillaRegs Platform
platform
, availFloatRegs :: [GlobalReg]
availFloatRegs = Platform -> [GlobalReg]
realFloatRegs Platform
platform
, availDoubleRegs :: [GlobalReg]
availDoubleRegs = Platform -> [GlobalReg]
realDoubleRegs Platform
platform
, availLongRegs :: [GlobalReg]
availLongRegs = Platform -> [GlobalReg]
realLongRegs Platform
platform
, availXMMRegs :: [Int]
availXMMRegs = Platform -> [Int]
realXmmRegNos Platform
platform }
allFloatRegs, allDoubleRegs, allLongRegs :: Platform -> [GlobalReg]
allVanillaRegs :: Platform -> [GlobalReg]
allXmmRegs :: Platform -> [Int]
allVanillaRegs :: Platform -> [GlobalReg]
allVanillaRegs Platform
platform = (Int -> GlobalReg) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> [a] -> [b]
map Int -> GlobalReg
VanillaReg ([Int] -> [GlobalReg]) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> a -> b
$ Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Vanilla_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
allFloatRegs :: Platform -> [GlobalReg]
allFloatRegs Platform
platform = (Int -> GlobalReg) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> [a] -> [b]
map Int -> GlobalReg
FloatReg ([Int] -> [GlobalReg]) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> a -> b
$ Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Float_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
allDoubleRegs :: Platform -> [GlobalReg]
allDoubleRegs Platform
platform = (Int -> GlobalReg) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> [a] -> [b]
map Int -> GlobalReg
DoubleReg ([Int] -> [GlobalReg]) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> a -> b
$ Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Double_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
allLongRegs :: Platform -> [GlobalReg]
allLongRegs Platform
platform = (Int -> GlobalReg) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> [a] -> [b]
map Int -> GlobalReg
LongReg ([Int] -> [GlobalReg]) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> a -> b
$ Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Long_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
allXmmRegs :: Platform -> [Int]
allXmmRegs Platform
platform = Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_XMM_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
realFloatRegs, realDoubleRegs, realLongRegs :: Platform -> [GlobalReg]
realVanillaRegs :: Platform -> [GlobalReg]
realVanillaRegs :: Platform -> [GlobalReg]
realVanillaRegs Platform
platform = (Int -> GlobalReg) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> [a] -> [b]
map Int -> GlobalReg
VanillaReg ([Int] -> [GlobalReg]) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> a -> b
$ Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Real_Vanilla_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
realFloatRegs :: Platform -> [GlobalReg]
realFloatRegs Platform
platform = (Int -> GlobalReg) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> [a] -> [b]
map Int -> GlobalReg
FloatReg ([Int] -> [GlobalReg]) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> a -> b
$ Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Real_Float_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
realDoubleRegs :: Platform -> [GlobalReg]
realDoubleRegs Platform
platform = (Int -> GlobalReg) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> [a] -> [b]
map Int -> GlobalReg
DoubleReg ([Int] -> [GlobalReg]) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> a -> b
$ Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Real_Double_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
realLongRegs :: Platform -> [GlobalReg]
realLongRegs Platform
platform = (Int -> GlobalReg) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> [a] -> [b]
map Int -> GlobalReg
LongReg ([Int] -> [GlobalReg]) -> [Int] -> [GlobalReg]
forall a b. (a -> b) -> a -> b
$ Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Real_Long_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
realXmmRegNos :: Platform -> [Int]
realXmmRegNos :: Platform -> [Int]
realXmmRegNos Platform
platform
| Platform -> Bool
isSse2Enabled Platform
platform Bool -> Bool -> Bool
|| Platform -> Arch
platformArch Platform
platform Arch -> Arch -> Bool
forall a. Eq a => a -> a -> Bool
== Arch
ArchAArch64
= Int -> [Int]
regList (PlatformConstants -> Int
pc_MAX_Real_XMM_REG (Platform -> PlatformConstants
platformConstants Platform
platform))
| Bool
otherwise
= []
regList :: Int -> [Int]
regList :: Int -> [Int]
regList Int
n = [Int
1 .. Int
n]
allRegs :: Platform -> AvailRegs
allRegs :: Platform -> AvailRegs
allRegs Platform
platform =
AvailRegs
{ availVanillaRegs :: [GlobalReg]
availVanillaRegs = Platform -> [GlobalReg]
allVanillaRegs Platform
platform
, availFloatRegs :: [GlobalReg]
availFloatRegs = Platform -> [GlobalReg]
allFloatRegs Platform
platform
, availDoubleRegs :: [GlobalReg]
availDoubleRegs = Platform -> [GlobalReg]
allDoubleRegs Platform
platform
, availLongRegs :: [GlobalReg]
availLongRegs = Platform -> [GlobalReg]
allLongRegs Platform
platform
, availXMMRegs :: [Int]
availXMMRegs = Platform -> [Int]
allXmmRegs Platform
platform }
nodeOnly :: AvailRegs
nodeOnly :: AvailRegs
nodeOnly = AvailRegs
noAvailRegs { availVanillaRegs = [VanillaReg 1] }
realArgRegsCover :: Platform
-> GlobalArgRegs
-> [GlobalReg]
realArgRegsCover :: Platform -> GlobalArgRegs -> [GlobalReg]
realArgRegsCover Platform
platform GlobalArgRegs
argRegs
= Platform -> [GlobalReg]
realVanillaRegs Platform
platform
[GlobalReg] -> [GlobalReg] -> [GlobalReg]
forall a. [a] -> [a] -> [a]
++ Platform -> [GlobalReg]
realLongRegs Platform
platform
[GlobalReg] -> [GlobalReg] -> [GlobalReg]
forall a. [a] -> [a] -> [a]
++ [[GlobalReg]] -> [GlobalReg]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat
( [ Platform -> [GlobalReg]
realFloatRegs Platform
platform | Bool
wantFP, Bool -> Bool
not (Platform -> Bool
passFloatArgsInXmm Platform
platform) ]
[[GlobalReg]] -> [[GlobalReg]] -> [[GlobalReg]]
forall a. [a] -> [a] -> [a]
++ [ Platform -> [GlobalReg]
realDoubleRegs Platform
platform | Bool
wantFP ]
)
[GlobalReg] -> [GlobalReg] -> [GlobalReg]
forall a. [a] -> [a] -> [a]
++ [ Int -> GlobalReg
mkVecReg Int
i | Int -> GlobalReg
mkVecReg <- Maybe (Int -> GlobalReg) -> [Int -> GlobalReg]
forall a. Maybe a -> [a]
maybeToList Maybe (Int -> GlobalReg)
mbMkVecReg
, Int
i <- Platform -> [Int]
realXmmRegNos Platform
platform ]
where
wantFP :: Bool
wantFP = case Arch -> RegArch
registerArch (Platform -> Arch
platformArch Platform
platform) of
RegArch
Unified -> GlobalArgRegs
argRegs GlobalArgRegs -> GlobalArgRegs -> Bool
forall a. Eq a => a -> a -> Bool
== GlobalArgRegs
SCALAR_ARG_REGS
RegArch
Separate -> GlobalArgRegs
argRegs GlobalArgRegs -> GlobalArgRegs -> Bool
forall a. Ord a => a -> a -> Bool
>= GlobalArgRegs
SCALAR_ARG_REGS
RegArch
NoVectors -> GlobalArgRegs
argRegs GlobalArgRegs -> GlobalArgRegs -> Bool
forall a. Ord a => a -> a -> Bool
>= GlobalArgRegs
SCALAR_ARG_REGS
mbMkVecReg :: Maybe (Int -> GlobalReg)
mbMkVecReg = case Arch -> RegArch
registerArch (Platform -> Arch
platformArch Platform
platform) of
RegArch
Unified -> Maybe (Int -> GlobalReg)
mb_xyzmm
RegArch
Separate -> Maybe (Int -> GlobalReg)
mb_xyzmm
RegArch
NoVectors -> Maybe (Int -> GlobalReg)
forall a. Maybe a
Nothing
mb_xyzmm :: Maybe (Int -> GlobalReg)
mb_xyzmm = case GlobalArgRegs
argRegs of
GlobalArgRegs
V16_ARG_REGS -> (Int -> GlobalReg) -> Maybe (Int -> GlobalReg)
forall a. a -> Maybe a
Just Int -> GlobalReg
XmmReg
GlobalArgRegs
V32_ARG_REGS -> (Int -> GlobalReg) -> Maybe (Int -> GlobalReg)
forall a. a -> Maybe a
Just Int -> GlobalReg
YmmReg
GlobalArgRegs
V64_ARG_REGS -> (Int -> GlobalReg) -> Maybe (Int -> GlobalReg)
forall a. a -> Maybe a
Just Int -> GlobalReg
ZmmReg
GlobalArgRegs
_ -> Maybe (Int -> GlobalReg)
forall a. Maybe a
Nothing
allArgRegsCover :: Platform
-> GlobalArgRegs
-> [GlobalReg]
allArgRegsCover :: Platform -> GlobalArgRegs -> [GlobalReg]
allArgRegsCover Platform
platform GlobalArgRegs
argRegs =
[GlobalReg] -> [GlobalReg]
forall a. Eq a => [a] -> [a]
nub (GlobalReg
node GlobalReg -> [GlobalReg] -> [GlobalReg]
forall a. a -> [a] -> [a]
: Platform -> GlobalArgRegs -> [GlobalReg]
realArgRegsCover Platform
platform GlobalArgRegs
argRegs)
where
node :: GlobalReg
node = Int -> GlobalReg
VanillaReg Int
1