From 9b7c3eea7e34f5eb0d91f93b803e853028c2cec8 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Sat, 22 Nov 2025 22:41:09 +0100
Subject: WIP: Think about fusion

---
 src/CHAD/Example/GMM.hs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src/CHAD/Example/GMM.hs')

diff --git a/src/CHAD/Example/GMM.hs b/src/CHAD/Example/GMM.hs
index 18641e8..2b2ac2b 100644
--- a/src/CHAD/Example/GMM.hs
+++ b/src/CHAD/Example/GMM.hs
@@ -112,7 +112,7 @@ gmmObjective wrong = fromNamed $
         qmat q l = inline qmat' (SNil .$ q .$ l)
     in let_ #k2arr (unit #k2) $
        - #k1
-       + idx0 (sum1i (build1 #N $ #i :->
+       + idx0 (sum1i (build1 #N $ #i :-> recompute $
              logsumexp (build1 #K $ #k :->
                  #alpha ! pair nil #k
                  + idx0 (sum1i (#Q .! #k))
-- 
cgit v1.2.3-70-g09d2