enable argument types in pga_{un}flatten

[ghc-base.git] / GHC / HetMet / CodeTypes.hs

diff --git a/GHC/HetMet/CodeTypes.hs b/GHC/HetMet/CodeTypes.hs
index 424551b..255e8b4 100644 (file)
--- a/GHC/HetMet/CodeTypes.hs
+++ b/GHC/HetMet/CodeTypes.hs
@@ -5,6 +5,7 @@ module GHC.HetMet.CodeTypes (
   hetmet_csp,
   hetmet_flatten,
   pga_flatten,
+  pga_unflatten,
   pga_flattened_id,
   GuestIntegerLiteral, guestIntegerLiteral,
   GuestStringLiteral, guestStringLiteral,
@@ -30,11 +31,11 @@ hetmet_csp = Prelude.error "hetmet_csp should never be evaluated; did you forget
 
 hetmet_flatten ::
    forall g .
-    GArrowSTKC g (,) () =>
+    GArrowSTKC g =>
       forall x y.
-         <[ x -> y ]>@g
+         <[ y ]>@g
          ->
-         (g x y)
+         (g (GArrowUnit g) y)
 hetmet_flatten x = unG (pga_flatten x)
 
 -- After the flattening pass the argument and result types of this
@@ -47,6 +48,11 @@ pga_flatten ::
      <[ x -> y ]>@g ->
      PGArrow g x y
 pga_flatten = error "hetmet_flatten should never be evaluated; did you forget to compile with -fcoqpass?"
+pga_unflatten ::
+   forall g x y.
+     PGArrow g x y ->
+     <[ x -> y ]>@g
+pga_unflatten = error "hetmet_flatten should never be evaluated; did you forget to compile with -fcoqpass?"
 
 pga_flattened_id ::
    forall g x y.