diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2025-11-22 22:40:40 +0100 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2025-11-22 22:40:40 +0100 |
| commit | b4f07c673b7c710f5861bb84e67233c63336c53d (patch) | |
| tree | 234b0a4708fc899f8cfe529a3d897942c5964171 | |
| parent | d70848e2b6dbba365216700dbd37de40684918a5 (diff) | |
| -rw-r--r-- | src/CHAD/Example/GMM.hs | 2 | ||||
| -rw-r--r-- | src/CHAD/Simplify.hs | 1 |
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)) |