From 4d456e4d34b1e4fb3725051d1b8a0c376b704692 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Thu, 30 Oct 2025 15:56:35 +0100
Subject: Implement reshape

---
 src/Analysis/Identity.hs | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/Analysis')

diff --git a/src/Analysis/Identity.hs b/src/Analysis/Identity.hs
index 9dc8811..b3a6664 100644
--- a/src/Analysis/Identity.hs
+++ b/src/Analysis/Identity.hs
@@ -244,6 +244,12 @@ idana env expr = case expr of
     res <- VIArr <$> genId <*> pure sh
     pure (res, EMinimum1Inner res e1')
 
+  EReshape _ dim e1 e2 -> do
+    (v1, e1') <- idana env e1
+    (_, e2') <- idana env e2
+    res <- VIArr <$> genId <*> shidsToVec dim v1
+    pure (res, EReshape res dim e1' e2')
+
   EFold1InnerD1 _ cm e1 e2 e3 -> do
     let t1 = typeOf e2
     x1 <- genIds t1
-- 
cgit v1.2.3-70-g09d2