> On Nov 30, 2020, at 11:41 AM, Christiaan Baaij <christiaan.ba...@gmail.com> 
> wrote:
> 
> Richard, I just watched your talk about simplifying the constraint solver: 
> https://www.youtube.com/watch?v=flwz6V5Ps8w 
> <https://www.youtube.com/watch?v=flwz6V5Ps8w>
> and then realised that no flattening at all means also no flattening of 
> [W]anteds.

Yes, that's true.

> 
> Now, let's say I have a type family called "Magic" for which I want to write 
> a constraint solver plugin; and some type family "Foo" and "Bar"
> And then lets say we start out with an original [W]anted: Foo (Magic 8) ~ Bar 
> 42
> 
> Then if I understand flattening correctly, that would get flattened to:
> [w] Magic 8 ~ fmv1
> [w] fmv1 ~ 42
> [w] Foo fmv1 ~ fmv2
> [w] fmv2 ~ Bar fmv1
> 
> Correct?

I get

[W] Magic 8 ~ fmv1
[W] Foo fmv1 ~ fmv2
[W] Bar 42 ~ fmv3
[W] fmv2 ~ fmv3

> 
> So currently, with flattened wanted, all a constraint solver plugin has to do 
> is look for [w]anted where the head of the type family application is 
> "Magic", [w] Magic 8 ~ fmv1.
> Then perhaps do some substitution for fmv1 to get back [w] Magic 8 ~ 42, use 
> its internal solving rules to reduce Magic 8 to 42, and conclude what "[w] 42 
> ~ 42" can be solved by reflexivity.

It looks like maybe you edited your example afterwards? This doesn't quite line 
up, because we don't have [W] Magic 8 ~ 42 anywhere. But I see your point, in 
that flattening brings the type family out to the top level.

> 
> But in a version without flattening, all we would see is:
> [W] Foo (Magic 8) ~ Foo 42 

Did you mean [W] Foo (Magic 8) ~ Bar 42? Then, yes.

> 
> And all I can do is to traverse the entire [W]anted, look for type family 
> applications of "Magic" an then improve to:
> [W] Foo 42 ~ Bar 42
> 
> Now... I can't declare that [W]anted solved, because i have no idea what Foo 
> and Bar do (they might also be type families that can only be solved through 
> constraint solver plugins).
> Currently, the API for constraint solving plugins allows me to return 
> _either_: 
> http://hackage.haskell.org/package/ghc-8.10.1/docs/TcRnTypes.html#t:TcPluginResult
>  
> <http://hackage.haskell.org/package/ghc-8.10.1/docs/TcRnTypes.html#t:TcPluginResult>
> A) A set of contradictions
> B) A set of solved [W] constraints + A set of new [W] constraints
> 
> With the above API, the only way to move forward is to: 1) declare "[w] Foo 
> (Magic 8) ~ Bar 42" solved, while at the same time emitting a new wanted "[w] 
> Foo 42 ~ Bar 42"

Yes, that's what you would do. Indeed, it's what GHC does whenever simplifying 
part of a Wanted.

> But it's basically the same wanted... so I would like a function of type  
> "updateWantedType :: Ct -> TcPredType -> Ct" (which I don't think exists in 
> the current GHC API)

This would not be hard to write, though you'd also need evidence relating the 
old type and the new one. Here's an attempt (that I have not tested, beyond the 
ability to compile it):

> import GHC.Plugins
> import GHC.Tc.Utils.TcType
> import GHC.Tc.Types.Evidence
> import GHC.Tc.Types.Constraint
> import GHC.Tc.Plugin
> 
> updateWantedType :: Ct            -- initial constraint; always unsolved 
> equality wanted
>                                   --   t1 ~N t2
>                  -> TcPredType    -- desired new constraint: s1 ~N s2
>                  -> CoercionN     -- lhs_co :: t1 ~N s1
>                  -> CoercionN     -- rhs_co :: t2 ~N s2
>                  -> TcPluginM (Ct, EvTerm)  -- new constraint, evidence for 
> *old* constraint
> -- NB: the *old* constraint is solved by the returned EvTerm;
> -- return the EvTerm with the *old* constraint in TcPluginOk
> updateWantedType old_ct new_pred lhs_co rhs_co = do
>   new_w <- newWanted (ctLoc old_ct) new_pred
>   let new_co = ctEvCoercion new_w    -- :: s1 ~N s2
>       old_co = lhs_co `mkTransCo` new_co `mkTransCo` mkSymCo rhs_co
>       old_ev = evCoercion old_co
>   return (mkNonCanonical new_w, old_ev)

That would seem to do the trick. This function is essentially extracted from 
GHC.Tc.Solver.Canonical.rewriteEqEvidence, and GHC does it many many times 
while working with equalities.

> 
> But you might think the above way of working with the constraint solver as a 
> plugin write, where we "solve" a wanted, only to emit an "improved" wanted, 
> is _not_ the way things should work.

I don't quite know how to interpret this statement. This way of working is 
commonplace. For example, it's what we do when trying to prove a goal by hand 
when working backwards: we simplify the goal Wanted by rewriting it to another 
(presumably simpler) Wanted -- as long as the new Wanted can prove the old one.

> If that's the case, how do you envision plugin writers interacting with the 
> solver?

As you've outlined above.

> Note that the above way of interacting with the constraint solver, "solving" 
> + emitting "improved", is something I already sorta do, but in a different 
> context:
> In "ghc-typelits-natnormalise", when asked to solve [w] "a + Foo b ~ Foo b + 
> c", I solve it, but then immediately emitting "[w] a ~ c".

And that's really the way to do it.

A slightly different problem than the one you've outlined here is that 
generating the coercions to pass to updatedWantedType might be challenging. So 
I wonder about a new kind of plugin, called a "rewriter plugin", that looks like

> rewriterPlugin :: TyCon -> [TcType] -> TcPluginM (Maybe TcType)

possibly with some other state passed in (including other constraints, if there 
is demand). The tycon would always be a type family, and the types would always 
exactly saturate the type family. Then, the plugin could reduce the type family 
if it knows how to do so. This seems to be a fairly simple interface and would 
be very, very easy to insert into the new infrastructure. (I'm a little worried 
about performance, because simplifying type families is code that's hammered 
on. But one step at a time.) Would that satisfy your need?

Richard

> 
> -- Christiaan
> 
> 
> 
> 
> 
> 
> On Fri, 20 Nov 2020 at 20:43, Richard Eisenberg <r...@richarde.dev 
> <mailto:r...@richarde.dev>> wrote:
> 
> 
>> On Nov 19, 2020, at 12:40 PM, Iavor Diatchki <iavor.diatc...@gmail.com 
>> <mailto:iavor.diatc...@gmail.com>> wrote:
>> 
>> Hello,
>> 
>> I haven't worked on that stuff in Haskell for a long time, but here are some 
>> thoughts:
>>   - I think plugins should generally be agnostic to the form of the 
>> constraints they are given, thus flattening or not should not affect 
>> them---after all, the user might have written the constraints in the 
>> "flattened" form in the first place.  So I think a plugin needs to convert 
>> the constraints to whatever form it assumes anyways.
>>   - I always thought that derived constraints were a pretty clever way for 
>> disseminating information in the constraint solver, is there a note 
>> somewhere on what's going to be the new mechanism?
> 
> This Note 
> (https://gitlab.haskell.org/ghc/ghc/-/blob/wip/derived-refactor/compiler/GHC/Tc/Types/Constraint.hs#L1678
>  
> <https://gitlab.haskell.org/ghc/ghc/-/blob/wip/derived-refactor/compiler/GHC/Tc/Types/Constraint.hs#L1678>)
>  is the best I can offer, though it may be out of date. In the final version 
> of the patch, that Note will naturally be updated. Essentially, the new plan 
> is just to use Wanteds instead of Deriveds, which will achieve the same goal 
> (but do not need to be distinct from Deriveds).
> 
> Richard
> 
>> 
>> -Iavor
>> 
>> 
>> On Thu, Nov 19, 2020 at 2:21 AM Christiaan Baaij <christiaan.ba...@gmail.com 
>> <mailto:christiaan.ba...@gmail.com>> wrote:
>> I always forget what flattening did/does. Is it the thing where it turns a 
>> "complex" type family application:
>> 
>> F (G y) (H z) ~ a
>> 
>> into:
>> 
>> G y ~ p
>> H z ~ q
>> F p q ~ a
>> 
>> ?
>> 
>> If so, then I'm all for removing that. Since I actually wrote/hacked a 
>> function that "reverts" that process (for [G]ivens only): 
>> https://hackage.haskell.org/package/ghc-tcplugins-extra-0.4/docs/src/GHC.TcPluginM.Extra.html#flattenGivens
>>  
>> <https://hackage.haskell.org/package/ghc-tcplugins-extra-0.4/docs/src/GHC.TcPluginM.Extra.html#flattenGivens>
>> Which I use in all of my plugins. (PS it should perhaps be called 
>> "unflattenGiven"? like I said, I always get confused about flatten vs 
>> unflatten).
>> 
>> On Thu, 19 Nov 2020 at 05:20, Richard Eisenberg <r...@richarde.dev 
>> <mailto:r...@richarde.dev>> wrote:
>> Hi all,
>> 
>> I'm hard at work on two significant refactorings within GHC's constraint 
>> solver. The first is at 
>> https://gitlab.haskell.org/ghc/ghc/-/merge_requests/4149 
>> <https://gitlab.haskell.org/ghc/ghc/-/merge_requests/4149>. It removes 
>> flattening meta-variables and flattening skolems. This is a very nice 
>> simplification. Instead, it just reduces type families directly. My other 
>> patch (held up by the first) is at 
>> https://gitlab.haskell.org/ghc/ghc/-/tree/wip/derived-refactor 
>> <https://gitlab.haskell.org/ghc/ghc/-/tree/wip/derived-refactor> and will 
>> remove Derived constraints, to be replaced by a little bit of cleverness in 
>> suppressing certain confusing error messages. My guess is that either or 
>> both of these will invalidate the current behavior of type-checker plugins. 
>> Sadly, this is not just a case of finding a new function that replicates the 
>> old behavior -- enough is changing under the hood that you might actually 
>> have to rewrite chunks of your code.
>> 
>> I have never written a type-checker plugin, and so I don't currently have 
>> advice for you. But if you are a plugin author affected by this change and 
>> want help, please reach out -- I would be happy to walk you through the 
>> changes, and then hopefully make a little video explaining the process to 
>> other plugin authors.
>> 
>> Neither patch will make it for 9.0, but I expect both to be in 9.2. There 
>> may be more where this came from 
>> (https://gitlab.haskell.org/ghc/ghc/-/issues/18965 
>> <https://gitlab.haskell.org/ghc/ghc/-/issues/18965>) in the future, but it's 
>> all for a good cause.
>> 
>> (I have bcc'd plugin authors that I'm aware of. Just adding this in case 
>> you're surprised at receiving this email.)
>> 
>> Thanks,
>> Richard
>> _______________________________________________
>> ghc-devs mailing list
>> ghc-devs@haskell.org <mailto:ghc-devs@haskell.org>
>> http://mail.haskell.org/cgi-bin/mailman/listinfo/ghc-devs 
>> <http://mail.haskell.org/cgi-bin/mailman/listinfo/ghc-devs>
> 

_______________________________________________
ghc-devs mailing list
ghc-devs@haskell.org
http://mail.haskell.org/cgi-bin/mailman/listinfo/ghc-devs

Reply via email to