tweag/asterius
1
1
Fork 0
mirror of https://github.com/tweag/asterius.git synced 2024-10-05 21:30:49 +03:00
Code Issues Projects Releases Wiki Activity

Make use of precomputed Int and Char closures in the runtime (#534)

Browse Source
This commit is contained in:
George Karachalias 2020-04-02 02:14:55 +02:00 committed by GitHub
parent eac16d6ea9
commit 661b08e949
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 39 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
asterius/src/Asterius/Builtins.hs
View File

@ -1162,14 +1162,47 @@ rtsMkCharFunction :: BuiltinsOptions -> AsteriusModule
rtsMkCharFunction _ = runEDSL "rts_mkChar" $ do
setReturnTypes [I64]
[i] <- params [I64]
p <- call' "allocate" [mainCapability, constI64 2] I64
storeI64 p 0 $ symbol "ghczmprim_GHCziTypes_Czh_con_info"
storeI64 p 8 i
emit p
if'
[I64]
(i `ltUInt64` constI64 256)
-- If the character in question is in the range [0..255] we use the
-- trick that GHC uses, and instead of generating a heap-allocated Char
-- closure, we simply return the address of the statically allocated
-- Char. See stg_CHARLIKE_closure in
-- ghc-toolkit/boot-libs/rts/StgMiscClosures.cmm
( let offset = i `mulInt64` constI64 16
in emit $ symbol "stg_CHARLIKE_closure" `addInt64` offset
)
-- Otherwise, we fall back to the more inefficient
-- approach and generate a dynamic closure.
$ do
p <- call' "allocate" [mainCapability, constI64 2] I64
storeI64 p 0 $ symbol "ghczmprim_GHCziTypes_Czh_con_info"
storeI64 p 8 i
emit p
rtsMkIntFunction :: BuiltinsOptions -> AsteriusModule
rtsMkIntFunction opts =
rtsMkHelper opts "rts_mkInt" "ghczmprim_GHCziTypes_Izh_con_info"
rtsMkIntFunction _ = runEDSL "rts_mkInt" $ do
setReturnTypes [I64]
[i] <- params [I64]
if'
[I64]
((i `leSInt64` constI64 16) `andInt32` (i `geSInt64` constI64 (-16)))
-- If the integer in question is in the range [-16..16] we use the
-- trick that GHC uses, and instead of generating a heap-allocated Int
-- closure, we simply return the address of the statically allocated
-- Int. See stg_INTLIKE_closure in
-- ghc-toolkit/boot-libs/rts/StgMiscClosures.cmm
( let offset = (i `addInt64` constI64 16) `mulInt64` constI64 16
in emit $ symbol "stg_INTLIKE_closure" `addInt64` offset
)
-- Otherwise, we fall back to the more inefficient
-- approach and generate a dynamic closure.
$ do
p <- call' "allocate" [mainCapability, constI64 2] I64
storeI64 p 0 $ symbol "ghczmprim_GHCziTypes_Izh_con_info"
storeI64 p 8 i
emit p
rtsMkWordFunction :: BuiltinsOptions -> AsteriusModule
rtsMkWordFunction opts =