Cool -- but please do write a blog post or something to distil what you have
learned. I have not followed this thread in detail, and I bet others haven't
either. But it'd be a pity for your learning not to be shared somehow!
Thanks
Simon
| -----Original Message-----
| From: ghc-devs <ghc-devs-boun...@haskell.org> On Behalf Of Alexis King
| Sent: 04 April 2020 02:46
| To: Sebastian Graf <sgraf1...@gmail.com>
| Cc: ghc-devs <ghc-devs@haskell.org>
| Subject: Re: Fusing loops by specializing on functions with SpecConstr?
|
|
| I fiddled with alternative representations for a while and didn’t make any
| progress—it was too easy to end up with code explosion in the presence of
| any unknown calls—but I seem to have found a RULES-based approach that
| works very well on the examples I’ve tried. It’s quite simple, which makes
| it especially appealing!
|
| I started by defining a wrapper around the `SF` constructor to attach
| rules to:
|
| mkSF :: (a -> s -> Step s b) -> s -> SF a b
| mkSF = SF
| {-# INLINE CONLIKE [1] mkSF #-}
|
| I then changed the definitions of (.), (***), (&&&), (+++), and (&&&) to
| use `mkSF` instead of `SF`, but I left the other methods alone, so they
| just use `SF` directly. Then I defined two rewrite rules:
|
| {-# RULES
| "mkSF @((), _)" forall f s. mkSF f ((), s) =
| SF (\a s1 -> case f a ((), s1) of Step ((), s2) b -> Step s2 b) s
| "mkSF @(_, ())" forall f s. mkSF f (s, ()) =
| SF (\a s1 -> case f a (s1, ()) of Step (s2, ()) b -> Step s2 b) s
| #-}
|
| That’s it. These two rules alone are enough to eliminate the redundant
| tupling. Now the optimized version of `mapMaybeSF` is beautiful!
|
| mapMaybeSF = \ @ a @ b f -> case f of { SF @ s f2 s2 ->
| SF (\ a1 s1 -> case a1 of {
| Nothing -> case s1 of dt { __DEFAULT -> Step dt Nothing }
| Just x -> case f2 x s1 of {
| Step s2' c1 -> Step s2' (Just c1) }})
| s2 }
|
| So unless this breaks down in some larger situation I’m not aware of, I
| think this solves my problem without the need for any fancy SpecConstr
| shenanigans. Many thanks to you, Sebastian, for pointing me in the right
| direction!
|
| Alexis
| _______________________________________________
| ghc-devs mailing list
| ghc-devs@haskell.org
| https://nam06.safelinks.protection.outlook.com/?url=http%3A%2F%2Fmail.hask
| ell.org%2Fcgi-bin%2Fmailman%2Flistinfo%2Fghc-
| devs&data=02%7C01%7Csimonpj%40microsoft.com%7Cfa33485e4b3643e695fe08d7
| d839ecb9%7C72f988bf86f141af91ab2d7cd011db47%7C1%7C0%7C637215615608529596&a
| mp;sdata=CSDPKcz%2BnVuQC%2BitP%2FZXpPpOtcTxUAfe0fxiNZAfTrs%3D&reserved
| =0
_______________________________________________
ghc-devs mailing list
ghc-devs@haskell.org
http://mail.haskell.org/cgi-bin/mailman/listinfo/ghc-devs
