aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorTom Smeding <tom@tomsmeding.com>2025-11-22 22:40:40 +0100
committerTom Smeding <tom@tomsmeding.com>2025-11-22 22:40:40 +0100
commitb4f07c673b7c710f5861bb84e67233c63336c53d (patch)
tree234b0a4708fc899f8cfe529a3d897942c5964171 /src
parentd70848e2b6dbba365216700dbd37de40684918a5 (diff)
Add a simplify ruleHEADmaster
Diffstat (limited to 'src')
-rw-r--r--src/CHAD/Example/GMM.hs2
-rw-r--r--src/CHAD/Simplify.hs1
2 files changed, 2 insertions, 1 deletions
diff --git a/src/CHAD/Example/GMM.hs b/src/CHAD/Example/GMM.hs
index 8f834e0..18641e8 100644
--- a/src/CHAD/Example/GMM.hs
+++ b/src/CHAD/Example/GMM.hs
@@ -3,7 +3,7 @@
{-# LANGUAGE TypeApplications #-}
module CHAD.Example.GMM where
-import CHAD.Data (SList(..), SNat(..))
+import CHAD.Data (SList(..))
import CHAD.Example.Types
import CHAD.Language
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))