pretty: Print build dimension

1 files changed, 3 insertions, 2 deletions

diff --git a/src/AST/Pretty.hs b/src/AST/Pretty.hs
index 01dfcf8..fb5e138 100644
--- a/src/AST/Pretty.hs
+++ b/src/AST/Pretty.hs
@@ -159,13 +159,14 @@ ppExpr' d val expr = case expr of
     a' <- ppExpr' 11 val a
     name <- genNameIfUsedIn' "i" (tTup (sreplicate n tIx)) IZ b
     e' <- ppExpr' 0 (Const name `SCons` val) b
+    let primName = ppString ("build" ++ intSubscript (fromSNat n))
     return $ ppParen (d > 0) $
       group $ flatAlt
         (hang 2 $
-           annotate AHighlight (ppString "build") <> ppX expr <+> a'
+           annotate AHighlight primName <> ppX expr <+> a'
            <+> ppString "$" <+> ppString "\\" <> ppString name <+> ppString "->"
            <> hardline <> e')
-        (ppApp (annotate AHighlight (ppString "build") <> ppX expr) [a', ppLam [ppString name] e'])
+        (ppApp (annotate AHighlight primName <> ppX expr) [a', ppLam [ppString name] e'])
 
   EFold1Inner _ cm a b c -> do
     name1 <- genNameIfUsedIn (typeOf a) (IS IZ) a