aboutsummaryrefslogtreecommitdiff
path: root/src/CHAD/Simplify.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/CHAD/Simplify.hs')
-rw-r--r--src/CHAD/Simplify.hs1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/CHAD/Simplify.hs b/src/CHAD/Simplify.hs
index 2510cc5..ea253d6 100644
--- a/src/CHAD/Simplify.hs
+++ b/src/CHAD/Simplify.hs
@@ -223,6 +223,7 @@ simplify'Rec = \case
-- TODO: more array shape
EShape _ (EBuild _ _ e1 e2) | not (hasAdds e2) -> acted $ simplify' e1
EShape _ (EMap _ e1 e2) | not (hasAdds e1) -> acted $ simplify' (EShape ext e2)
+ EShape _ (EReplicate1Inner _ en earr) -> acted $ simplify' (EPair ext (EShape ext earr) en)
-- TODO: more constant folding
EOp _ OIf (EConst _ STBool True) -> acted $ return (EInl ext STNil (ENil ext))