From b4f07c673b7c710f5861bb84e67233c63336c53d Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Sat, 22 Nov 2025 22:40:40 +0100
Subject: Add a simplify rule

---
 src/CHAD/Example/GMM.hs | 2 +-
 src/CHAD/Simplify.hs    | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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))
-- 
cgit v1.2.3-70-g09d2