Add IR parsing for ArrayVal.

diku-dk · Jul 3, 2024 · 402493f · 402493f
1 parent 4800376
commit 402493f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/Futhark/IR/Parse.hs b/src/Futhark/IR/Parse.hs
@@ -314,6 +314,10 @@ pBasicOp =
  <*> pFlatSlice
  <* lexeme "="
  <*> pVName,
+ try $
+ ArrayVal
+ <$> brackets (pPrimValue `sepBy` pComma)
+ <*> (lexeme ":" *> "[]" *> pPrimType),
  ArrayLit
  <$> brackets (pSubExp `sepBy` pComma)
  <*> (lexeme ":" *> "[]" *> pType),

diff --git a/src/Futhark/IR/Pretty.hs b/src/Futhark/IR/Pretty.hs
@@ -198,7 +198,11 @@ instance Pretty BasicOp where
  <+> colon
  <+> "[]"
  <> pretty rt
- pretty (ArrayVal vs _) = brackets $ commasep $ map pretty vs
+ pretty (ArrayVal vs t) =
+ brackets (commasep $ map pretty vs)
+ <+> colon
+ <+> "[]"
+ <> pretty t
  pretty (BinOp bop x y) = pretty bop <> parens (pretty x <> comma <+> pretty y)
  pretty (CmpOp op x y) = pretty op <> parens (pretty x <> comma <+> pretty y)
  pretty (ConvOp conv x) =