From 81d88dbc430ca6ec8390636f8b7162887b390873 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Mon, 3 Nov 2025 23:09:37 +0100
Subject: WIP map + zip

---
 src/Compile.hs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'src/Compile.hs')

diff --git a/src/Compile.hs b/src/Compile.hs
index bf7817a..d6ad7ec 100644
--- a/src/Compile.hs
+++ b/src/Compile.hs
@@ -802,6 +802,15 @@ compile' env = \case
 
     return (CELit arrname)
 
+  -- TODO: actually generate decent code here
+  EMap _ e1 e2 -> do
+    let STArr n _ = typeOf e2
+    compile' env $
+      elet e2 $
+        EBuild ext n (EShape ext (evar IZ)) $
+          elet (EIdx ext (evar (IS IZ)) (EVar ext (tTup (sreplicate n tIx)) IZ)) $
+            weakenExpr (WCopy (WSink .> WSink)) e1
+
   EFold1Inner _ commut efun ex0 earr -> do
     let STArr (SS n) t = typeOf earr
 
@@ -951,6 +960,16 @@ compile' env = \case
               [("buf", CEProj (CELit arrname) "buf")
               ,("sh", CELit ("{" ++ intercalate ", " [printCExpr 0 e "" | e <- indexTupleComponents dim shname] ++ "}"))])
 
+  -- TODO: actually generate decent code here
+  EZip _ e1 e2 -> do
+    let STArr n _ = typeOf e1
+    compile' env $
+      elet e1 $
+      elet (weakenExpr WSink e2) $
+        EBuild ext n (EShape ext (evar (IS IZ))) $
+          EPair ext (EIdx ext (evar (IS (IS IZ))) (EVar ext (tTup (sreplicate n tIx)) IZ))
+                    (EIdx ext (evar (IS IZ)) (EVar ext (tTup (sreplicate n tIx)) IZ))
+
   EFold1InnerD1 _ commut efun ex0 earr -> do
     let STArr (SS n) t = typeOf earr
         STPair _ bty = typeOf efun
-- 
cgit v1.2.3-70-g09d2