Skip to content

Commit

Permalink
ScopedSnocList: WIP: rewrites to Snock ANF and LambdaLift
Browse files Browse the repository at this point in the history
  • Loading branch information
GulinSS committed Oct 1, 2024
1 parent a14a3bf commit bbf9475
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 9 deletions.
2 changes: 1 addition & 1 deletion src/Compiler/ANF.idr
Original file line number Diff line number Diff line change
Expand Up @@ -264,7 +264,7 @@ toANF : LiftedDef -> Core ANFDef
toANF (MkLFun args scope sc)
= do v <- newRef Next (the Int 0)
(iargs, vsNil) <- bindArgs args []
let vs : AVars args = rewrite sym (appendNilRightNeutral args) in
let vs : AVars args = rewrite sym (appendLinLeftNeutral args) in
vsNil
(iargs', vs) <- bindArgs scope vs
pure $ MkAFun (iargs ++ reverse iargs') !(anf vs sc)
Expand Down
23 changes: 15 additions & 8 deletions src/Compiler/LambdaLift.idr
Original file line number Diff line number Diff line change
Expand Up @@ -358,10 +358,10 @@ initUsed : {vars : _} -> Used vars
initUsed {vars} = MkUsed (replicate (length vars) False)

lengthDistributesOverAppend
: (xs, ys : List a)
-> length (xs ++ ys) = length xs + length ys
lengthDistributesOverAppend [] ys = Refl
lengthDistributesOverAppend (x :: xs) ys =
: (xs, ys : SnocList a)
-> length (ys ++ xs) = length xs + length ys
lengthDistributesOverAppend [<] ys = Refl
lengthDistributesOverAppend (xs :< x) ys =
cong S $ lengthDistributesOverAppend xs ys

weakenUsed : {outer : _} -> Used vars -> Used (vars ++ outer)
Expand Down Expand Up @@ -581,11 +581,18 @@ mutual
LConCase fc (dropUnused unused sc) alts' (map (dropUnused unused) def)
where
dropConCase : LiftedConAlt (vars ++ outer) ->
LiftedConAlt ((dropped vars unused) ++ outer)
LiftedConAlt (dropped vars unused ++ outer)
dropConCase (MkLConAlt n ci t args sc) =
let sc' = (rewrite sym $ appendAssociative args outer vars in sc)
droppedSc = dropUnused {vars=vars} {outer=outer++args} unused sc' in
MkLConAlt n ci t args (rewrite appendAssociative args outer (dropped vars unused) in droppedSc)
MkLConAlt n ci t args (?sdf droppedSc)
where
sc' : Lifted (vars ++ (outer ++ args))
sc' = (rewrite appendAssociative vars outer args in sc)

droppedSc : Lifted ((dropped vars unused ++ outer) ++ args)
droppedSc = do
rewrite sym $ appendAssociative (dropped vars unused) outer args
dropUnused {vars=vars} {outer=outer++args} unused sc'

dropUnused {vars} {outer} unused (LConstCase fc sc alts def) =
let alts' = map dropConstCase alts in
LConstCase fc (dropUnused unused sc) alts' (map (dropUnused unused) def)
Expand Down

0 comments on commit bbf9475

Please sign in to comment.