diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2025-11-22 22:41:09 +0100 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2025-11-22 22:41:09 +0100 |
| commit | 9b7c3eea7e34f5eb0d91f93b803e853028c2cec8 (patch) | |
| tree | 25b906bb49218d2743631d0c83e23717012e3b9b /src/CHAD/Example/GMM.hs | |
| parent | b4f07c673b7c710f5861bb84e67233c63336c53d (diff) | |
WIP: Think about fusionfusion
Diffstat (limited to 'src/CHAD/Example/GMM.hs')
| -rw-r--r-- | src/CHAD/Example/GMM.hs | 2 |
1 files changed, 1 insertions, 1 deletions
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)) |