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/Interpreter.hs | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/Interpreter.hs')

diff --git a/src/Interpreter.hs b/src/Interpreter.hs
index db7033d..79d5014 100644
--- a/src/Interpreter.hs
+++ b/src/Interpreter.hs
@@ -146,6 +146,10 @@ interpret'Rec env = \case
         sh `ShCons` n = arrayShape arr
     numericIsNum t $ return $
       arrayGenerate sh (\idx -> minimum [arrayIndex arr (idx `IxCons` i) | i <- [0 .. n-1]])
+  EReshape _ dim esh e -> do
+    sh <- unTupRepIdx ShNil ShCons dim <$> interpret' env esh
+    arr <- interpret' env e
+    return $ arrayReshape sh arr
   EFold1InnerD1 _ _ a b c -> do
     let t = typeOf b
     let f = \x y -> interpret' (V t y `SCons` V t x `SCons` env) a
-- 
cgit v1.2.3-70-g09d2